import streamlit as st
from transformers import pipeline

# Charger un modèle en français
@st.cache_resource
def load_model():
    return pipeline("text-generation", model="dbmdz/german-gpt2")  # Modèle en français

bot = load_model()

# Interface du chatbot
st.title("Chatbot Restaurant 🍽️")
st.write("Posez vos questions sur le restaurant !")

user_input = st.text_input("Votre question :")

if user_input:
    # Ajouter un prompt pour guider le modèle
    prompt = f"Tu es un chatbot pour un restaurant. Réponds aux questions de manière amicale et informative. Voici la question : {user_input}"
    
    # Générer la réponse du bot
    response = bot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
    
    # Afficher la réponse
    st.write(response)