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