Update app.py

app.py

@@ -27,7 +27,16 @@ def Searchtool(question: str) -> str:
     """
     search_tool = DuckDuckGoSearchTool()  # variable différente du nom de la fonction
     response = search_tool(question)
-    return response
+    return response 
+
+@tool 
+def generate_image(prompt : str) -> str : 
+    """ genere une image a partir d'une description textuelle .
+    Args : 
+    prompt : une description de l'image a générer 
+    returns : URL ou chemin de l'image générer """
+
+    return image_generation_tool(prompt)
 
 
 @tool
@@ -67,7 +76,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer , Searchtool ,  get_current_time_in_timezone ], ## add your tools here (don't remove final answer)
+    tools=[final_answer , Searchtool ,  get_current_time_in_timezone , generate_image ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,