Update api/main.py

api/main.py

@@ -1,106 +1,100 @@
 """
-FastAPI Application Factory
-Main entry point for para.AI API
+Aplicação FastAPI Principal - para.AI v3.0
+Endpoint para processar JSONL e gerar TAR.GZ
 """
+import logging
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from contextlib import asynccontextmanager
-import logging
-import time
 from datetime import datetime
+import time
 
-from api.config import settings
-from api.routes import debug_routes, process_routes, status_routes, health_routes
-
-
-logging.basicConfig(
-    level=settings.LOG_LEVEL,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-)
-logger = logging.getLogger(__name__)
+from main.config import settings
+from main.api.routes import health, test, process, download
+from main.utils.logger import setup_logger
 
+# Configurar logging
+logger = setup_logger()
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Lifespan events"""
-    logger.info(f"🚀 Starting {settings.APP_NAME} v{settings.APP_VERSION}")
-    logger.info(f"📊 Environment: {'DEBUG' if settings.DEBUG else 'PRODUCTION'}")
-    logger.info(f"🤖 LLM Model: {settings.LLM_MODEL}")
-
+    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")
+
+# Criar aplicação
+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,
+)
 
-    logger.info("🛑 Shutting down API")
-
-
-def create_app() -> FastAPI:
-    """Factory function to create FastAPI app"""
-
-    app = FastAPI(
-        title=settings.APP_NAME,
-        version=settings.APP_VERSION,
-        description="API para análise jurisprudencial com LLMs - Sistema de 9 Especialistas",
-        lifespan=lifespan,
-        docs_url="/docs" if settings.DEBUG else None,
-        redoc_url="/redoc" if settings.DEBUG else None,
-    )
-
-    app.add_middleware(
-        CORSMiddleware,
-        allow_origins=settings.CORS_ORIGINS,
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-    @app.middleware("http")
-    async def add_process_time_header(request: Request, call_next):
-        start_time = time.time()
-        response = await call_next(request)
-        process_time = time.time() - start_time
-        response.headers["X-Process-Time"] = str(process_time)
-        return response
-
-    @app.exception_handler(Exception)
-    async def global_exception_handler(request: Request, exc: Exception):
-        logger.error(f"Global exception: {exc}", exc_info=True)
-        return JSONResponse(
-            status_code=500,
-            content={
-                "error": "Internal Server Error",
-                "detail": str(exc) if settings.DEBUG else "An error occurred",
-                "timestamp": datetime.now().isoformat(),
-                "path": str(request.url)
-            }
-        )
-
-    app.include_router(health_routes.router, prefix="/health", tags=["Health"])
-    app.include_router(process_routes.router, prefix="/process", tags=["Process"])
-    app.include_router(status_routes.router, prefix="/status", tags=["Status"])
-
-    if settings.DEBUG:
-        app.include_router(debug_routes.router, prefix="/debug", tags=["Debug"])
+# CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=settings.CORS_ORIGINS,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-    @app.get("/")
-    async def root():
-        return {
-            "app": settings.APP_NAME,
-            "version": settings.APP_VERSION,
-            "status": "running",
+# 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
+
+# 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(),
-            "docs": "/docs" if settings.DEBUG else "disabled"
+            "path": str(request.url)
         }
+    )
 
-    return app
-
-
-app = create_app()
-
+# 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": {
+            "health": "/health",
+            "test_specialists": "/test/specialists",
+            "process_jsonl": "/process/jsonl",
+            "download": "/download/{batch_id}"
+        }
+    }
 
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(
-        "api.main:app",
+        "main.app:app",
         host=settings.API_HOST,
         port=settings.API_PORT,
         reload=settings.DEBUG