Update app.py

app.py

@@ -97,6 +97,11 @@ class SuperheroPartyThemeTool(Tool):
         
         return themes.get(category.lower(), "Themed party idea not found. Try 'classic heroes', 'villain masquerade', or 'futuristic Gotham'.")
 
+image_generation_tool = Tool.from_space(
+    "black-forest-labs/FLUX.1-schnell",
+    name="image_generator",
+    description="Generate an image from a prompt"
+)
 
 
 final_answer = FinalAnswerTool()
@@ -126,7 +131,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, my_custom_tool, get_current_time_in_timezone, web_search, visit_webpage, python_tool,
+    tools=[final_answer, image_generation_tool, my_custom_tool, get_current_time_in_timezone, web_search, visit_webpage, python_tool, image_generation_tool,
           suggest_menu, catering_service_tool, SuperheroPartyThemeTool()], ## add your tools here (don't remove final answer)
     max_steps=10,
     verbosity_level=1,