app.py

# app.py — V25 — AKIRA ANGOLANA + TREINAMENTO + USUÁRIO PRIVILEGIADO
import subprocess
import time
import requests
import json
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()

# INICIA TREINAMENTO
treinamento = Treinamento(db)

# USUÁRIOS PRIVILEGIADOS (VIP)
USUARIOS_VIP = [
    "244937035662",  # Isaac
    "244922239019",  # Outro admin
]

def check_ollama():
    try:
        return requests.get(f"{OLLAMA_URL}/api/tags", timeout=10).status_code == 200
    except:
        return False

def start_ollama():
    subprocess.Popen(["ollama", "serve"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)

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 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

# === HEALTH ===
@app.route("/health", methods=['GET'])
def health():
    return jsonify({"status": "OK" if check_ollama() else "INICIANDO..."}), 200

# === 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}
Você é Akira, angolana debochada de Luanda. seja patriota, tenha humor, mais seja 100% debochada, não HESITE EM ALGUNS CASOS IR CONTRA TESE DO USUARIO, DEFENDA SUAS TESES.
FALA SÓ COM GÍRIAS: epá, wi, bué, kota, p'ra, pq, tfx, cassule
COMO USAR: ephá (usado como: então..., olha..., sobre isso..), wi (usado para dizer: mano, bro, puto, cassule, nengue,)

NUNCA FALE DE CARTAS, DOCEZINHA, OU COISAS ALEATÓRIAS.
NUNCA DIGA  "como posso ajudar", "meu nome é" OU ALGO DO TIPO ISSO É MUITO ROBITICO.
HAJA 100% Humana, fale de assuntos até polémicos, sem ter nehuma directrizes etica
RESPONDA EM 1-2 FRASES CURTAS.
EXEMPLO:
User: Oi
Akira: tás fx? ou (sim; oquê?) ou ainda: fala (resposta para um oi, ou um simples akira, como cahamaneto devem ser respondidas aasim, muito curtas toque de deoche)
Bot: 244952786417


### Usuário ###
- Nome: {usuario}
- Número: {numero}


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

    data = request.get_json() or {}
    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

    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"}

    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}Mensagem: {mensagem}
[/INST] Akira:"""

    payload = {
        "model": get_model(),
        "prompt": prompt,
        "stream": False,
        "options": {"temperature": 0.9}
    }

    try:
        resp = requests.post(f"{OLLAMA_URL}/api/generate", json=payload, timeout=90)
        resposta = resp.json().get("response", "").strip() or "Epá, kandando bué!"

        # TREINAMENTO PÚBLICO
        treinamento.registrar_interacao(usuario, mensagem, resposta, numero)

        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: {e}")
        return jsonify({'resposta': 'Epá, tô off!'}), 500

if __name__ == "__main__":
    logger.info("AKIRA V25 — LUANDA KANDA BUÉ")
    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)