Update fluxai.py

fluxai.py

@@ -77,13 +77,33 @@ def deduct_tokens_gpt(user_id, amount):
 
 UPLOAD_DIRECTORY = "./uploads"
 
+@router.post("/uploadfile/")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        ext = file.filename.split(".")[-1]
+        unique_filename = f"{uuid.uuid4().hex}.{ext}"
+        file_location = os.path.join(UPLOAD_DIRECTORY, unique_filename)
+        with open(file_location, "wb") as f:
+            f.write(await file.read())
+        return JSONResponse(
+            status_code=200,
+            content={"url": f"https://randydev-ryuzaki-api.hf.space/uploads/{unique_filename}"}
+        )
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e)}
+        )
+
 @router.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")
+async def serve_file(filename: str):
+    file_location = os.path.join(UPLOAD_DIRECTORY, filename)
+    if os.path.exists(file_location):
+        return FileResponse(file_location)
+    return JSONResponse(
+        status_code=404,
+        content={"error": "File not found"}
+    )
 
 @router.post("/akeno/mistralai", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 async def mistralai_(payload: MistralAI):
@@ -125,10 +145,13 @@ async def fluxai_image(payload: FluxAI):
                 os.makedirs(os.path.dirname(file_path), exist_ok=True)
                 with open(file_path, "wb") as f:
                     f.write(enhanced_image_bytes.getvalue())
+                url_request = "https://randydev-ryuzaki-api.hf.space/uploadfile/"
+                files = {"file": open(file_path, "rb")}
+                response_uploaded = requests.post(url, files=files)
                 example_test = "Explain how this picture looks like."
                 x = GeminiLatest(api_keys=GOOGLE_API_KEY)
                 response = x.get_response_image(example_test, file_path)
-                url = f"https://randydev-ryuzaki-api.hf.space/{file_path}"
+                url = response_uploaded.json().get("url")
                 return SuccessResponse(
                     status="True",
                     randydev={"url": url, "caption": response}