Update app.py

app.py

@@ -15,7 +15,6 @@ if not firebase_key:
     raise RuntimeError("❌ FIREBASE_KEY não encontrado nos Secrets do Space.")
 
 try:
-    # Garante que o JSON da chave seja carregado corretamente
     cred_dict = json.loads(firebase_key)
     cred = credentials.Certificate(cred_dict)
     if not firebase_admin._apps:
@@ -35,49 +34,113 @@ if not HF_TOKEN:
 
 client = InferenceClient(token=HF_TOKEN)
 
-# =============================
-# Rota de teste
-# =============================
+# Mapeamento de emoções em português
+EMOTION_MAP = {
+    "joy": "alegria",
+    "sadness": "tristeza",
+    "anger": "raiva",
+    "fear": "medo",
+    "surprise": "surpresa",
+    "disgust": "desgosto",
+    "neutral": "neutro",
+    "depression": "depressão"  # nova emoção
+}
+
+# Sugestões simples conforme emoção
+SUGESTOES = {
+    "alegria": "Continue fazendo o que te faz bem!",
+    "tristeza": "Tente conversar com alguém de confiança.",
+    "raiva": "Respire fundo e dê um tempo antes de reagir.",
+    "medo": "Procure entender a origem do seu medo.",
+    "surpresa": "Nem tudo é previsível — abrace o inesperado.",
+    "desgosto": "Tente mudar o foco e procurar algo positivo.",
+    "depressão": "Considere buscar apoio emocional ou psicológico.",
+    "neutro": "Você parece equilibrado no momento."
+}
+
 @app.route("/")
 def index():
-    return jsonify({"status": "online", "message": "API Flask funcionando no Space!"})
+    return jsonify({"status": "online", "message": "API Flask funcionando!"})
 
 # =============================
-# Rota principal de análise de emoção
+# Função de normalização
+# =============================
+def normalizar_probabilidades(results):
+    total = sum([r["score"] for r in results])
+    if total == 0:
+        return results
+    for r in results:
+        r["score"] = r["score"] / total
+    return results
+
+# =============================
+# Rota principal /analyze
 # =============================
 @app.route("/analyze", methods=["POST"])
 def analyze():
     try:
         data = request.get_json()
         if not data or "text" not in data:
-            return jsonify({"error": "Campo 'text' é obrigatório"}), 400
+            return jsonify({"status": "error", "error": "Campo 'text' é obrigatório"}), 400
 
         text = data["text"]
 
-        # Chamada ao modelo Hugging Face
+        # Chama modelo HF
         response = client.text_classification(
             text,
             model="SamLowe/roberta-base-go_emotions"
         )
 
-        emotion = response[0]["label"]
-        score = round(response[0]["score"], 4)
+        # Normaliza scores
+        results = normalizar_probabilidades(response)
+
+        # Monta dicionário de probabilidades traduzidas
+        probabilidades = {}
+        for item in results:
+            label = item["label"].lower()
+            score = round(item["score"], 4)
 
-        result = {"emotion": emotion, "score": score}
+            # Regra adicional para identificar "depressão"
+            if label in ["sadness", "grief", "hopelessness"]:
+                label = "depression"
+
+            label_pt = EMOTION_MAP.get(label, "neutro")
+            probabilidades[label_pt] = score
+
+        # Define emoção principal
+        principal = max(probabilidades, key=probabilidades.get)
+        sugestao = SUGESTOES.get(principal, "Mantenha o equilíbrio emocional.")
+
+        result = {
+            "status": "ok",
+            "emotion": principal,
+            "emode": [principal],
+            "probabilities": probabilidades,
+            "suggestion": sugestao
+        }
 
         # Armazena no Firestore
         if db:
-            db.collection("emotions").add(result)
+            db.collection("emotions").add({
+                "text": text,
+                "emotion": principal,
+                "probabilities": probabilidades
+            })
 
         return jsonify(result)
 
     except Exception as e:
         print(f"❌ Erro na rota /analyze: {e}")
-        return jsonify({"error": str(e)}), 500
+        return jsonify({
+            "status": "fallback",
+            "emotion": "neutro",
+            "emode": ["neutro"],
+            "suggestion": "Não foi possível usar a API. Fallback ativado.",
+            "debug": str(e)
+        }), 500
+
 
-# =============================
-# Inicialização do servidor
-# =============================
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port, debug=False)
+