Update app.py

implementar error para determinar por que no se genera la imagen

app.py

@@ -23,8 +23,11 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     elif arg2 == 2:
         # Generate image using image generation tool
         prompt = f"Generate an image based on the following input: {arg1}"
-        image_response = image_generation_tool(prompt)
-        return f"Image generated: {image_response['image_url']}"
+        try:
+            image_response = image_generation_tool(prompt)
+            return f"Image generated: {image_response['image_url']}"
+        except Exception as e:
+            return f"Failed to generate image. Error: {str(e)}"
     else:
         return "Invalid argument for arg2. Use 1 for text generation and 2 for image generation."