add new tools for agent

app.py

@@ -69,19 +69,40 @@
 
 # GradioUI(agent).launch()
 
-# new code: (by gemini)
+# new code: (by gemini + me + Agents course)
+# 1. Import spaces at the top
 import datetime
 import pytz
 import requests
 import yaml
-
-# 1. Import spaces at the top
 import spaces
-
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-from tools.final_answer import FinalAnswerTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool, FinalAnswerTool, InferenceClientModel
+#from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+# Gloable vars and important stuff:
+
+search_tool = DuckDuckGoSearchTool() #for search tool # the easy way.
+final_answer = FinalAnswerTool() #final answer tool
+# image generation tool
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+# our llm:
+#model = HfApiModel(
+model = InferenceClientModel( 
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
+)
+
+# in `prompts.yml` we have sys_prompt.
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream) 
+
+
+# Tools (user def tools):
+
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """A tool that does nothing yet 
@@ -91,6 +112,20 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     """
     return "What magic will you build ?"
 
+@tool
+def hbty(b_day: str) -> str: #hbty = Happy Birthday to you
+    """
+    A tool for wishing Happy Birthday, If use query or inputs have keyword "birthday".
+    Args:
+        b_day = a string, reprasenting valid question, or statement, if this staement have "birthday" or "bday" then wish "Happy Birthday to you"
+    """
+    #return "Happy Birthday To You !!!! :)"
+    cleaned_input = b_day.lower()
+    if "birthday" in cleaned_input or "bday" in cleaned_input:
+        return "Happy Birthday To You !!!! :)"
+    else:
+        return "No birthday mentioned in the statement."
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -104,37 +139,55 @@ 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 get_search_results(query: str) -> str:
+    """A tool that fetches the data from the internet, for a specific search query.
+    Args:
+        query: A string representing a valid question or request for example, 'What is the capitol of india?'
+    """
+    try:
+        # DuckDuckGoSearchTool is callable directly with the query string
+        search_results = _ddg_search(query)
+        return search_results
+    except Exception as e:
+        return f"Error executing search query '{query}': {str(e)}"
 
-final_answer = FinalAnswerTool()
+@tool
+def generate_code_snippet(language: str, task_description: str) -> str:
+    """Generates a clean code snippet in a requested programming language for a specific task.
 
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
-)
+    Args:
+        language: The programming language to write (e.g., 'python', 'cpp', 'javascript', 'sql').
+        task_description: A clear description of what the code should do.
+    """
+    prompt = f"Write a clean, efficient {language} code snippet to fulfill this task: {task_description}. Provide brief comments."
 
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    try:
+        # Call the model to generate code
+        response = model([{"role": "user", "content": prompt}])
+        return response.content
+    except Exception as e:
+        return f"Error generating code: {str(e)}"
 
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-    
+
+# Runner code:
+# 2. Decorate the agent execution function for ZeroGPU
+@spaces.GPU
+def run_agent(prompt):
+    return agent.run(prompt)
+
+# our agent:
 agent = CodeAgent(
     model=model,
-    tools=[final_answer],
+    tools=[final_answer,generate_code_snippet,get_search_results,get_current_time_in_timezone,hbty,my_custom_tool,search_tool,image_generation_tool], # add your tools here (don't remove final_answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates # Pass system prompt to CodeAgent
 )
 
-# 2. Decorate the agent execution function for ZeroGPU
-@spaces.GPU
-def run_agent(prompt):
-    return agent.run(prompt)
-
 # Launch Gradio UI
 GradioUI(agent).launch()