|
import gradio as gr |
|
import requests |
|
|
|
API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1" |
|
import os |
|
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 quelques mots sensoriels : {mots}. |
|
Rédige une suggestion hypnotique de 5 phrases. |
|
Chaque phrase doit évoquer une sensation corporelle ou symbolique, dans un style poétique, immersif et doux. |
|
Termine toujours par •°○ |
|
""" |
|
|
|
payload = { |
|
"inputs": prompt, |
|
"parameters": { |
|
"max_new_tokens": 160, |
|
"temperature": 0.85, |
|
"top_p": 0.9, |
|
"do_sample": True |
|
} |
|
} |
|
|
|
response = requests.post(API_URL, headers=headers, json=payload) |
|
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 : aucune réponse utile." |
|
|
|
iface = gr.Interface( |
|
fn=generate_suggestion, |
|
inputs=gr.Textbox(label="Liste de mots sensoriels (séparés par virgule)"), |
|
outputs="text", |
|
title="Résonance Hypnopoétique •°○" |
|
) |
|
|
|
iface.launch() |