from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
import gradio as gr

model_id = "HuggingFaceH4/zephyr-1.3b"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)

chatbot = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512)

def repondre(prompt):
    system_prompt = """Tu es un assistant spécialisé en éthique et en critique de l'intelligence artificielle.
    Réponds de manière claire, structurée, et sans inventer de faits.
    Question : """
    full_prompt = system_prompt + prompt
    result = chatbot(full_prompt)[0]["generated_text"]
    return result[len(full_prompt):]

interface = gr.Interface(
    fn=repondre,
    inputs=gr.Textbox(label="Pose ta question critique sur l'IA"),
    outputs="text",
    title="Assistant IA Critique (Zephyr 1.3B)",
    description="Ce chatbot répond à des questions critiques sur l'IA, les biais, l'éthique et les données."
)

interface.launch()