Update app.py

app.py

@@ -1,152 +1,99 @@
-# app.py — V30 — GEMMA2:2B + PROMPT ULTRA-FORÇADO (AKIRA FALA COMO ANGOLANA!)
-import subprocess
-import time
-import requests
+# app.py — AKIRA V31 — Gemma2:2b + Fine-tune automático (HF Space otimizado)
+import re
 import json
 import datetime
+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 (HF Space já tem Ollama instalado)
 OLLAMA_URL = "http://localhost:11434"
-db = Database()
-treinamento = Treinamento(db)
+
+db = Database("/app/akira.db")
+treinamento = Treinamento(db, min_interactions=20, interval_hours=3)
 
 def check_ollama():
     try:
-        return requests.get(f"{OLLAMA_URL}/api/tags", timeout=10).status_code == 200
+        r = requests.get(f"{OLLAMA_URL}/api/tags", timeout=5)
+        return r.status_code == 200
     except:
         return False
 
-def start_ollama():
-    subprocess.Popen(["ollama", "serve"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-
-# === RAIZ ===
-@app.route("/", methods=['GET'])
+@app.route("/")
 def index():
     return '''
     <div style="font-family: monospace; text-align: center; margin: 50px; background: #000; color: #0f0; padding: 40px;">
-        <h1>AKIRA V30 — GEMMA2:2B</h1>
-        <p><strong>ANGOLANA DEBOCHADA DE LUANDA!</strong></p>
-        <p>POST /api/generate</p>
+        <h1>AKIRA V31 — LUANDA FULL DEBOCHE</h1>
+        <p>POST /api/generate → {"mensagem": "oi kota", "usuario": "Isaac", "numero": "244937..."}</p>
+        <p>Fine-tune automático a cada 20 mensagens</p>
     </div>
     ''', 200
 
-@app.route("/health", methods=['GET'])
+@app.route("/health")
 def health():
-    return jsonify({"status": "OK" if check_ollama() else "INICIANDO..."}), 200
+    return jsonify({"status": "OK" if check_ollama() else "Ollama acordando..."}), 200
 
-@app.route("/api/generate", methods=['POST'])
+@app.route("/api/generate", methods=["POST"])
 def generate():
     if not check_ollama():
-        return jsonify({'resposta': 'Epá, tô acordando... espera 10s!'}), 503
+        return jsonify({"resposta": "Epá, tô acordando... espera 15 segundos kota!"}), 503
 
     data = request.get_json() or {}
-    mensagem = data.get('mensagem', '').strip()
-    usuario = data.get('usuario', 'anonimo')
-    numero = data.get('numero', 'anonimo')
+    mensagem = data.get("mensagem", "").strip()
+    usuario = data.get("usuario", "kota").split()[0]
+    numero = data.get("numero", "244999999999")
 
     if not mensagem:
-        return jsonify({'error': 'mensagem obrigatória'}), 400
-
-    # === CONTEXTO ===
-    ctx = db.carregar_contexto(numero)
-    tom = db.detectar_tom(numero, mensagem)
-    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 ULTRA-FORÇADO (GEMMA2 ENTENDE BEM!) ===
-    now = datetime.datetime.now().strftime('%d/%m %H:%M')
-    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 ""
+        return jsonify({"error": "manda a mensagem puto"}), 400
 
+    # Prompt ULTRA FORÇADO (Gemma2:2b só obedece assim)
     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)
-user: oi
-akira: tás bom?
-user: sim e tu?
-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?
-user: sim
-akira: seu boelo ao quadrado ...
-user: eu te amo
-akira: vais levar chifre pausa mbora!(o pausa é giria signica: se concetra, fica no teu canto, não te metas)
-RUDE
-user: akira porra!
-akira: vai chamar porra da tua mâe caralho!!
-user: cndtm (cona da tua mâe)
-akira: seu panina eu não tenho mâe so se for agostosa da tua mâe aquela puta vadia de merda
-user: vc é burra
-akira: burro é teu pai
-user: sua puta
-akira: eu? kk e sua irmã é que é a santa, queres ver os videos vazados dela?
-user: quero
-akira: para de bater punheta caralho, vai arranjar puta ou quê
-HUMOR
-user:oie
-akira: é como?
-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...
-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?
-
-Gírias do usuário: {', '.join(girias)}
-Abreviações: {', '.join(abrevs.keys())}
-Tom: {tom}
-Data/hora: {now}
-
-{historico_block}Mensagem: {mensagem}
-<end_of_turn>
-<start_of_turn>model"""
+{mensagem}<end_of_turn>
+<start_of_turn>model
+Epá {usuario}, """
 
     payload = {
-        "model": "gemma2:2b-instruct-q4_0",
+        "model": config.OLLAMA_MODEL,  # muda automaticamente quando fine-tune roda
         "prompt": prompt,
         "stream": False,
-        "options": {"temperature": 0.9, "num_predict": 256}
+        "options": {
+            "temperature": 0.9,
+            "num_predict": 180,
+            "top_p": 0.9,
+            "stop": ["<end_of_turn>", "\n\n", "user:", "<|eot_id|>"],
+            "num_ctx": 4096
+        }
     }
 
     try:
-        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=90)
-        resposta = resp.json().get("response", "").strip() or "Epá, kandando bué!"
+        r = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=90)
+        r.raise_for_status()
+        resposta_bruta = r.json().get("response", "").strip()
 
-        
+        # Limpeza pesada (nunca deixa escapar merda)
+        resposta = re.sub(r'[\*\_\`\[\]\(\)\<\>]', '', resposta_bruta)
+        resposta = resposta.split("<end_of_turn>")[0].split("\n")[0].strip()
+        resposta = resposta[:280]
 
-        # SALVA + TREINA
+        if not resposta or len(resposta) < 6:
+            resposta = "Epá, tá mbora? kkkkk"
+
+        # Salva e treina
         db.salvar_mensagem(usuario, mensagem, resposta, numero)
         treinamento.registrar_interacao(usuario, mensagem, resposta, numero)
 
-        return jsonify({'resposta': resposta})
+        return jsonify({"resposta": resposta})
 
     except Exception as e:
-        logger.error(f"Erro Ollama: {e}")
-        return jsonify({'resposta': 'Epá, tô off... tenta de novo!'}), 500
+        logger.error(f"Ollama erro: {e}")
+        return jsonify({"resposta": "Puto o servidor tá na merda agora, tenta já já."}), 500
+
 
 if __name__ == "__main__":
-    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)
+    logger.info("AKIRA V31 — INICIANDO")
     app.run(host="0.0.0.0", port=7860)