Update app.py

app.py

@@ -1,38 +1,50 @@
-import requests
 import gradio as gr
-import spaces 
-import os 
-
+import requests
+import os
+import random
 
+# Configuration de l'API
+API_URL = "https://api-inference.huggingface.co/models/gpt2"
 api_token = os.environ.get("TOKEN")
-
-
-API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-llama-3-8B-Instruct"
 headers = {"Authorization": f"Bearer {api_token}"}
 
-
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
-def detect_sentiment(message):
-    prompt = f"Détecte le sentiment de ce message. Réponds par 'positif' ou 'négatif' :\nMessage : \"{message}\""
+def generate_yes_no_response(question):
+    # Créez un prompt qui demande explicitement une réponse Oui/Non
+    prompt = f"Réponds par 'Oui' ou 'Non' à la question suivante : {question}"
+    
+    # Faites la requête à l'API
     response = query({"inputs": prompt})
-    sentiment = response[].['generated_text'].strip().lower()
-    return sentiment
-
-def sentiment_analysis_interface(message):
-    sentiment = detect_sentiment(message)
-    return sentiment
-
-# Créer l'interface Gradio
-iface = gr.Interface(
-    fn=sentiment_analysis_interface,
-    inputs="text",
-    outputs="text",
-    title="Détection de Sentiment",
-    description="Entrez un message pour détecter si le sentiment est positif ou négatif.",
+    
+    # Extrayez le texte de la réponse
+    if isinstance(response, list) and len(response) > 0:
+        full_response = response[0].get('generated_text', '')
+    else:
+        full_response = str(response)  # Fallback si la réponse n'est pas comme prévu
+    
+    # Analysez la réponse pour extraire Oui ou Non
+    if "oui" in full_response.lower():
+        return "Oui"
+    elif "non" in full_response.lower():
+        return "Non"
+    else:
+        # Si la réponse n'est ni oui ni non, choisissez aléatoirement
+        return random.choice(["Oui", "Non"])
+
+# Fonction pour l'interface Gradio
+def chatbot(message, history):
+    response = generate_yes_no_response(message)
+    return response
+
+# Création de l'interface Gradio
+iface = gr.ChatInterface(
+    fn=chatbot,
+    title="Chatbot Oui/Non",
+    description="Posez une question, et je répondrai par Oui ou Non."
 )
 
-# Lancer l'interface
+# Lancement de l'interface
 iface.launch()