from transformers import pipeline

# Charger un modèle spécifique au français
@st.cache_resource
def load_model():
    return pipeline("text-generation", model="flaubert/flaubert_base_cased")

# Créer l'interface Streamlit
st.title("Chatbot Restaurant 🍽️")
st.write("Posez vos questions sur le restaurant !")

# Demander à l'utilisateur de saisir une question
user_input = st.text_input("Votre question :")

# Si l'utilisateur a posé une question
if user_input:
    bot = load_model()  # Charger le modèle
    prompt = f"Tu es un chatbot pour un restaurant. Réponds de manière amicale, polie et précise. Voici la question : {user_input}"

    # Générer la réponse du modèle
    response = bot(prompt, max_length=50, do_sample=False)[0]["generated_text"]

    # Afficher la réponse
    st.write(response)