Update app.py

app.py

@@ -27,12 +27,19 @@ dynamicTool_TextToImage = smolagents.load_tool(
     trust_remote_code=True
 )
 
+spaceTool_ImageGenerator = smolagents.Tool.from_space(
+    "black-forest-labs/FLUX.1-schnell",
+    name="image_generator",
+    description="Generate an image from a prompt"
+)
+
 with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 
 codeAgent_Party = CodeAgent(model=InferenceClientModel(model_id='Qwen/Qwen2.5-Coder-3B-Instruct'), max_steps=10, verbosity_level=2,
-    tools   =  [final_answer, web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator, dynamicTool_TextToImage],
+    tools   =  [final_answer, web_search, visit_webpage, suggest_menu, catering_service_tool, 
+                superhero_party_theme_generator, dynamicTool_TextToImage, spaceTool_ImageGenerator],
     executor_type='local',
     executor_kwargs={},
     max_print_outputs_length=None,