import gradio as gr
import joblib

# Load the model and vectorizer
model = joblib.load("scam_classifier_model.joblib")
vectorizer = joblib.load("tfidf_vectorizer.joblib")

# Define prediction function
def predict_sms(text):
    vect = vectorizer.transform([text])
    pred = model.predict(vect)[0]
    return f"Prediction: {pred}"

# Gradio interface
demo = gr.Interface(
    fn=predict_sms,
    inputs=gr.Textbox(lines=4, placeholder="Andika ujumbe wa SMS hapa..."),
    outputs="text",
    title="Bongo SMS Scam Detector",
    description="Changanua kama ujumbe wa SMS ni wa 'scam' au 'trust'."
)

demo.launch()