Update app.py

app.py

@@ -32,13 +32,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Add image generation tool (simple example)
-@tool
-def image_generation_tool(prompt: str) -> str:
-    """Generate an image from text prompt. Returns image URL."""
-    # In real usage, replace with actual API call
-    return f"https://dummyimage.com/600x400/000/fff&text={prompt.replace(' ', '+')}"
-
 final_answer = FinalAnswerTool()
 model = HfApiModel(
     max_tokens=2096,
@@ -47,6 +40,9 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
@@ -54,7 +50,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     #tools=[final_answer], # add your tools here (don't remove final_answer)
-    tools=[final_answer,image_generation_tool, DuckDuckGoSearchTool],
+    tools=[final_answer,image_generation_tool,DuckDuckGoSearchTool],
     max_steps=6,
     verbosity_level=1,
     grammar=None,