Update treinamento.py

treinamento.py

@@ -4,7 +4,6 @@ import os
 import threading
 import time
 import requests
-from log pérd
 from loguru import logger
 from database import Database
 from sentence_transformers import SentenceTransformer
@@ -68,7 +67,7 @@ class Treinamento:
             self.thread.start()
             logger.info("Loop de fine-tune iniciado!")
 
-    def registrar_interacao(self, usuario, mensagem, resposta, numero):
+    def registrar_interacao(self, usuario, mensagem, resposta, numero, is_reply=False, mensagem_original=""):
         try:
             # === SALVA NO BANCO ===
             self.db.salvar_mensagem(usuario, mensagem, resposta, numero)
@@ -86,7 +85,7 @@ class Treinamento:
                 json.dump(entry, f, ensure_ascii=False)
                 f.write("\n")
 
-            logger.info(f"Kandando salvo: {usuario[:10]}... ({len(_dataset)} total)")
+            logger.info(f"Kandando salvo: {len(_dataset)} total")
 
             # === TREINA SE CHEGAR A 25 ===
             if len(_dataset) >= self.min_interactions:
@@ -98,7 +97,6 @@ class Treinamento:
     def _treinar(self):
         if len(_dataset) < self.min_interactions:
             return
-
         logger.info(f"INICIANDO FINE-TUNE → {MODEL_FINE} com {len(_dataset)} kandandos")
 
         try:
@@ -109,7 +107,6 @@ class Treinamento:
 
             files = {'modelfile': open(MODelfile_PATH, 'rb')}
             data = {'name': MODEL_FINE}
-
             resp = requests.post("http://localhost:11434/api/create", files=files, data=data, timeout=600)
 
             if resp.status_code == 200:
@@ -119,7 +116,6 @@ class Treinamento:
                 logger.error(f"Erro Ollama: {resp.status_code} {resp.text}")
 
             os.remove(MODelfile_PATH)
-
         except Exception as e:
             logger.error(f"Erro no fine-tune: {e}")