import gradio as gr
from transformers import pipeline

# GPT-2 public
generator = pipeline("text-generation", model="gpt2")

def respond(message, history):
    history = history or []

    # Générer la réponse
    answer = generator(message, max_length=50, do_sample=True)[0]["generated_text"]

    # Ajouter le tuple (user, bot) à l'historique
    history.append((message, answer))

    # Retourner la liste de tuples
    return history

# Gradio 3.43
iface = gr.ChatInterface(respond, title="Chat Demo")
iface.launch()