Create app.py

app.py

@@ -0,0 +1,66 @@
+# -----------------------------
+# app.py pour déploiement Spam Detector
+# -----------------------------
+import gradio as gr
+import joblib
+import re
+from nltk.corpus import stopwords
+from nltk.stem import PorterStemmer
+import nltk
+nltk.download('stopwords')
+
+# -----------------------------
+# 1️⃣ Prétraitement d'un message
+# -----------------------------
+stop_words = set(stopwords.words('english'))
+stemmer = PorterStemmer()
+
+def preprocess_message(text):
+    """Prétraite un message pour qu'il corresponde au format du modèle."""
+    if not text:
+        return ""
+    text = text.lower()
+    text = re.sub(r'http\S+|www\S+', '', text)
+    text = re.sub(r'\S+@\S+', '', text)
+    text = re.sub(r'\+?\d[\d -]{8,}\d', '', text)
+    text = re.sub(r'\d+', '', text)
+    text = re.sub(r'[^a-z\s!/+>]', '', text)  # garder ponctuation utile pour spam
+    words = [stemmer.stem(word) for word in text.split() if word not in stop_words]
+    return " ".join(words)
+
+# -----------------------------
+# 2️⃣ Charger modèle et TF-IDF
+# -----------------------------
+model = joblib.load("spam_model.pkl")
+vectorizer = joblib.load("tfidf_vectorizer.pkl")
+
+# -----------------------------
+# 3️⃣ Fonction de prédiction
+# -----------------------------
+def predict_message(message):
+    cleaned = preprocess_message(message)
+    X = vectorizer.transform([cleaned])
+    prediction = model.predict(X)[0]
+    probability = model.predict_proba(X)[0][1] if hasattr(model, 'predict_proba') else None
+    return {
+        "Message": message,
+        "Prediction": prediction,
+        "Spam Probability": round(float(probability), 4) if probability is not None else None
+    }
+
+# -----------------------------
+# 4️⃣ Interface Gradio
+# -----------------------------
+iface = gr.Interface(
+    fn=predict_message,
+    inputs=gr.Textbox(lines=3, placeholder="Entrez votre message..."),
+    outputs="json",
+    title="📩 Spam Detector",
+    description="Entrez un message pour savoir s'il s'agit de spam ou non."
+)
+
+# -----------------------------
+# 5️⃣ Lancer l'application
+# -----------------------------
+if __name__ == "__main__":
+    iface.launch()