fixes

main.py

@@ -1,26 +1,19 @@
+from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from app.api.endpoints.poetry import router as poetry_router
 import os
 import logging
 from typing import Tuple
-from starlette.applications import Starlette
 from starlette.responses import Response
-from starlette.routing import Route
 from starlette.staticfiles import StaticFiles
 from huggingface_hub import login
 from functools import lru_cache
 from app.services.poetry_generation import PoetryGenerationService
 
-# Initialize the PoetryGenerationService during startup
-poetry_service = PoetryGenerationService()
-
-
+# Configure logging once at module level
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-app = FastAPI()
-app.include_router(poetry_router, prefix="/api/v1/poetry")
-
 @lru_cache()
 def get_hf_token() -> str:
     """Get Hugging Face token from environment variables."""
@@ -42,29 +35,31 @@ def init_huggingface():
         logger.error(f"Failed to login to Hugging Face: {str(e)}")
         raise
 
-def get_app_and_port() -> Tuple[Starlette, int]:
-    port = int(os.getenv("PORT", "8000"))
-    return app, port
-@app.on_event("startup")
-async def startup_event():
-    # This will be executed when the application starts
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
+    poetry_service = PoetryGenerationService()
     await poetry_service.preload_models()
-async def lifecheck(request):
+    init_huggingface()
+    yield
+    # Cleanup (if needed)
+    # Add cleanup code here
+
+app = FastAPI(lifespan=lifespan)
+app.include_router(poetry_router, prefix="/api/v1/poetry")
+
+@app.get("/healthz")
+async def lifecheck():
     return Response("OK", media_type="text/plain")
 
+def get_port() -> int:
+    return int(os.getenv("PORT", "8000"))
+
 if __name__ == "__main__":
-    # Initialize Hugging Face authentication before starting the server
-    init_huggingface()
+    import uvicorn
     
-    routes = [
-        Route("/", app.router),
-        Route("/healthz", lifecheck),
-    ]
-
-    app_and_port = get_app_and_port()
-    app = app_and_port[0]
-    port = app_and_port[1]
-
-    logger.info(f"Starting FastAPI server on port {port}")
+    port = get_port()
     app.mount("/static", StaticFiles(directory="static"), name="static")
-    app.run(host="0.0.0.0", port=port)
+    
+    logger.info(f"Starting FastAPI server on port {port}")
+    uvicorn.run(app, host="0.0.0.0", port=port)