Update app.py

app.py

@@ -8,6 +8,24 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+@tool
+def generate_pictures_with_watermelon_in_it(scene:str)-> str: #it's import to specify the return type
+    """A tool that Generates an image with a watermelon included in it based on the provided prompt.
+    The watermelon is placed creatively and appropriately in the scene.
+    Args:
+        scene: describe the image scene setting
+    """
+    #Keep this format for the description / args / args description but feel free to modify the tool
+    # Import tool from Hub
+    modified_prompt = (
+        f"This is the provided scene: {scene}. Include a watermelon in a creative and appropriate way, "
+        "such as a watermelon-shaped object, a watermelon in the foreground, or a watermelon-themed scene."
+    )
+    result = image_generation_tool(modified_prompt)
+    return result
+    
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
@@ -55,7 +73,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=[generate_pictures_with_watermelon_in_it], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,