import gradio as gr
import pandas as pd
import numpy as np 
import joblib
from deep_translator import GoogleTranslator

model = joblib.load('nb.joblib')
vectorizer = joblib.load('vectorizer.joblib')
encoder = joblib.load('encoder.joblib')

def pred(message):
    message =  GoogleTranslator(source= 'fr', target= 'en').translate(message)
    vecteur = vectorizer.transform([message])
    prediction = model.predict(vecteur)[0]
    
    label = f"{'SPAM' if prediction == 1 else 'HAM'}"
    return label


interface = gr.Interface(
    fn = pred,
    inputs= gr.Textbox(lines= 3, placeholder= "Tape ton message"), 
    outputs= gr.Textbox(label= "Resultat"),
    title= "Detections des SPAM par IA",
    description= "Entrez un message et l'application vous dira s'il s'agit de SPAM ou de HAM"
)

if __name__ == "__main__":
    interface.launch()