Update app.py

app.py

@@ -36,9 +36,9 @@ emotion_labels = {
     "sadness": "tristeza",
     "joy": "alegria",
     "anger": "raiva",
-    "fear": "medo",
-    "disgust": "repulsa",
-    "surprise": "surpresa",
+    "fear": "ansiedade",
+    "disgust": "insegurança",
+    "surprise": "alegria",
     "others": "neutro"
 }
 
@@ -49,9 +49,8 @@ def gerar_sugestao(emotion_pt):
         "depressão": "Você não está sozinho. Considere conversar com alguém de confiança ou buscar apoio profissional.",
         "alegria": "Continue aproveitando esse momento positivo e compartilhe boas energias!",
         "raiva": "Afaste-se da situação e respire antes de reagir. Canalize essa energia em algo produtivo.",
-        "medo": "Tente identificar o que está te causando insegurança e dê pequenos passos para enfrentá-lo.",
-        "repulsa": "Evite focar em situações desagradáveis. Redirecione sua atenção para algo positivo.",
-        "surpresa": "Abrace o inesperado! Às vezes, surpresas podem trazer aprendizados valiosos.",
+        "ansiedade": "Tente identificar o que está te causando insegurança e dê pequenos passos para enfrentá-lo.",
+        "insegurança": "Analise os pontos que causam insegurança e busque soluções práticas.",
         "neutro": "Mantenha o equilíbrio emocional e cuide de si mesmo.",
         "desconhecido": "Emoção não identificada com precisão."
     }
@@ -60,11 +59,11 @@ def gerar_sugestao(emotion_pt):
 # ====== FALLBACK LOCAL ======
 def fallback_emotion(text):
     emotion_map = {
-        "tristeza": ["triste", "desanimado", "deprimido"],
-        "ansiedade": ["ansioso", "preocupado", "tenso"],
+        "tristeza": ["triste", "desanimado", "deprimido", "melancólico"],
+        "ansiedade": ["ansioso", "preocupado", "tenso", "nervoso"],
         "insegurança": ["inseguro", "incerto", "receoso"],
-        "raiva": ["irritado", "zangado", "revoltado"],
-        "alegria": ["feliz", "animado", "contente"]
+        "raiva": ["irritado", "zangado", "revoltado", "furioso"],
+        "alegria": ["feliz", "animado", "contente", "alegre"]
     }
     text_lower = text.lower()
     for emotion, keywords in emotion_map.items():
@@ -122,7 +121,6 @@ def analyze():
         if emotion_pt == "tristeza" and confidence >= 0.9:
             emotion_pt = "depressão"
 
-        # Geração de resposta
         response = {
             "status": "ok",
             "emotion": emotion_pt,
@@ -142,4 +140,3 @@ def analyze():
 # ====== EXECUÇÃO ======
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)
-