Update fluxai.py

fluxai.py

@@ -10,7 +10,7 @@ from pydantic import BaseModel
 from pymongo import MongoClient
 from models import *
 from huggingface_hub import InferenceClient
-from fastapi import UploadFile
+from fastapi import UploadFile, File
 from fastapi.responses import JSONResponse
 import uuid
 from RyuzakiLib import GeminiLatest
@@ -90,7 +90,10 @@ async def mistralai_(payload: MistralAI):
             )
 
 @router.post("/akeno/fluxai", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
-async def fluxai_image(payload: FluxAI, file: UploadFile):
+async def fluxai_image(
+    payload: FluxAI = Depends(), 
+    file: UploadFile = File(None)
+):
     if deduct_tokens_gpt(payload.user_id, amount=20):
         try:
             image_bytes = await schellwithflux(payload.args)
@@ -110,7 +113,10 @@ async def fluxai_image(payload: FluxAI, file: UploadFile):
                     enhanced_image_bytes = io.BytesIO()
                     image.save(enhanced_image_bytes, format="JPEG", quality=95)
                     enhanced_image_bytes.seek(0)
-                ext = file.filename.split(".")[-1]
+                if file:
+                    ext = file.filename.split(".")[-1]
+                else:
+                    ext = "jpg"
                 unique_filename = f"{uuid.uuid4().hex}.{ext}"
                 file_path = os.path.join("uploads", unique_filename)
                 os.makedirs(os.path.dirname(file_path), exist_ok=True)