Update app.py

app.py

@@ -11,21 +11,21 @@ from tools.web_search import DuckDuckGoSearchTool
 
 from Gradio_UI import GradioUI
 
-# Example tool: a custom tool that currently does nothing—feel free to enhance it!
+# Example Tool (non-functioning): provided purely as a template.
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
-    """A tool that does nothing yet.
+    """Example Tool: A non-functional tool provided as a template.
     
     Args:
-        arg1: the first argument.
-        arg2: the second argument.
+        arg1: The first argument.
+        arg2: The second argument.
     """
     return "What magic will you build ?"
 
-# Example tool: get current time in a specified timezone
+# Working Tool: Fetches the current local time in a specified timezone.
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+    """Working Tool: Fetches the current local time in a specified timezone.
     
     Args:
         timezone: A string representing a valid timezone (e.g., 'America/New_York').
@@ -37,21 +37,21 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Instantiate the FinalAnswerTool (required)
+# Instantiate the FinalAnswerTool (required for returning final answers)
 final_answer = FinalAnswerTool()
 
-# Import an image generation tool from the Hub (if desired)
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Instantiate local tools for web search and webpage visits
+# Instantiate working local tools:
+# Web Search Tool: Uses DuckDuckGo to perform a web search.
 search_tool = DuckDuckGoSearchTool()   # Defined in tools/web_search.py
+
+# Webpage Visit Tool: Visits a URL and converts its content to Markdown.
 visit_tool = VisitWebpageTool()         # Defined in tools/visit_webpage.py
 
 # Define the model configuration using HfApiModel
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # You can change this if the model is overloaded.
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # Adjust if this model is overloaded.
     custom_role_conversions=None,
 )
 
@@ -59,16 +59,15 @@ model = HfApiModel(
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create the CodeAgent, including all the tools we want to enable:
+# Create the CodeAgent, including our working tools and the example tool.
 agent = CodeAgent(
     model=model,
     tools=[
         final_answer,               # Must remain included.
         search_tool,                # Enables web search.
         visit_tool,                 # Enables webpage visits.
-        my_custom_tool,             # Your custom tool (modify as desired).
-        get_current_time_in_timezone,
-        image_generation_tool       # Enables image generation via a Hub tool.
+        get_current_time_in_timezone,  # Working tool for time queries.
+        my_custom_tool,             # Example (non-functioning) tool.
     ],
     max_steps=6,
     verbosity_level=1,