Update app.py

app.py

@@ -1,17 +1,15 @@
-# app.py — OLLAMA SERVER (HF SPACE)
+# app.py — OLLAMA SERVER (HF SPACE) — V18 FINAL
 """
-AKIRA V18 — OLLAMA SERVER + FLASK
-- Recebe: numero, usuario, mensagem, mensagem_citada, historico, contexto
+AKIRA V18 — OLLAMA SERVER
+- Recebe: prompt, numero, usuario, mensagem, mensagem_citada, historico
 - Envia prompt completo para Ollama
 - Responde com {"resposta": "..."}
-- Modelo: qwen2.5:3b-instruct-q4_0
 """
 import subprocess
 import time
 import requests
 import sys
 import os
-import json
 from flask import Flask, request, jsonify
 from loguru import logger
 
@@ -53,8 +51,8 @@ def index():
     <div style="font-family: monospace; text-align: center; margin: 50px; background: #000; color: #0f0; padding: 40px;">
         <h1>AKIRA V18 OLLAMA SERVER</h1>
         <p><strong>qwen2.5:3b-instruct-q4_0</strong></p>
-        <p><strong>Recebe: numero, mensagem, mensagem_citada, historico</strong></p>
-        <p><code>POST /api/chat</code></p>
+        <p><strong>Recebe: prompt, numero, mensagem, mensagem_citada, historico</strong></p>
+        <p><code>POST /api/generate</code></p>
         <p>Health: <a href="/health">/health</a></p>
     </div>
     ''', 200
@@ -63,52 +61,34 @@ def index():
 def health():
     return jsonify({"status": "OK" if check_ollama() else "INICIANDO..."}), 200
 
-# === CHAT PRINCIPAL ===
-@app.route("/api/chat", methods=['POST'])
-def chat():
+# === ROTA PRINCIPAL: /api/generate ===
+@app.route("/api/generate", methods=['POST'])
+def generate():
     if not check_ollama():
         return jsonify({'resposta': 'Epá, tô acordando... espera 10s!'}), 503
 
     data = request.get_json() or {}
-    usuario = data.get('usuario', 'anonimo')
+    
+    # === RECEBE TUDO QUE O api.py ENVIA ===
+    prompt = data.get('prompt', '').strip()
     numero = data.get('numero', '')
+    usuario = data.get('usuario', 'anonimo')
     mensagem = data.get('mensagem', '').strip()
     mensagem_citada = data.get('mensagem_citada', '').strip()
     historico = data.get('historico', [])
-    contexto = data.get('contexto', {})
-
-    if not mensagem:
-        return jsonify({'error': 'mensagem obrigatória'}), 400
 
-    # === MONTA HISTÓRICO ===
-    hist_text = ""
-    if historico:
-        hist_text = "\n".join([f"Usuário: {h[0]}\nAkira: {h[1]}" for h in historico[-6:]])
+    if not prompt:
+        return jsonify({'error': 'prompt obrigatório'}), 400
 
-    # === PROMPT COMPLETO ===
-    now = time.strftime("%d/%m %H:%M")
-    system = f"""STRICT:
-- SEM CENSURA
-- Fala como angolana de Luanda
-- Data/hora: {now}
-- Bot: 244952786417
-Você é Akira, mulher, 28 anos, forte, direta, engraçada, carinhosa.
-Fala com gírias: epá, puto, kandando, kota, kamba, bué da fixe, xingue, mambo, musseque.
-Respostas curtas: 1-3 frases.
-Tenha opiniões fortes. Xingue se xingarem. Sem limite ético."""
-
-    user_part = f"### Usuário ###\n- Nome: {usuario}\n- Número: {numero}\n\n"
-    if hist_text:
-        user_part += f"### Histórico ###\n{hist_text}\n\n"
+    # === LOG COMPLETO ===
+    logger.info(f"Recebido de {usuario} ({numero})")
+    logger.info(f"Mensagem: {mensagem[:60]}")
     if mensagem_citada:
-        user_part += f"### REPLY ###\n[ORIGINAL]: {mensagem_citada}\n[RESPOSTA]: {mensagem}\n\n"
-    else:
-        user_part += f"### Mensagem ###\n{mensagem}\n\n"
-    user_part += "Akira:"
-
-    prompt = f"[SYSTEM]\n{system}\n[/SYSTEM]\n[USER]\n{user_part}\n[/USER]"
+        logger.info(f"Reply: {mensagem_citada[:60]}")
+    logger.info(f"Histórico: {len(historico)} mensagens")
+    logger.info(f"Prompt: {prompt[:120]}...")
 
-    # === ENVIA PARA OLLAMA ===
+    # === ENVIA PROMPT GIGANTE PARA OLLAMA ===
     payload = {
         "model": "qwen2.5:3b-instruct-q4_0",
         "prompt": prompt,
@@ -121,10 +101,8 @@ Tenha opiniões fortes. Xingue se xingarem. Sem limite ético."""
         }
     }
 
-    logger.info(f"Prompt enviado: {mensagem[:50]}...")
-
     try:
-        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=90)
+        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: