Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py — V22 — PROMPT DINÂMICO COM GÍRIAS + ABREVIÇÕES DO BANCO
+# app.py — V23 — QWEN2.5:0.5B + PROMPT DINÂMICO COM GÍRIAS + ABREVIÇÕES DO BANCO
 import subprocess
 import time
 import requests
@@ -12,9 +12,14 @@ app = Flask(__name__)
 OLLAMA_URL = "http://localhost:11434"
 db = Database()
 
+# MODELO LEVE, BOM EM PORTUGUÊS, FINE-TUNEÁVEL
+MODEL_NAME = "qwen2.5:0.5b-instruct-q4_0"
+
 def check_ollama():
-    try: return requests.get(f"{OLLAMA_URL}/api/tags", timeout=10).status_code == 200
-    except: return False
+    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)
@@ -63,42 +68,49 @@ Fala com gírias: {', '.join(girias)}
 Abreviações: {', '.join(abrevs.keys())}
 Tom: {tom}
 Data/hora: {now}
-
 ### Usuário: {usuario} ({numero}) ###
-
 {hist_text and f"### Histórico ###\n{hist_text}\n\n"}
-
 ### Mensagem ###
 {mensagem}
-
 Akira:"""
 
-    # === GERA ===
+    # === GERA COM QWEN2.5:0.5B ===
     payload = {
-        "model": "phi3.5:mini-instruct-q4_0",
+        "model": MODEL_NAME,
         "prompt": prompt,
         "stream": False,
-        "options": {"temperature": 0.9, "num_predict": 256}
+        "options": {
+            "temperature": 0.9,
+            "num_predict": 256,
+            "top_k": 40,
+            "top_p": 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é!"
 
-        # SALVA
+        # SALVA NO BANCO
         db.salvar_mensagem(usuario, mensagem, resposta, numero)
         novo_hist = ctx["historico"] + [[mensagem, resposta]]
         db.salvar_contexto(numero, novo_hist, girias, tom)
 
         return jsonify({'resposta': resposta})
+
     except Exception as e:
-        logger.error(f"Erro: {e}")
+        logger.error(f"Erro Ollama: {e}")
         return jsonify({'resposta': 'Epá, tô off... tenta de novo!'}), 500
 
 if __name__ == "__main__":
-    logger.info("AKIRA V22 — PROMPT DINÂMICO COM BANCO")
+    logger.info(f"AKIRA V23 — RODANDO COM {MODEL_NAME}")
     start_ollama()
     for i in range(60):
-        if check_ollama(): break
+        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)