Add logging

app.py

@@ -1,31 +1,39 @@
-#!/usr/bin/env python3
-import os
-import json
-import uuid
-import uvicorn
-import pika
-from fastapi import FastAPI, Body, Header, HTTPException
-from fastapi.middleware.cors import CORSMiddleware
-from worker import main as rabbit_worker
-
-app = FastAPI()
-
-API_KEY = os.getenv("SECRET_KEY")
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-@app.get("/")
-async def root():
-    return {"status": "ok", "message": "API is running"}
-
-
-if __name__ == "__main__":
-    os.system('python download_models_hf.py')
-    rabbit_worker()
+#!/usr/bin/env python3
+import os
+import json
+import uuid
+import uvicorn
+import pika
+import logging
+from fastapi import FastAPI, Body, Header, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from worker import main as rabbit_worker
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(name)s - %(message)s"
+)
+logger = logging.getLogger(__name__)
+
+app = FastAPI()
+
+API_KEY = os.getenv("SECRET_KEY")
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@app.get("/")
+async def root():
+    logger.info("Root endpoint called.")
+    return {"status": "ok", "message": "API is running"}
+
+
+if __name__ == "__main__":
+    os.system('python download_models_hf.py')
+    rabbit_worker()
     uvicorn.run(app, host="0.0.0.0", port=8000)