Update app.py

app.py

@@ -3,29 +3,54 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 
 # Télécharger le modèle depuis Hugging Face
-model_path = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="model.bin")
+model_path = hf_hub_download(
+    repo_id="facebook/fasttext-language-identification", 
+    filename="model.bin"
+)
 
 # Charger le modèle avec FastText
 model = fasttext.load_model(model_path)
 
 def predict_language(text):
-    # Supprimer les retours à la ligne et les espaces inutiles
+    if not text.strip():
+        return "", 0.0
+    
+    # Nettoyage du texte
     text = text.replace('\n', ' ').strip()
     
-    # Effectuer la prédiction de la langue
+    # Prédiction
     labels, probs = model.predict(text)
     
-    # Retourner la langue prédite et la probabilité correspondante
-    return labels[0], probs[0]
+    # Formatage du résultat
+    lang_code = labels[0].replace("__label__", "")
+    probability = round(float(probs[0]), 4)
+    
+    return lang_code, probability
 
-# Interface Gradio
-iface = gr.Interface(
-    fn=predict_language, 
-    inputs=gr.Textbox(label="Entrez votre texte ici"), 
-    outputs=[gr.Text(label="Langue prédite"), gr.Text(label="Probabilité")],
-    live=True
+# Interface Gradio améliorée
+interface = gr.Interface(
+    fn=predict_language,
+    inputs=gr.Textbox(
+        label="Texte à analyser",
+        placeholder="Écrivez ou collez votre texte ici..."
+    ),
+    outputs=[
+        gr.Text(label="Code langue"),
+        gr.Number(label="Confiance", precision=4)
+    ],
+    examples=[
+        ["Bonjour, comment ça va ?"], 
+        ["Hello, how are you?"], 
+        ["Hola, ¿cómo estás?"]
+    ],
+    title="Détection de Langue avec FastText",
+    description="Identifie la langue d'un texte entré. Supporte 176 langues différentes."
 )
 
-# Lancer l'application Gradio
+# Configuration de déploiement
 if __name__ == "__main__":
-    iface.launch(share=True)
+    interface.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False  # À désactiver en production
+    )