import fasttext
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="lid.218.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
    text = text.replace('\n', ' ').strip()
    
    # Effectuer la prédiction de la langue
    labels, probs = model.predict(text)
    
    # Retourner la langue prédite et la probabilité correspondante
    return labels[0], probs[0]

# 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
)

# Lancer l'application Gradio
if __name__ == "__main__":
    iface.launch(share=True)