Update app.py

app.py

@@ -15,24 +15,35 @@ from Gradio_UI import GradioUI
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True,hf_token=os.getenv("HF_TOKEN"))
 
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# Image generation !
 
-@tool
-def generate_image_tool(prompt: str) -> Image:
-    """Generates an image from text using the image generation tool.
-    
-    Args:
-        prompt: A description of the image to generate.
-    
-    Returns:
-        The generated image.
-    """
+def generate_image_tool(prompt: str) -> str:
+    """Generates an image from text and returns the file path."""
     try:
-        return image_generation_tool(prompt)
+        image = image_generation_tool(prompt)  # Get PIL Image
+        file_path = "generated_image.jpg"
+        image.save(file_path)  # Save it
+        return file_path  # Return the file path
     except Exception as e:
         return f"Error generating image: {str(e)}"
 
 
+# @tool
+# def generate_image_tool(prompt: str) -> Image:
+#     """Generates an image from text using the image generation tool.
+    
+#     Args:
+#         prompt: A description of the image to generate.
+    
+#     Returns:
+#         The generated image.
+#     """
+#     try:
+#         return image_generation_tool(prompt)
+#     except Exception as e:
+#         return f"Error generating image: {str(e)}"
+
+
     
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !