Update app.py

app.py

@@ -36,19 +36,25 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+image_generation_tool = Tool.from_space(
+    "black-forest-labs/FLUX.1-schnell",
+    name="image_generator",
+    description="Generate an image from a prompt"
+)
+
 @tool
-def random_quote_tool()-> str: 
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that fetches a random quote
-    No arguments are taken
+def display_image(path: str) -> str:
+    """
+    Use the PIL library to display an image from a given path.
+
+    Args:
+        path: output path from the image generation tool
     """
-    response = requests.get("https://zenquotes.io/api/quotes")
-    if response.status_code == 200:
-        quote_data = response.json()
-        return random.choice(quote_data).get('q')
-    else:
-        return "Failed to fetch a quote"
+    image = Image.open(path)
+    image.show()
 
+    return "Success"
+    
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -70,7 +76,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, get_current_time_in_timezone, wiki_of_person], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, get_current_time_in_timezone, wiki_of_person, display_image], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,