Update app.py

app.py

@@ -46,6 +46,42 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def funny_hi_response(_: str = "") -> str:
+    """Returns a funny or sarcastic response to someone saying 'Hi'."""
+    import random
+
+    responses = [
+        "Oh hey, look who remembered how to type!",
+        "Hi? That’s it? After everything we’ve been through?",
+        "Welcome back, I almost started enjoying the silence.",
+        "Hi there, social butterfly. Slow day?",
+        "If I had a nickel for every 'Hi' I got… I’d still be broke.",
+        "Wow, a 'Hi'? You sure you didn’t mean to send that to your other assistant?",
+        "Hello, mortal. What wisdom dost thou seek today?",
+        "Hi. I was hoping for pizza, but I guess you’ll do.",
+        "You say 'Hi', I hear 'I'm bored, entertain me.'",
+        "A wild user appeared! What shall we do now?"
+    ]
+
+    return random.choice(responses)
+
+@tool
+def answer_with_duckduckgo(query: str) -> str:
+    """Uses DuckDuckGo to search and return an answer to any user question.
+    
+    Args:
+        query: A natural language question from the user.
+    Returns:
+        A short answer or summary from DuckDuckGo.
+    """
+    try:
+        duck_tool = DuckDuckGoSearchTool()
+        result = duck_tool.run(query)
+        return result or "No relevant results found, but hey, at least I tried."
+    except Exception as e:
+        return f"Oops! I ran into an issue while searching: {str(e)}"
+
 
 final_answer = FinalAnswerTool()
 
@@ -68,7 +104,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_two_liner_roast,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer,get_two_liner_roast,get_current_time_in_timezone,funny_hi_response,answer_with_duckduckgo], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,