import gradio as gr
import requests
import os

API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta"
HF_TOKEN = os.environ["HF_TOKEN"]
headers = {"Authorization": f"Bearer {HF_TOKEN}"}

def generate_suggestion(mots):
    prompt = f"""
Tu es un thérapeute-poète.
Voici des mots sensoriels ou symboliques : {mots}.
Rédige 5 phrases poétiques, variées, sensibles.
Aucune structure figée. Évoque des sensations, des images, des émotions.
Termine par ☆
"""
    response = requests.post(API_URL, headers=headers, json={
        "inputs": prompt,
        "parameters": {
            "temperature": 0.95,
            "top_p": 0.9,
            "max_new_tokens": 180,
            "do_sample": True
        }
    })

    output = response.json()
    if isinstance(output, list) and "generated_text" in output[0]:
        return output[0]["generated_text"]
    elif "generated_text" in output:
        return output["generated_text"]
    else:
        return "⚠️ Erreur de génération IA."

iface = gr.Interface(
    fn=generate_suggestion,
    inputs=gr.Textbox(label="Mots ou question", placeholder="frisson, vertige, lumière..."),
    outputs="text",
    title="Résonance Hypnopoétique •°○",
    description="Une suggestion poétique inspirée de vos mots-clés sensoriels ou symboliques."
)

iface.launch()