Update app.py

app.py

@@ -5,7 +5,6 @@ from firebase_admin import credentials, firestore
 from flask import Flask, request, jsonify
 from huggingface_hub import InferenceClient
 
-# Inicialização do Flask
 app = Flask(__name__)
 
 # ====== CONFIGURAÇÃO FIREBASE ======
@@ -31,7 +30,7 @@ if not HF_TOKEN:
 
 client = InferenceClient(token=HF_TOKEN)
 
-# ====== MAPEAMENTO DE EMOÇÕES SUPORTADAS ======
+# ====== EMOÇÕES SUPORTADAS ======
 emotion_map = {
     "tristeza": ["triste", "desanimado", "desmotivado", "abatido", "deprimido"],
     "ansiedade": ["ansioso", "preocupado", "tenso", "nervoso", "agitado"],
@@ -40,7 +39,19 @@ emotion_map = {
     "alegria": ["feliz", "contente", "animado", "satisfeito", "positivo"]
 }
 
-# ====== FUNÇÃO DE FALLBACK LOCAL ======
+# ====== GERAR SUGESTÕES ======
+def gerar_sugestao(emotion):
+    sugestoes = {
+        "tristeza": "Procure fazer algo que te acalme e traga conforto emocional.",
+        "ansiedade": "Respire fundo e faça pausas para relaxar. Técnicas de respiração podem ajudar.",
+        "insegurança": "Acredite no seu valor. Lembre-se das suas conquistas.",
+        "raiva": "Afaste-se um pouco da situação e respire antes de reagir.",
+        "alegria": "Continue aproveitando esse momento positivo e compartilhe com os outros!",
+        "desconhecido": "Emoção não identificada com precisão."
+    }
+    return sugestoes.get(emotion, "Mantenha o equilíbrio emocional e cuide de você mesmo.")
+
+# ====== FALLBACK LOCAL ======
 def fallback_emotion(text):
     text_lower = text.lower()
     for emotion, keywords in emotion_map.items():
@@ -59,19 +70,7 @@ def fallback_emotion(text):
         "suggestion": "Emoção não identificada com precisão."
     }
 
-# ====== GERAR SUGESTÃO CONFORME EMOÇÃO ======
-def gerar_sugestao(emotion):
-    sugestoes = {
-        "tristeza": "Procure fazer algo que te acalme e traga conforto emocional.",
-        "ansiedade": "Respire fundo e faça pausas para relaxar. Técnicas de respiração podem ajudar.",
-        "insegurança": "Acredite no seu valor. Lembre-se das suas conquistas.",
-        "raiva": "Afaste-se um pouco da situação e respire antes de reagir.",
-        "alegria": "Continue aproveitando esse momento positivo e compartilhe com os outros!",
-        "desconhecido": "Emoção não identificada com precisão."
-    }
-    return sugestoes.get(emotion, "Mantenha o equilíbrio emocional e cuide de você mesmo.")
-
-# ====== ROTA PRINCIPAL DE TESTE ======
+# ====== ROTA PRINCIPAL ======
 @app.route("/")
 def home():
     return jsonify({"message": "API Flask funcionando no Space!", "status": "online"})
@@ -86,35 +85,32 @@ def analyze():
 
         text = data["text"]
 
-        # Tenta IA do Hugging Face
         try:
             result = client.text_classification(text)
             if result and isinstance(result, list) and len(result) > 0:
                 label = result[0].get("label", "").lower()
-
-                # Mapeamento IA → emoções suportadas
                 if label in emotion_map.keys():
                     emotion = label
                 else:
-                    emotion = "desconhecido"
-
-                return jsonify({
-                    "status": "ok",
-                    "emotion": emotion,
-                    "emode": [emotion],
-                    "suggestion": gerar_sugestao(emotion)
-                })
+                    # 🔁 Novo: força fallback se IA não reconheceu emoção suportada
+                    return jsonify(fallback_emotion(text))
             else:
-                # Fallback local
                 return jsonify(fallback_emotion(text))
 
+            return jsonify({
+                "status": "ok",
+                "emotion": emotion,
+                "emode": [emotion],
+                "suggestion": gerar_sugestao(emotion)
+            })
+
         except Exception as e:
-            print(f"⚠️ Erro na IA principal: {e}")
+            print(f"⚠️ Erro IA principal: {e}")
             return jsonify(fallback_emotion(text))
 
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-# ====== EXECUÇÃO LOCAL (porta 7860) ======
+# ====== EXECUÇÃO ======
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)