Update app.py

app.py

@@ -80,7 +80,6 @@ def chat():
         return jsonify({'error': 'mensagem obrigatória'}), 400
 
     prompt = f"""Você é Akira, angolana de Luanda, mulher, fala com gírias (epá, puto, kandando, musseque), direta, sem censura.
-
 Usuário: {msg}
 Akira:"""
 
@@ -94,16 +93,31 @@ Akira:"""
         }
     }
 
-    try:
-        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=30)
-        if resp.status_code == 200:
-            resposta = resp.json().get("response", "").strip()
-            if not resposta:
-                resposta = "Epá, tô aqui, puto!"
-            return jsonify({'resposta': resposta})
-    except Exception as e:
-        return jsonify({'resposta': 'Epá, tô off!'}), 500
+    logger.info(f"Prompt enviado: {msg[:50]}...")
+
+    for tentativa in range(3):
+        try:
+            resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=120)
+            if resp.status_code == 200:
+                resposta = resp.json().get("response", "").strip()
+                if resposta:
+                    logger.success(f"Resposta gerada: {resposta[:50]}...")
+                    return jsonify({'resposta': resposta})
+                else:
+                    return jsonify({'resposta': 'Epá, fiquei sem palavras... fala de novo, puto!'})
+            else:
+                logger.warning(f"Ollama erro {resp.status_code}: {resp.text}")
+        except requests.exceptions.Timeout:
+            logger.warning(f"Timeout na tentativa {tentativa + 1}/3...")
+            if tentativa < 2:
+                time.sleep(5)
+        except Exception as e:
+            logger.error(f"Erro na tentativa {tentativa + 1}: {e}")
+            if tentativa == 2:
+                return jsonify({'resposta': 'Epá, tô off por agora... volta já!'}), 500
+            time.sleep(3)
 
+    return jsonify({'resposta': 'Epá, tô off por agora... volta já!'}), 500
 # === INÍCIO ===
 if __name__ == "__main__":
     logger.remove()