Update main.py

main.py

@@ -481,13 +481,18 @@ 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():
+        try:
+            response = BingImages.images.generate(
+                prompt=item.prompt,
+                model=item.model,
+            )
+        except BaseException as e
+            return {"status": "false", "message": f"Internal Error: {e}"}
+        return response
     try:
-        response = BingImages.images.generate(
-            prompt=item.prompt,
-            model=item.model,
-        )
-        return {"status": "true", "sukuna": {"message": response.data[0].url}}
-
+        response = await generate_image()
+        return = {"status": "true", "sukuna": {"message": response.data[0].url}}
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}