Update fluxai.py

fluxai.py

@@ -11,7 +11,7 @@ from pymongo import MongoClient
 from models import *
 from huggingface_hub import InferenceClient
 from fastapi import UploadFile, File
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, FileResponse
 import uuid
 from RyuzakiLib import GeminiLatest
 
@@ -75,6 +75,16 @@ def deduct_tokens_gpt(user_id, amount):
     else:
         return False
 
+UPLOAD_DIRECTORY = "uploads"
+
+@app.get("/uploads/{filename}")
+async def get_upload(filename: str):
+    file_path = os.path.join(UPLOAD_DIRECTORY, filename)
+    if os.path.exists(file_path):
+        return FileResponse(file_path)
+    else:
+        raise HTTPException(status_code=404, detail="File not found")
+
 @router.post("/akeno/mistralai", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 async def mistralai_(payload: MistralAI):
         try: