import pickle  # Pour sauvegarder et charger le modèle
# Gradio est une bibliothèque pour créer des interfaces utilisateur interactives
import gradio as gr

# Charger le modèle sauvegardé
with open('sms_spam_detector_model_fr.pkl','rb') as file:  
    model=pickle.load(file)
    
# Charger le vectorizer
with open('sms_vectorizer_fr.pkl','rb') as file: 
    vectorizer=pickle.load(file)
    
# Fonction pour prédire si un message est du spam ou non
def predict_spam(message):
    # Transformer le message en utilisant le vectorizer
    message_transformed = vectorizer.transform([message])
    prediction = model.predict(message_transformed)
    return "SPAM ❌" if prediction[0] == 1 else "HAM ✅"

# Créer une interface Gradio pour l'application
interface_gradio = gr.Interface(fn=predict_spam, inputs="text", outputs="text",
                                title="Sms Spam Detector",
                                 description="print message to detect spam or ham."
                                 )
interface_gradio.launch()