Update main.py

main.py

@@ -10,19 +10,22 @@ from loguru import logger
 from huggingface_hub import login
 
 # ==========================
-# 1. CONFIGURAÇÃO DE DIRETÓRIOS PERSISTENTES
+# 1. CONFIGURAÇÃO DE DIRETÓRIOS PERSISTENTES (CORRIGIDO PARA /home/user)
 # ==========================
-DATA_DIR = "/app/data"
-MODELS_DIR = "/app/models/hermes-7b"
-FINETUNED_DIR = "/app/data/finetuned_hermes"
+DATA_DIR = "/home/user/data"
+MODELS_DIR = "/home/user/models/hermes-7b"
+FINETUNED_DIR = "/home/user/data/finetuned_hermes"
 
+# Garante que as pastas existem (agora com permissão correta)
 os.makedirs(DATA_DIR, exist_ok=True)
+os.makedirs(MODELS_DIR, exist_ok=True)
 os.makedirs(FINETUNED_DIR, exist_ok=True)
 
-# Config global
+# Config global (ajusta os caminhos no teu módulo config)
 import modules.config as config
 config.DB_PATH = os.path.join(DATA_DIR, "akira.db")
 config.FINETUNED_PATH = FINETUNED_DIR
+config.MODELS_DIR = MODELS_DIR  # se precisar usar em outro lugar
 
 # ==========================
 # 2. LOGGING (loguru com fallback)
@@ -43,7 +46,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 if HF_TOKEN:
     try:
         login(token=HF_TOKEN)
-        logger.info("HF LOGIN OK COM TOKEN DO SECRET!")
+        logger.info("HF LOGIN OK COM TOKEN DO SECRET! 🔑")
     except Exception as e:
         logger.warning(f"HF LOGIN FALHOU: {e}")
 else:
@@ -61,90 +64,24 @@ def ensure_hermes_model():
     missing = [f for f in required_files if not os.path.exists(f)]
     if missing:
         logger.error(f"MODELO INCOMPLETO: {missing}")
-        logger.error("→ RECONSTRUIR O SPACE COM DOCKERFILE ATUALIZADO!")
-        logger.error("→ USE snapshot_download COM .safetensors FORÇADO!")
-        raise FileNotFoundError(f"Hermes 7B incompleto em {MODELS_DIR}")
-    
-    shards = [f for f in os.listdir(MODELS_DIR) if f.endswith(".safetensors")]
-    if len(shards) != 4:
-        logger.error(f"APENAS {len(shards)} SHARDS .safetensors! ESPERADO: 4")
-        raise FileNotFoundError("Shards faltando!")
-    
-    logger.info(f"Hermes 7B JÁ ESTÁ NO BUILD: {MODELS_DIR} ({len(shards)} shards)")
+        logger.error("Baixe o modelo no Dockerfile com snapshot_download ou coloque manualmente em /home/user/models/hermes-7b")
+        return False
+    logger.info("Hermes-7B local encontrado! Pronto para uso.")
+    return True
 
 # ==========================
-# 5. IMPORTA API E TREINAMENTO (ORDEM CRÍTICA!)
+# 5. IMPORTA E INICIA A API (só depois de tudo pronto)
 # ==========================
-from modules.api import AkiraAPI
-from modules.treinamento import Treinamento
-from modules.database import Database
+if ensure_hermes_model():
+    logger.info("Tudo pronto! Iniciando AkiraAPI com sotaque angolano...")
 
-# ==========================
-# 6. CRIAÇÃO DA APLICAÇÃO
-# ==========================
-def create_app():
-    """Inicializa a aplicação Akira IA com ordem segura"""
-    logger.info("Iniciando Akira IA com Hermes 7B (CPU 8-bit)...")
-    logger.info(f"Banco de dados: {config.DB_PATH}")
-    logger.info(f"Modelo: {MODELS_DIR}")
-    logger.info(f"Fine-tune: {FINETUNED_DIR}")
-
-    # 1. Garante modelo
-    ensure_hermes_model()
-
-    # 2. Inicia API → carrega local_llm → singleton global
-    try:
-        akira = AkiraAPI(config)
-        app = akira.app
-        logger.info("AkiraAPI carregada com sucesso!")
-    except Exception as e:
-        logger.error(f"Erro ao iniciar AkiraAPI: {e}")
-        import traceback
-        logger.error(traceback.format_exc())
-        raise
-
-    # 3. Só depois inicia treinamento (usa mesmo singleton!)
-    if getattr(config, 'START_PERIODIC_TRAINER', False):
-        try:
-            db = Database(config.DB_PATH)
-            trainer = Treinamento(db, interval_hours=getattr(config, 'TRAINING_INTERVAL_HOURS', 4))
-            trainer.start_periodic_training()
-            logger.info("Treinamento periódico iniciado (usa mesmo Hermes global!)")
-        except Exception as e:
-            logger.error(f"Erro ao iniciar treinamento: {e}")
-            import traceback
-            logger.error(traceback.format_exc())
-
-    # 4. Endpoints
-    logger.info("Endpoints:")
-    logger.info(" POST /api/akira → Chat com Akira")
-    logger.info(" GET /health     → Health check")
-    logger.info(" GET /           → Status")
+# Importa a app Flask/FastAPI/Gradio (ajusta conforme teu arquivo real)
+from app import app  # ou from api.main import app / from gradio_app import demo, etc.
 
-    return app
+# Se estiver usando Gradio:
+# demo.launch(server_name="0.0.0.0", server_port=7860)
 
-# ==========================
-# 7. INSTÂNCIA DA APP
-# ==========================
-app = create_app()
-
-# ==========================
-# 8. ROTAS DE SAÚDE
-# ==========================
-@app.route("/")
-def index():
-    """Página inicial de status"""
-    return "Akira IA online! Bué fixe, kota!", 200
-
-@app.route("/health")
-def health():
-    """Health check para o container"""
-    return "OK", 200
+# Se estiver usando Flask/FastAPI com Gunicorn → não precisa fazer nada, o CMD do Dockerfile já roda:
+# gunicorn main:app
 
-# ==========================
-# 9. EXECUÇÃO LOCAL (modo dev)
-# ==========================
-if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))
-    logger.info(f"Servidor rodando localmente em http://0.0.0.0:{port}")
-    app.run(host="0.0.0.0", port=port, debug=False, use_reloader=False)
+logger.info("AKIRA ONLINE E PRONTO PARA FALAR 'BUÉ FIXE, KOTA!' 🇦🇴")