Update app.py

app.py

@@ -7,6 +7,28 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+@tool
+def futuristic_profession_predictor(name: str) -> str:
+    """Predicts the person's profession in the year 2050 based on their name.
+    
+    Args:
+        name: The name of the person.
+    """
+    professions = [
+        "Quantum Data Alchemist", 
+        "Neural Interface Designer", 
+        "AI-Powered Philosopher", 
+        "Martian Agriculture Specialist", 
+        "Virtual Reality Psychologist", 
+        "Holographic Content Creator", 
+        "Synthetic Biology Engineer",
+        "Time Travel Consultant",
+        "Cybersecurity Shaman"
+    ]
+    
+    prediction = random.choice(professions)
+    return f"In the year 2050, {name} will be a {prediction}!"
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
@@ -69,7 +91,8 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    #tools=[final_answer, image_generation_tool, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
+    tools = [final_answer, futuristic_profession_predictor],
     max_steps=20,
     verbosity_level=1,
     grammar=None,