Update app.py

app.py

@@ -15,25 +15,24 @@ def query(payload):
 	response = requests.post(API_URL, headers=headers, json=payload)
 	return response.json()
 	
-
 def analyze_sentiment(text):
-    prompt = f"Tu es un analyseur de sentiment. Ton rôle est d'évaluer le sentiment général du texte fourni. Réponds uniquement par 'positif' ou 'négatif'. N'ajoute aucune explication. Voici le texte à analyser : {text}"
+    prompt = f"Analyse le sentiment de ce texte et réponds uniquement par 'positif' ou 'négatif' sans aucune autre explication : {text}"
     
     output = query({
         "inputs": prompt,
     })
     
     if isinstance(output, list) and len(output) > 0:
-        response = output[0].get('generated_text', '')
-        # Chercher 'positif' ou 'négatif' dans la réponse
-        sentiment = re.search(r'\b(positif|négatif)\b', response.lower())
-        if sentiment:
-            return sentiment.group()
+        response = output[0].get('generated_text', '').strip().lower()
+        if 'positif' in response:
+            return "positif"
+        elif 'négatif' in response:
+            return "négatif"
         else:
             return "Sentiment non détecté"
     else:
         return "Erreur: Réponse inattendue de l'API"
-
+    
 demo = gr.Interface(
     fn = analyze_sentiment,
     inputs=["text"],