Enhance lifespan management in FastAPI by initializing PoetryGenerationService and handling model preloading asynchronously

main.py

@@ -1,3 +1,4 @@
+import asyncio
 from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from app.api.endpoints.poetry import router as poetry_router
@@ -37,13 +38,21 @@ def init_huggingface():
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    # Startup
-    poetry_service = PoetryGenerationService()
-    await poetry_service.preload_models()
+    # Initialize Hugging Face authentication first
     init_huggingface()
-    yield
-    # Cleanup (if needed)
-    # Add cleanup code here
+    
+    # Initialize poetry service and preload models
+    poetry_service = PoetryGenerationService()
+    
+    # Only await if `preload_models` is asynchronous
+    if callable(getattr(poetry_service, "preload_models", None)):
+        result = poetry_service.preload_models()
+        if asyncio.iscoroutine(result):
+            await result
+        else:
+            result()  # Call directly if synchronous
+
+    yield  # Continue to application startup
 
 app = FastAPI(lifespan=lifespan)
 app.include_router(poetry_router, prefix="/api/v1/poetry")