Update main.py

main.py

@@ -8,6 +8,7 @@ from datetime import datetime
 
 from fastapi import FastAPI, File, UploadFile, HTTPException, Query, Depends
 from fastapi.responses import FileResponse, StreamingResponse
+from fastapi.staticfiles import StaticFiles  # <-- Imported for serving static files
 
 from llm_initialization import get_llm
 from embedding import get_embeddings
@@ -30,6 +31,13 @@ MONGO_CLUSTER_URL = os.getenv("CONNECTION_STRING")
 
 app = FastAPI(title="VectorStore & Document Management API")
 
+# ----- New: Setup avatars static file serving using absolute path -----
+AVATAR_DIR = os.path.join(os.getcwd(), "avatars")
+if not os.path.exists(AVATAR_DIR):
+    os.makedirs(AVATAR_DIR)
+app.mount("/avatars", StaticFiles(directory=AVATAR_DIR), name="avatars")
+# --------------------------------------------------------------------------
+
 # Import auth router and dependencies
 from auth import router as auth_router, get_current_user, users_collection
 
@@ -304,7 +312,6 @@ async def root():
         "message": "Welcome to the EduLearn AI."
     }
 
-
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)