update img tool

app.py

@@ -13,7 +13,6 @@ from Gradio_UI import GradioUI
 @tool
 # it's import to specify the return type
 def generate_image(prompt: str, style: str) -> str:
-    # Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that generates an image based on a prompt and a style.
     Args:
         prompt: the prompt to generate the image
@@ -25,9 +24,13 @@ def generate_image(prompt: str, style: str) -> str:
 
         # Generate the image
         image = image_generation_tool(full_prompt)
-
-        # Return the image
-        return image
+        
+        # Convert the image to a string representation that can be displayed
+        if hasattr(image, 'to_string'):
+            return image.to_string()
+        else:
+            # If we can't convert the image, return a message with the image object
+            return f"Image generated successfully! Please check the image object: {str(image)}"
     except Exception as e:
         return f"Error generating image: {str(e)}"