Update app.py

app.py

@@ -97,27 +97,27 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',#
 custom_role_conversions=None,
 )
 
-@tool 
-def image_generation_tool(promt: str)->any:
-    """A tool that generate image from user`s promt.
-    Args:
-        promt: description of image
-    """
-    try:
-        image = image_generation_tool(promt)
-        return image
+#@tool 
+#def image_generation_tool(promt: str)->any:
+#    """A tool that generate image from user`s promt.
+#    Args:
+#        promt: description of image
+#    """
+#   try:
+#        image = image_generation_tool(promt)
+#        return image
         
-    except Exception as e:
-        return f'We have some problem {e}'
+#    except Exception as e:
+#        return f'We have some problem {e}'
 # Import tool from Hub
-#image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+search_tool = DuckDuckGoSearchTool()
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,enhanced_translate,image_generation_tool,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    tools=[final_answer,enhanced_translate,image_generation_tool,search_tool,DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,