import gradio as gr
import joblib

with open('intent_classifier', 'rb') as file:
    model = joblib.load(file)

with open('c_intent_vectorizer', 'rb') as file:
    vectorizer = joblib.load(file)

def preprocess_input(data):
  tokens = data.lower().split()
  filtered = [text for text in tokens if text.isalpha()]
  return ' '.join(filtered)

def predict(text):
    model_input = preprocess_input(text)
    vectorized_input = vectorizer.transform([model_input])
    prediction = model.predict(vectorized_input)
    return prediction
    

demo = gr.Interface(fn=predict, inputs="textbox", outputs="textbox")

demo.launch()