Update main.py

main.py

@@ -8,7 +8,8 @@ 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
+# Removed static files mounting for avatars as avatars are now served via GridFS in auth
+#from fastapi.staticfiles import StaticFiles
 
 from llm_initialization import get_llm
 from embedding import get_embeddings
@@ -31,12 +32,8 @@ 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")
-# --------------------------------------------------------------------------
+# Note: Since user avatars are now stored in MongoDB via GridFS and served via /auth/avatar,
+# we no longer mount a local avatars directory.
 
 # Import auth router and dependencies
 from auth import router as auth_router, get_current_user, users_collection
@@ -190,7 +187,7 @@ def chat(
 # ----------------------- Remaining Endpoints -----------------------
 @app.post("/add_document")
 async def add_document(
-    file: Optional[UploadFile] = File(None),  # Changed type to UploadFile
+    file: Optional[UploadFile] = File(None),  # File parameter now is an UploadFile
     wiki_query: Optional[str] = Query(None),
     wiki_url: Optional[str] = Query(None)
 ):