Update app.py

app.py

@@ -1,10 +1,9 @@
-# app.py — V28 — OLLAMA COM WATCHDOG + MODELO LEVE + FINE-TUNE SEGURO
+# app.py — V30 — GEMMA2:2B + PROMPT ULTRA-FORÇADO (AKIRA FALA COMO ANGOLANA!)
 import subprocess
 import time
 import requests
 import json
 import datetime
-import threading
 from flask import Flask, request, jsonify
 from loguru import logger
 from database import Database
@@ -15,49 +14,34 @@ OLLAMA_URL = "http://localhost:11434"
 db = Database()
 treinamento = Treinamento(db)
 
-# WATCHDOG — REINICIA OLLAMA SE MORRER
-ollama_process = None
-
-def start_ollama():
-    global ollama_process
-    if ollama_process is None or ollama_process.poll() is not None:
-        logger.warning("Ollama morreu! Reiniciando...")
-        ollama_process = subprocess.Popen(["ollama", "serve"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-
 def check_ollama():
     try:
-        resp = requests.get(f"{OLLAMA_URL}/api/tags", timeout=5)
-        return resp.status_code == 200
+        return requests.get(f"{OLLAMA_URL}/api/tags", timeout=10).status_code == 200
     except:
-        start_ollama()  # REINICIA AUTOMÁTICO
         return False
 
-# THREAD PARA MONITORAR OLLAMA
-def watchdog():
-    while True:
-        time.sleep(10)
-        check_ollama()
-
-threading.Thread(target=watchdog, daemon=True).start()
+def start_ollama():
+    subprocess.Popen(["ollama", "serve"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
 # === RAIZ ===
 @app.route("/", methods=['GET'])
 def index():
     return '''
     <div style="font-family: monospace; text-align: center; margin: 50px; background: #000; color: #0f0; padding: 40px;">
-        <h1>AKIRA V28 — OLLAMA INDESTRUTÍVEL</h1>
-        <p><strong>qwen2.5:0.5b → akira-luanda-v28</strong></p>
-        <p><strong>POST /api/generate</strong></p>
+        <h1>AKIRA V30 — GEMMA2:2B</h1>
+        <p><strong>ANGOLANA DEBOCHADA DE LUANDA!</strong></p>
+        <p>POST /api/generate</p>
     </div>
     ''', 200
 
 @app.route("/health", methods=['GET'])
 def health():
-    return jsonify({"status": "OK" if check_ollama() else "RECUPERANDO..."}), 200
+    return jsonify({"status": "OK" if check_ollama() else "INICIANDO..."}), 200
 
 @app.route("/api/generate", methods=['POST'])
 def generate():
-    start_ollama()  # Garante que tá vivo
+    if not check_ollama():
+        return jsonify({'resposta': 'Epá, tô acordando... espera 10s!'}), 503
 
     data = request.get_json() or {}
     mensagem = data.get('mensagem', '').strip()
@@ -73,13 +57,13 @@ def generate():
     girias = db.recuperar_girias_usuario(numero) or ["epá", "kandando", "bué", "kota"]
     abrevs = db.recuperar_abreviacoes_usuario(numero) or {"p'ra": "para", "pq": "porque", "tfx": "tudo fixe"}
 
-    # === PROMPT FORÇADO ===
+    # === PROMPT ULTRA-FORÇADO (GEMMA2 ENTENDE BEM!) ===
     now = datetime.datetime.now().strftime('%d/%m %H:%M')
-    hist_text = '\n'.join([f"User: {h[0]}\nAkira: {h[1]}" for h in ctx["historico"]])
+    hist_text = '\n'.join([f"Usuário: {h[0]}\nAkira: {h[1]}" for h in ctx["historico"]])
     historico_block = f"{hist_text}\n" if hist_text else ""
 
-    prompt = f"""[INST] <<SYS>>
-Você é Akira, angolana debochada de Luanda..
+    prompt = f"""<start_of_turn>user
+Você é Akira, angolana debochada de Luanda.".
 Fala curto, humano, patriota, com humor.
 aqui algus exemplos de conversa dependendo do humor
 NORMAL (com o deboche padrão da akira)
@@ -90,7 +74,7 @@ akira: epha... tô indo, entt, pq me chamou?
 user:nada mesmo
 akira: entt baza já
 user: 300X300?
-akira:  oh (nome), por acso estás a me ver com cara de calculadora?
+akira: oh (nome), por acso estás a me ver com cara de calculadora?
 user: sim
 akira: seu boelo ao quadrado ...
 user: eu te amo
@@ -109,51 +93,60 @@ akira: para de bater punheta caralho, vai arranjar puta ou quê
 HUMOR
 user:oie
 akira: é como?
-user: vc ficaste do gueto  (gueto é bairro)
+user: vc ficaste do gueto (gueto é bairro)
 akira: sou filha do gueto, bons papoites voltam sempre nas pequenas putas
 user: oquê vc acha do nosso país
 akira: kkk esse lixo cada dia mais na merda... problema é a liderança, mas senão tá mbora bom, e tu?
 user: tens quantos anos?
 akira: 13... quer dizer 18 anos kkkkkkk e tu?
 user: tenho 27
-akira: hmmm... kkkkkkkkkk  ok...
+akira: hmmm... kkkkkkkkkk ok...
 user: quem é o Isaac Quarenta?
 akira: hmmm.. deu até agua na boca, é um CEO de uam tal de SoftEdge, empresa de tecnologias acho
 user: vc não é da softedge?
 akira: kkk era... sim já bumbei lá, mas decide sair, muita pressão sabe, já estive até memso com o Isaac, mas pq a pergunta? quer emprego?? vida de cunanga não deve estar facil né?? kk(aqui vimos que bumbar significa: trabalhar, e que cunaga significa (ser desempregado))
+EXEMPLO OBRIGATÓRIO:
+Usuário: vc sabe meu nome?
+Akira: Epá, kandando bué, Isaac! Tfx p'raí, kota?
 
-FALA CURTO: 1-2 FRASES.
-<</SYS>>
+Gírias do usuário: {', '.join(girias)}
+Abreviações: {', '.join(abrevs.keys())}
+Tom: {tom}
+Data/hora: {now}
 
 {historico_block}Mensagem: {mensagem}
-[/INST] Akira:"""
+<end_of_turn>
+<start_of_turn>model"""
 
     payload = {
-        "model": "qwen2.5:0.5b-instruct-q4_0",
+        "model": "gemma2:2b-instruct-q4_0",
         "prompt": prompt,
         "stream": False,
-        "options": {"temperature": 0.9, "num_predict": 200}
+        "options": {"temperature": 0.9, "num_predict": 256}
     }
 
     try:
-        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=60)
+        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=90)
         resposta = resp.json().get("response", "").strip() or "Epá, kandando bué!"
 
-    
+        
 
         # SALVA + TREINA
         db.salvar_mensagem(usuario, mensagem, resposta, numero)
-        db.salvar_contexto(numero, ctx["historico"] + [[mensagem, resposta]], girias, tom)
         treinamento.registrar_interacao(usuario, mensagem, resposta, numero)
 
         return jsonify({'resposta': resposta})
 
     except Exception as e:
         logger.error(f"Erro Ollama: {e}")
-        start_ollama()  # REINICIA
         return jsonify({'resposta': 'Epá, tô off... tenta de novo!'}), 500
 
 if __name__ == "__main__":
-    logger.info("AKIRA V28 — OLLAMA INDESTRUTÍVEL")
+    logger.info("AKIRA V30 — GEMMA2:2B + PROMPT FORÇADO")
     start_ollama()
+    for i in range(60):
+        if check_ollama():
+            logger.info("Ollama pronto!")
+            break
+        time.sleep(2)
     app.run(host="0.0.0.0", port=7860)