Update app.py

ajustar implementation de generacion de imagen

app.py

@@ -23,8 +23,8 @@ 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_url = image_generation_tool(prompt)
-        return f"Image generated: {image_url}"
+        image_response = image_generation_tool(prompt)
+        return f"Image generated: {image_response['image_url']}"
     else:
         return "Invalid argument for arg2. Use 1 for text generation and 2 for image generation."