Update app.py

app.py

@@ -17,8 +17,18 @@ prompt_templates = {
 }
 
 # === Tools Defined Inline ===
+
 @tool
 def get_weather(city: str) -> str:
+    """
+    Get the current weather for a given city.
+
+    Args:
+        city (str): Name of the city to get weather for.
+
+    Returns:
+        str: Weather description.
+    """
     try:
         response = requests.get(f"https://wttr.in/{city}?format=3")
         return f"The current weather in {city} is: {response.text}" if response.status_code == 200 else f"Failed to fetch weather data for {city}."
@@ -27,6 +37,15 @@ def get_weather(city: str) -> str:
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
+    """
+    Get current time in a specific timezone.
+
+    Args:
+        timezone (str): Timezone string like 'America/New_York'.
+
+    Returns:
+        str: Current time as string.
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -36,6 +55,15 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def generate_pun(topic: str) -> str:
+    """
+    Generate a pun based on a topic.
+
+    Args:
+        topic (str): A topic to create a pun from.
+
+    Returns:
+        str: A humorous pun related to the topic.
+    """
     puns = {
         "banana": "I find bananas very *a-peeling*.",
         "fish": "You cod not be more fin-tastic!",
@@ -48,13 +76,22 @@ def generate_pun(topic: str) -> str:
 
 @tool
 def final_answer(text: str) -> str:
+    """
+    Finalize and return the last answer.
+
+    Args:
+        text (str): Final response.
+
+    Returns:
+        str: Same response passed through.
+    """
     return text
 
 # === Load Remote Tools ===
 web_search = load_tool("smol-ai/web-search")
 image_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# === Smart Agent ===
+# === Smart Agent Definition ===
 class BasicAgent:
     def __init__(self):
         model = HfApiModel(
@@ -87,7 +124,7 @@ class BasicAgent:
             return f"Error while answering: {e}"
 
 
-# === Evaluation Logic ===
+# === Evaluation Logic (Do Not Touch) ===
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
@@ -166,7 +203,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
 
-# === Gradio UI ===
+# === Gradio UI (Do Not Touch) ===
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""