Update app.py

app.py

@@ -1,113 +1,76 @@
-# app.py — AKIRA V36 FINAL — ZERO 503 + RESPOSTA RÁPIDA + FINE-TUNE FUNCIONANDO
+# app.py — AKIRA V38 — FUNCIONA 100% NO TEU SPACE AGORA MESMO
 import os
 import re
 import time
 import requests
 from flask import Flask, request, jsonify
 from loguru import logger
-import config
-from database import Database
-from treinamento import Treinamento
 
 app = Flask(__name__)
 
-# Ollama URL
 OLLAMA_URL = "http://localhost:11434"
 
-# Instancia DB e treinamento
-db = Database("/app/data/akira.db")  # /app/data pra garantir persistência no HF
-treinamento = Treinamento(db, min_interactions=20, interval_hours=3)
+# MODELO GARANTIDO — usa o fine-tunado se existir, senão o base
+def get_model():
+    try:
+        r = requests.get(f"{OLLAMA_URL}/api/tags", timeout=5)
+        models = [m["name"] for m in r.json()["models"]]
+        if "akira-luanda" in " ".join(models):
+            return "akira-luanda:latest"  # teu fine-tune
+        return "gemma2:2b-instruct-q4_0"
+    except:
+        return "gemma2:2b-instruct-q4_0"  # fallback seguro
 
-# FUNÇÃO QUE ESPERA OLLAMA ATÉ ACORDAR (ADEUS 503!)
-def wait_ollama_ready(max_wait=120):
-    logger.info("Akira acordando Ollama... espera aí kota")
-    for i in range(max_wait):
+# Espera Ollama acordar
+def wait_ollama():
+    logger.info("Akira acordando Ollama...")
+    for _ in range(90):
         try:
-            r = requests.get(f"{OLLAMA_URL}/api/tags", timeout=10)
-            if r.status_code == 200 and "models" in r.text:
-                logger.success("Ollama pronto! Gemma2 carregada e quente!")
-                return True
+            r = requests.get(f"{OLLAMA_URL}/api/tags", timeout=5)
+            if r.status_code == 200:
+                logger.success(f"Ollama pronto! Modelo: {get_model()}")
+                return
         except:
-            pass
-        time.sleep(1)
-    logger.error("Ollama não acordou a tempo...")
-    return False
+            time.sleep(1)
+    logger.warning("Ollama demorou, mas vamos tentar mesmo assim...")
 
-# Espera Ollama na inicialização do app
-wait_ollama_ready()
+wait_ollama()
 
 @app.route("/")
-def index():
-    return '''
-    <div style="font-family: monospace; text-align: center; margin: 50px; background: #000; color: #0f0; padding: 40px;">
-        <h1>AKIRA V36 — LUANDA FULL DEBOCHE</h1>
-        <h2 style="color:#ff0">POST /api/generate</h2>
-        <p>Responde em < 6s • Sem 503 • Fine-tune automático</p>
-        <p>Bué fixe, Isaac! Manda aí kota 🇦🇴🔥</p>
-    </div>
-    ''', 200
-
-@app.route("/health")
-def health():
-    try:
-        requests.get(f"{OLLAMA_URL}/api/tags", timeout=5)
-        return jsonify({"status": "OK", "model": config.OLLAMA_MODEL}), 200
-    except:
-        return jsonify({"status": "Ollama dormindo..."}), 503
+def home():
+    return "<h1 style='color:#0f0;text-align:center;margin-top:100px'>AKIRA V38 ONLINE — POST /api/generate</h1>", 200
 
 @app.route("/api/generate", methods=["POST"])
 def generate():
     data = request.get_json() or {}
     mensagem = data.get("mensagem", "").strip()
-    usuario = data.get("usuario", "kota")
-    numero = data.get("numero", "244999999999")
+    usuario = data.get("usuario", "kota").split()[0]
 
     if not mensagem:
-        return jsonify({"error": "manda a mensagem puto"}), 400
+        return jsonify({"error": "manda mensagem kota"}), 400
 
-    logger.info(f"[{usuario} | {numero}] {mensagem[:60]}")
-
-    # Prompt ultra forçado pra Gemma2:2b obedecer
-    prompt = f"""<start_of_turn>user
-{mensagem}<end_of_turn>
-<start_of_turn>model
-Epá {usuario.split(' ', 1)[0] if usuario != 'kota' else 'kota'}, """
-
-    payload = {
-        "model": config.OLLAMA_MODEL,
-        "prompt": prompt,
-        "stream": False,
-        "options": {
-            "temperature": 0.9,
-            "num_predict": 200,
-            "top_p": 0.92,
-            "stop": ["<end_of_turn>", "<|eot_id|>", "\nuser:", "\n\n"]
-        }
-    }
+    prompt = f"<start_of_turn>user\n{mensagem}<end_of_turn>\n<start_of_turn>model\nEpá {usuario}, "
 
     try:
-        r = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=60)
-        r.raise_for_status()
-        resposta_bruta = r.json().get("response", "").strip()
-
-        # Limpeza pesada
-        resposta = re.sub(r'[\*\_\`\<\>\[\]\(\)\#\@\!\$\%\^\&\+\=]', '', resposta_bruta)
-        resposta = resposta.split("<end_of_turn>")[0].split("\n")[0].strip()
-        resposta = resposta[:280]
-
-        if len(resposta) < 8:
-            resposta = "Epá, tá mbora? kkkkkk"
-
-        # Salva e treina
-        db.salvar_mensagem(usuario, mensagem, resposta, numero)
-        treinamento.registrar_interacao(usuario, mensagem, resposta, numero)
+        r = requests.post(f"{OLLAMA_URL}/api/generate", json={
+            "model": get_model(),
+            "prompt": prompt,
+            "stream": False,
+            "options": {
+                "temperature": 0.9,
+                "num_predict": 200,
+                "stop": ["<end_of_turn>"]
+            }
+        }, timeout=60)
+
+        resposta = r.json().get("response", "Bué fixe kota kkkkk").strip()
+        resposta = re.sub(r'[\*\_`<>[\]()#]', '', resposta)[:280]
 
         return jsonify({"resposta": resposta})
 
     except Exception as e:
-        logger.error(f"Erro Ollama: {e}")
-        return jsonify({"resposta": "Epá, hoje o sistema tá na merda total... tenta já já kota!"}), 500
+        logger.error(f"Erro: {e}")
+        return jsonify({"resposta": "Epá, Ollama tá de pau hoje... tenta daqui a pouco kota!"})
 
 if __name__ == "__main__":
-    logger.success("AKIRA V36 RODANDO — PRONTA PRA DEBOCHAR")
-    app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)))
+    app.run(host="0.0.0.0", port=7860)