Update api/main.py

api/main.py

@@ -1,39 +1,26 @@
 """
-Aplicação FastAPI Principal - para.AI v3.0
-Endpoint para processar JSONL e gerar TAR.GZ
+FastAPI Application Principal
 """
-import logging
-from fastapi import FastAPI, Request
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
-from contextlib import asynccontextmanager
-from datetime import datetime
-import time
-
 from api.config import settings
-from api.routes import health, test, process, download
-from api.utils.logger import setup_logger
-
-# Configurar logging
-logger = setup_logger()
+from api.routes import health_routes, debug_routes, status_routes
+from api.routes import test_routes, process_routes
+import logging
 
-@asynccontextmanager
-async def lifespan(app: FastAPI):
-    """Lifespan events"""
-    logger.info(f"🚀 Iniciando {settings.APP_NAME} v{settings.APP_VERSION}")
-    logger.info(f"📊 Modo: {'DEBUG' if settings.DEBUG else 'PRODUCTION'}")
-    logger.info(f"🔧 Workers: {settings.MAX_WORKERS}")
-    yield
-    logger.info("🛑 Encerrando API")
+# Setup logging
+logging.basicConfig(
+    level=settings.LOG_LEVEL,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger("para_ai")
 
-# Criar aplicação
+# Criar app
 app = FastAPI(
     title=settings.APP_NAME,
     version=settings.APP_VERSION,
-    description="Sistema de processamento de jurisprudências com 9 especialistas LLM",
-    lifespan=lifespan,
-    docs_url="/docs" if settings.DEBUG else None,
-    redoc_url="/redoc" if settings.DEBUG else None,
+    description="Sistema para.AI - 9 Especialistas LLM para análise jurisprudencial",
+    docs_url="/docs" if settings.DEBUG else None
 )
 
 # CORS
@@ -45,57 +32,27 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Middleware de tempo de processamento
-@app.middleware("http")
-async def add_process_time(request: Request, call_next):
-    start = time.time()
-    response = await call_next(request)
-    process_time = time.time() - start
-    response.headers["X-Process-Time"] = f"{process_time:.4f}"
-    return response
+# Rotas
+app.include_router(health_routes.router, prefix="/health", tags=["Health"])
+app.include_router(test_routes.router, prefix="/test", tags=["Tests"])
+app.include_router(process_routes.router, prefix="/process", tags=["Process"])
+app.include_router(status_routes.router, prefix="/status", tags=["Status"])
 
-# Exception handler global
-@app.exception_handler(Exception)
-async def global_exception_handler(request: Request, exc: Exception):
-    logger.error(f"❌ Erro global: {exc}", exc_info=True)
-    return JSONResponse(
-        status_code=500,
-        content={
-            "error": "Internal Server Error",
-            "detail": str(exc) if settings.DEBUG else "Erro no processamento",
-            "timestamp": datetime.now().isoformat(),
-            "path": str(request.url)
-        }
-    )
+if settings.DEBUG:
+    app.include_router(debug_routes.router, prefix="/debug", tags=["Debug"])
 
-# Rotas
-app.include_router(health.router, prefix="/health", tags=["Health"])
-app.include_router(test.router, prefix="/test", tags=["Tests"])
-app.include_router(process.router, prefix="/process", tags=["Process"])
-app.include_router(download.router, prefix="/download", tags=["Download"])
 
 @app.get("/")
 async def root():
-    """Endpoint raiz"""
     return {
         "app": settings.APP_NAME,
         "version": settings.APP_VERSION,
         "status": "running",
-        "timestamp": datetime.now().isoformat(),
-        "docs": "/docs" if settings.DEBUG else "disabled",
         "endpoints": {
+            "docs": "/docs",
             "health": "/health",
             "test_specialists": "/test/specialists",
             "process_jsonl": "/process/jsonl",
             "download": "/download/{batch_id}"
         }
-    }
-
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(
-        "api.app:app",
-        host=settings.API_HOST,
-        port=settings.API_PORT,
-        reload=settings.DEBUG
-    )
+    }