Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+from PIL import Image
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -33,6 +33,25 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_image_from_description(description: str) -> str:
+    """A tool that fetches and displays an image based on a description.
+    Args:
+        description: A string describing the image to generate.
+    """
+    image_generation_tool = Tool.from_space(
+        "black-forest-labs/FLUX.1-dev",
+        name="image_generator",
+        description="Generate an image from a prompt")
+    image_obj = image_generation_tool(description)
+    if hasattr(image_obj, "_path"):
+        image_path = image_obj._path
+        image = Image.open(image_path)
+        image.show()
+        return "Image displayed."
+    else:
+        return "Unexpected output: " + str(image_obj)
+
 
 final_answer = FinalAnswerTool()