Update app.py

app.py

@@ -10,14 +10,28 @@ from Gradio_UI import GradioUI
 
 @tool
 def web_search(query: str) -> str:
-    """Searches DuckDuckGo and returns top results as a string."""
+    """Searches DuckDuckGo and returns top results as a string.
+
+    Args:
+        query (str): The search query.
+
+    Returns:
+        str: The search results concatenated into a string.
+    """
     search_tool = DuckDuckGoSearchTool()
     results = search_tool(query)
     return "\n".join(results)
 
 @tool
 def visit_webpage(url: str) -> str:
-    """Fetches the content of a web page."""
+    """Fetches the content of a web page.
+
+    Args:
+        url (str): The URL to visit.
+
+    Returns:
+        str: The raw text content of the page (first 5000 characters).
+    """
     try:
         response = requests.get(url, timeout=5)
         return response.text[:5000]
@@ -26,24 +40,38 @@ def visit_webpage(url: str) -> str:
 
 @tool
 def text_splitter(text: str) -> List[str]:
-    """Splits text into chunks using langchain's CharacterTextSplitter."""
+    """Splits text into chunks using langchain's CharacterTextSplitter.
+
+    Args:
+        text (str): The input text to split.
+
+    Returns:
+        List[str]: A list of text chunks.
+    """
     splitter = CharacterTextSplitter(chunk_size=450, chunk_overlap=10)
     return splitter.split_text(text)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Gets current time in a given timezone."""
+    """Gets the current time in a specified timezone.
+
+    Args:
+        timezone (str): A timezone string, like 'America/New_York'.
+
+    Returns:
+        str: The local time formatted as a string.
+    """
     try:
         tz = pytz.timezone(timezone)
         return datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
     except Exception as e:
         return f"Error fetching time: {e}"
 
-# Load prompts.yaml
+# Load prompt templates
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Define model
+# Define the model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -53,13 +81,13 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Required final answer tool
+# Load the final answer tool
 final_answer = FinalAnswerTool()
 
 # Optional additional tool
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Build agent
+# Create the agent
 agent = CodeAgent(
     model=model,
     tools=[final_answer, web_search, visit_webpage, text_splitter, image_generation_tool],
@@ -72,5 +100,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch UI with visible errors
+# Launch the UI
 GradioUI(agent).launch(show_error=True)