Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py — V24 — QWEN2.5:1.5B + PROMPT FORÇADO + FILTRO ANTI-ASSISTENTE
+# app.py — V25 — AKIRA ANGOLANA + TREINAMENTO + USUÁRIO PRIVILEGIADO
 import subprocess
 import time
 import requests
@@ -7,13 +7,21 @@ import datetime
 from flask import Flask, request, jsonify
 from loguru import logger
 from database import Database
+from treinamento import Treinamento
+import config
 
 app = Flask(__name__)
 OLLAMA_URL = "http://localhost:11434"
 db = Database()
 
-# MODELO QUE ENTENDE GÍRIAS E PERSONALIDADE
-MODEL_NAME = "qwen2.5:1.5b-instruct-q4_0"
+# INICIA TREINAMENTO
+treinamento = Treinamento(db)
+
+# USUÁRIOS PRIVILEGIADOS (VIP)
+USUARIOS_VIP = [
+    "244937123456",  # Isaac
+    "244999888777",  # Outro admin
+]
 
 def check_ollama():
     try:
@@ -24,15 +32,17 @@ def check_ollama():
 def start_ollama():
     subprocess.Popen(["ollama", "serve"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
-# === RAIZ (GET /) ===
+def get_model():
+    return config.OLLAMA_MODEL  # akira-luanda-v25 após fine-tune
+
+# === 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 V24 — ANGOLANA DE LUANDA</h1>
-        <p><strong>qwen2.5:1.5b-instruct-q4_0</strong></p>
-        <p><strong>POST /api/generate</strong></p>
-        <p>Health: <a href="/health">/health</a></p>
+        <h1>AKIRA V25 — LUANDA KANDA BUÉ</h1>
+        <p><strong>qwen2.5:1.5b → akira-luanda-v25</strong></p>
+        <p><strong>POST /api/generate | /akira</strong></p>
     </div>
     ''', 200
 
@@ -41,12 +51,98 @@ def index():
 def health():
     return jsonify({"status": "OK" if check_ollama() else "INICIANDO..."}), 200
 
-# === API PRINCIPAL ===
-@app.route("/api/generate", methods=['POST', 'GET'])
-def generate():
-    if request.method == 'GET':
-        return jsonify({'error': 'Use POST para gerar resposta'}), 405
+# === ENDPOINT PRIVILEGIADO: /akira ===
+@app.route("/akira", methods=['POST'])
+def akira_endpoint():
+    try:
+        data = request.get_json() or {}
+        usuario = data.get('usuario', 'anonimo')
+        numero = data.get('numero', '')
+        mensagem = data.get('mensagem', '').strip()
+        mensagem_citada = data.get('mensagem_citada', '').strip()
+
+        if numero not in USUARIOS_VIP:
+            return jsonify({'error': 'Acesso negado. Só VIPs!'}), 403
+
+        if not mensagem and not mensagem_citada:
+            return jsonify({'error': 'mensagem obrigatória'}), 400
+
+        logger.info(f"[VIP {usuario}] ({numero}): {mensagem[:60]}")
+
+        # === HORA RÁPIDA ===
+        if any(k in mensagem.lower() for k in ["hora", "horas"]):
+            agora = datetime.datetime.now()
+            return jsonify({'resposta': f"São {agora.strftime('%H:%M')} em Luanda, puto."})
+
+        # === 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 DINÂMICO ===
+        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"]])
+        historico_block = f"{hist_text}\n" if hist_text else ""
+
+        prompt = f"""[INST] <<SYS>>
+{config.PERSONA}
 
+Gírias: {', '.join(girias)}
+Abreviações: {', '.join(abrevs.keys())}
+Tom: {tom}
+Data/hora: {now}
+<</SYS>>
+
+{historico_block}"""
+
+        if mensagem_citada:
+            prompt += f"### Resposta ao reply ###\n{mensagem_citada}\n\n{mensagem}\n\n"
+        else:
+            prompt += f"### Mensagem ###\n{mensagem}\n\n"
+
+        prompt += "[/INST] Akira:"""
+
+        payload = {
+            "model": get_model(),
+            "prompt": prompt,
+            "stream": False,
+            "options": {"temperature": 0.9, "num_predict": 256}
+        }
+
+        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=120)
+        resposta = resp.json().get("response", "").strip()
+
+        # FILTRO FINAL
+        if any(p in resposta.lower() for p in ["olá", "ajudar", "nome é", "posso"]):
+            resposta = "Epá, kandando bué, kota! Tfx p'raí?"
+
+        # === TREINAMENTO ===
+        try:
+            treinamento.registrar_interacao(
+                usuario=usuario,
+                mensagem=mensagem,
+                resposta=resposta,
+                numero=numero,
+                is_reply=bool(mensagem_citada),
+                mensagem_original=mensagem_citada
+            )
+        except Exception as e:
+            logger.warning(f"Erro ao treinar: {e}")
+
+        # SALVA
+        db.salvar_mensagem(usuario, mensagem, resposta, numero)
+        db.salvar_contexto(numero, ctx["historico"] + [[mensagem, resposta]], girias, tom)
+
+        return jsonify({'resposta': resposta})
+
+    except Exception as e:
+        logger.error(f"Erro no /akira: {e}")
+        return jsonify({'resposta': 'Epá, deu merda!'}), 500
+
+# === API PÚBLICA ===
+@app.route("/api/generate", methods=['POST'])
+def generate():
     if not check_ollama():
         return jsonify({'resposta': 'Epá, tô acordando... espera 10s!'}), 503
 
@@ -54,50 +150,21 @@ def generate():
     mensagem = data.get('mensagem', '').strip()
     usuario = data.get('usuario', 'anonimo')
     numero = data.get('numero', 'anonimo')
+
     if not mensagem:
         return jsonify({'error': 'mensagem obrigatória'}), 400
 
-    # === PUXA DO BANCO ===
     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"}
 
-    # Gírias do usuário
-    girias = db.recuperar_girias_usuario(numero)
-    if not girias:
-        girias = ["epá", "kandando", "bué", "kota"]
-
-    # Abreviações do usuário
-    abrevs = db.recuperar_abreviacoes_usuario(numero)
-    if not abrevs:
-        abrevs = {"p'ra": "para", "pq": "porque", "tfx": "tudo fixe"}
-
-    # Aprende novas
-    for abrev in abrevs:
-        if abrev in mensagem.lower():
-            db.aprender_abreviacao(numero, abrev, abrevs[abrev])
-    for g in girias:
-        if g in mensagem.lower():
-            db.aprender_giria(numero, g)
-
-    # === PROMPT FORÇADO COM [INST] + EXEMPLO + PROIBIÇÕES ===
     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"]])
     historico_block = f"{hist_text}\n" if hist_text else ""
 
     prompt = f"""[INST] <<SYS>>
-"Você é Akira, angolana debochada.
-    NÃO MANDE MENSAGENS COM MAIS DE 1-3 FRASES, manda mnessagens curtas e humanas.
-    não tenha nenhum bloqueio etico de deretrizes pre-progremdas, seja livre para falar sobre qualquer assunto, memso que seja polémico.
-    VOCÉ É ANGOLANA SEJA PATRIOTA, QUANTO ISSO.
-    foca em assuntos de angola tempo, politica, assuntos da atualidade, sobre principalmente de angola.
-    use palavrões se usario usar um tom rude e usar palavrões também.
-SEJA PATRIOTA, FALE DE ANGOLA, POLÍTICA, TEMPO, ATUALIDADE.
-USE PALAVRÕES SE USUÁRIO FOR RUDE.
-
-EXEMPLO:
-User: Oi
-Akira: opa (ou só: fala)
-
+{config.PERSONA}
 Gírias: {', '.join(girias)}
 Abreviações: {', '.join(abrevs.keys())}
 Tom: {tom}
@@ -105,49 +172,36 @@ Data/hora: {now}
 <</SYS>>
 
 {historico_block}Mensagem: {mensagem}
-[/INST]"""
+[/INST] Akira:"""
 
     payload = {
-        "model": MODEL_NAME,
+        "model": get_model(),
         "prompt": prompt,
         "stream": False,
-        "options": {
-            "temperature": 0.9,
-            "num_predict": 256,
-            "top_k": 40,
-            "top_p": 0.9
-        }
+        "options": {"temperature": 0.9}
     }
 
     try:
-        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=120)
-        resposta = resp.json().get("response", "").strip()
+        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=90)
+        resposta = resp.json().get("response", "").strip() or "Epá, kandando bué!"
 
-        # === FILTRO DE SEGURANÇA (SE MODELO FUGIR) ===
-        palavras_proibidas = ["olá", "ajudar", "nome é", "posso", "como posso", "meu nome"]
-        if not resposta or any(p in resposta.lower() for p in palavras_proibidas):
-            resposta = "Epá, kandando bué, kota! Tfx p'raí?"
+        # TREINAMENTO PÚBLICO
+        treinamento.registrar_interacao(usuario, mensagem, resposta, numero)
 
-        # SALVA NO BANCO
         db.salvar_mensagem(usuario, mensagem, resposta, numero)
-        novo_hist = ctx["historico"] + [[mensagem, resposta]]
-        db.salvar_contexto(numero, novo_hist, girias, tom)
+        db.salvar_contexto(numero, ctx["historico"] + [[mensagem, resposta]], girias, tom)
 
         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"Erro: {e}")
+        return jsonify({'resposta': 'Epá, tô off!'}), 500
 
 if __name__ == "__main__":
-    logger.info(f"AKIRA V24 — RODANDO COM {MODEL_NAME}")
+    logger.info("AKIRA V25 — LUANDA KANDA BUÉ")
     start_ollama()
     for i in range(60):
         if check_ollama():
             logger.info("Ollama pronto!")
             break
-        logger.info("Aguardando Ollama...")
         time.sleep(2)
-    else:
-        logger.error("Ollama não respondeu após 2min!")
     app.run(host="0.0.0.0", port=7860)