Update app.py

app.py

@@ -1,21 +1,19 @@
 from smolagents import CodeAgent, InferenceClientModel, load_tool, tool
 import datetime
 import pytz
-import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 
-# ────────────────────────────────────────────────
-# Custom Tools
-# ────────────────────────────────────────────────
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Get the current local time in the requested timezone.
 
     Args:
-        timezone: IANA timezone name (e.g. 'Asia/Kolkata', 'America/New_York')
+        timezone: IANA timezone name (e.g. 'Asia/Kolkata', 'America/New_York', 'Europe/London')
+
+    Returns:
+        Human-readable string with current time or error message
     """
     try:
         tz = pytz.timezone(timezone)
@@ -29,13 +27,17 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
-    """Placeholder custom tool — replace with real logic as needed."""
-    return f"Tool called → arg1 = {arg1!r} | arg2 = {arg2} → What would you like to build?"
+    """Placeholder custom tool — replace with real logic as needed.
+
+    Args:
+        arg1: some text input (e.g. a name, description, or keyword)
+        arg2: some integer input (e.g. a count, year, or multiplier)
 
+    Returns:
+        Demonstration response string
+    """
+    return f"Tool called → arg1 = {arg1!r} | arg2 = {arg2} → What would you like to build?"
 
-# ────────────────────────────────────────────────
-# Main
-# ────────────────────────────────────────────────
 
 def main():
     final_answer_tool = FinalAnswerTool()
@@ -46,36 +48,14 @@ def main():
         max_tokens=2048,
     )
 
-    # Optional text-to-image tool from Hub
+    # Optional: text-to-image tool
     image_tool = None
     try:
         image_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-        print("✅ Text-to-image tool loaded from Hub")
-    except Exception as e:
-        print(f"⚠️ Text-to-image tool failed to load: {e}")
-
-    # ── Load your custom prompts.yaml + FIX the missing final_answer key ──
-    prompt_templates = None
-    try:
-        with open("prompts.yaml", encoding="utf-8") as f:
-            prompt_templates = yaml.safe_load(f)
-        print("✅ Custom prompt templates loaded")
-
-        # CRITICAL FIX for latest smolagents (v1.17+)
-        # The library now strictly requires the "final_answer" key
-        if prompt_templates and "final_answer" not in prompt_templates:
-            prompt_templates["final_answer"] = {
-                "pre_messages": "Here is the final answer:\n",
-                "post_messages": ""
-            }
-            print("✅ Added missing 'final_answer' template (required by latest smolagents)")
-
-    except FileNotFoundError:
-        print("⚠️ prompts.yaml not found → using default system prompt")
+        print("Text-to-image tool loaded")
     except Exception as e:
-        print(f"⚠️ Failed to load prompts.yaml: {e}")
+        print(f"Text-to-image load failed: {e}")
 
-    # Tools list
     tools = [
         final_answer_tool,
         get_current_time_in_timezone,
@@ -84,22 +64,20 @@ def main():
     if image_tool is not None:
         tools.append(image_tool)
 
-    # Create the agent
     agent = CodeAgent(
         tools=tools,
         model=model,
-        max_steps=10,                    # increased for better performance
+        max_steps=10,
         verbosity_level=1,
         stream_outputs=True,
-        # planning_interval=4,           # uncomment if you want planning mode
-        prompt_templates=prompt_templates,   # now guaranteed to be valid
+        # planning_interval=4,           # optional
+        # prompt_templates=...           # only if you really need custom prompts
     )
 
-    # Launch Gradio UI (perfect for HF Spaces / local)
     GradioUI(agent).launch(
         server_name="0.0.0.0",
         server_port=7860,
-        # share=True,                    # uncomment for temporary public link
+        # share=True,
     )