| 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() |