Update main.py

main.py

@@ -481,18 +481,19 @@ async def bing_dalle(item: BingDalle):
     except requests.exceptions.RequestException:
         raise HTTPException(status_code=500, detail="Invalid cookie string, check your cookie string and try again")
 
-    async def generate_image():
+    async def generate_image(prompt, model):
         try:
             response = BingImages.images.generate(
-                prompt=item.prompt,
-                model=item.model,
+                prompt=prompt,
+                model=model,
             )
         except BaseException as e:
             return {"status": "false", "message": f"Internal Error: {e}"}
         return response
     try:
-        response = await generate_image()
-        return {"status": "true", "sukuna": {"message": response.data[0].url}}
+        response = await generate_image(item.prompt, item.model)
+        image_url = response.data[0].url
+        return {"status": "true", "sukuna": {"message": image_url}}
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}