import gradio as gr
from transformers import pipeline

# Création du pipeline de génération
model = pipeline("text2text-generation", model="t5-small")

# Fonction d'inférence
def chat(prompt):
    try:
        result = model(prompt, max_length=100)
        return result[0]["generated_text"]
    except Exception as e:
        return f"Erreur : {str(e)}"

# Interface Gradio
interface = gr.Interface(
    fn=chat,
    inputs=gr.Textbox(lines=2, placeholder="Écris quelque chose..."),
    outputs="text",
    title="T5 Chatbot",
    description="API de génération avec T5-small"
)

# Lancement du serveur Gradio
interface.launch(server_name="0.0.0.0", server_port=7860)