import gradio as gr
import fasttext
import os
from huggingface_hub import hf_hub_download

# Téléchargement du modèle depuis Hugging Face
model_path = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="lid.176.bin")

# Vérifier si le modèle a bien été téléchargé
if not os.path.exists(model_path):
    raise ValueError(f"Le modèle {model_path} n'a pas pu être téléchargé!")

# Charger le modèle avec FastText
model = fasttext.load_model(model_path)

# Fonction de prédiction de la langue
def predict_language(text):
    prediction = model.predict(text)
    return prediction[0][0]  # Retourne la langue prédite

# Interface Gradio
iface = gr.Interface(fn=predict_language, inputs="text", outputs="text", live=True)

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