Update main.py

main.py

@@ -501,13 +501,17 @@ async def bing_dalle(item: BingDalle):
 
     try:
         response = await generate_image(item.prompt, item.model)
-        # Check if response contains the expected data
-        if 'data' in response and response['data']:
-            # Assuming the URL is stored in 'url' key of the first item
-            image_url = response['data'][0]['url']
+    except BaseException as e:
+        return {"status": "false", "message": f"Something went wrong: {e}"}
+    try:
+        image_url = response.data[0].url
+    except BaseException as e:
+        return {"status": "false", "message": f"No image URL found in the {response} Error: {e}"
+
+        if image_url is not None:
             return {"status": "true", "sukuna": {"message": image_url}}
         else:
-            return {"status": "false", "message": f"No image URL found in the {response.data}"}
+            return {"status": "false", "message": f"No image URL found in the {response}"}
     except BaseException as e:
         return {"status": "false", "message": f"Something went wrong: {e}"}