import gradio as gr
import requests
import os

API_URL = "https://api-inference.huggingface.co/models/gpt2"
api_token = os.environ.get("TOKEN")
headers = {"Authorization": f"Bearer {api_token}"}

def query(payload):
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()

def reponse(question):
    # Créez un prompt qui force le modèle à choisir entre Oui et Non
    prompt = {question}
    
   # Configurez les paramètres pour forcer une réponse courte
    payload = {
        "inputs": prompt,
        "parameters": {
            "max_new_tokens": 1,
            "return_full_text": False,
            "do_sample": True,
            "top_k": 2,
            "temperature": 0.7
        }
    }
    
    response = query(payload)
    
  
def chatbot(message, history):
    response = reponse(message)
    return response

iface = gr.ChatInterface(
    fn=chatbot,
    title="Chatbot Oui/Non",
    description="Posez une question, et je répondrai par Oui ou Non."
)

iface.launch()