Update fluxai.py

fluxai.py

@@ -17,10 +17,11 @@ import uuid
 from RyuzakiLib import GeminiLatest
 
 class FluxAI(BaseModel):
-    user_id: int
+    user_id: int = 1191668125
     api_key: str
     args: str
     auto_enhancer: bool = False
+    is_flux_dev: bool = False
 
 class MistralAI(BaseModel):
     args: str
@@ -46,6 +47,16 @@ async def schellwithflux(args):
         return None
     return response.content
 
+async def devwithflux(args):
+    API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
+    headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
+    payload = {"inputs": args}
+    response = requests.post(API_URL, headers=headers, json=payload)
+    if response.status_code != 200:
+        print(f"Error status {response.status_code}")
+        return None
+    return response.content
+
 async def mistralai_post_message(message_str):
     client = InferenceClient(
         "mistralai/Mixtral-8x7B-Instruct-v0.1",
@@ -121,7 +132,10 @@ async def fluxai_image(payload: FluxAI):
         USERS_API_KEYS = get_all_api_keys()
         if payload.api_key in USERS_API_KEYS:
             try:
-                image_bytes = await schellwithflux(payload.args)
+                if payload.is_flux_dev:
+                    image_bytes = await devwithflux(payload.args)
+                else:
+                    image_bytes = await schellwithflux(payload.args)
                 if image_bytes is None:
                     return SuccessResponse(
                         status="False",