Refactor Dockerfile and enhance main.py with Hugging Face integration

Dockerfile

@@ -19,12 +19,5 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy the application code
 COPY . .
 
-# Create a non-root user
-RUN useradd -m -u 1000 appuser
-USER appuser
-
-# Expose the port the app runs on
-EXPOSE 8000
-
 # Command to run the application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

main.py

@@ -1,28 +1,53 @@
 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
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 app = FastAPI()
 app.include_router(poetry_router, prefix="/api/v1/poetry")
 
-if __name__ == "__main__":
-    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
-
-    logging.basicConfig(level=logging.INFO)
-    logger = logging.getLogger(__name__)
-
-    def get_app_and_port() -> Tuple[Starlette, int]:
-        port = int(os.getenv("PORT", "8000"))
-        return app, port
-
-    async def lifecheck(request):
-        return Response("OK", media_type="text/plain")
+@lru_cache()
+def get_hf_token() -> str:
+    """Get Hugging Face token from environment variables."""
+    token = os.getenv("HF_TOKEN")
+    if not token:
+        raise EnvironmentError(
+            "HF_TOKEN environment variable not found. "
+            "Please set your Hugging Face access token."
+        )
+    return token
+
+def init_huggingface():
+    """Initialize Hugging Face authentication."""
+    try:
+        token = get_hf_token()
+        login(token=token)
+        logger.info("Successfully logged in to Hugging Face")
+    except Exception as e:
+        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
+
+async def lifecheck(request):
+    return Response("OK", media_type="text/plain")
 
+if __name__ == "__main__":
+    # Initialize Hugging Face authentication before starting the server
+    init_huggingface()
+    
     routes = [
         Route("/", app.router),
         Route("/healthz", lifecheck),

requirements.txt

@@ -1,9 +1,8 @@
 # Core dependencies
-gradio==4.19.2
 fastapi==0.109.2
 uvicorn==0.27.1
 pydantic==2.6.1
-
+huggingface_hub
 # For API requests and handling
 requests==2.31.0
 python-multipart==0.0.9