Update app.py

app.py

@@ -1,115 +1,106 @@
 #!/usr/bin/env python
 # coding=utf-8
+"""
+Robust startup file that guarantees the prompt template dict matches what
+smolagents.CodeAgent expects, without importing internal constants.
+"""
 
-import copy
 import datetime
-import requests
+import copy
 import pytz
 import yaml
 
-# smolagents imports:
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
-from smolagents import prompts as _prompts
-from smolagents.agents import DEFAULT_PROMPT_TEMPLATES
-
+# ------------- smolagents -------------
+from smolagents import CodeAgent, HfApiModel, tool
+# Local Gradio wrapper
+from Gradio_UI import GradioUI
 
-# Tools from your local folder
+# ------------ local tools -------------
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
 
-# Gradio UI from your local file
-from Gradio_UI import GradioUI
-
-###############################################################
-# Example custom tool: a non-functional template
+# ---------- example custom tool -------
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
-    """Example Tool: A non-functional tool provided as a template.
-    
-    Args:
-        arg1: The first argument.
-        arg2: The second argument.
-    """
+    """A placeholder tool to illustrate custom tool wiring."""
     return "What magic will you build ?"
 
-###############################################################
-# A working tool: get the current time in a given timezone
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Working Tool: Fetches the current local time in a specified timezone.
-    
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
+    """Return current local time in a given timezone."""
     try:
         tz = pytz.timezone(timezone)
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-###############################################################
-# Instantiate final_answer (required for returning final answers)
-final_answer = FinalAnswerTool()
+        return datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
+    except Exception as exc:
+        return f"Invalid timezone ({exc})"
 
-# Other local tools: web search & visit page
-search_tool = DuckDuckGoSearchTool()
-visit_tool = VisitWebpageTool()
 
-###############################################################
-# Define model configuration using HfApiModel
+# ---------------- model ----------------
 model = HfApiModel(
+    model_id="Qwen/Qwen3-32B",
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen3-32B',  # Adjust if overloaded or unavailable
     custom_role_conversions=None,
 )
 
-###############################################################
-# 1) Read your custom prompts.yaml
+# ------------- load YAML --------------
 try:
-    with open("prompts.yaml", 'r') as stream:
-        custom_prompts = yaml.safe_load(stream)
-except Exception as e:
-    print("Error loading prompts.yaml:", e)
-    custom_prompts = {}
-
-# 2) Merge with the built-in defaults so no keys or subkeys are missing
-prompt_templates = copy.deepcopy(DEFAULT_PROMPT_TEMPLATES)
-
-def deep_merge(base_dict, override_dict):
-    """Recursively merge override_dict into base_dict."""
-    for k, v in override_dict.items():
-        if isinstance(v, dict) and isinstance(base_dict.get(k), dict):
-            deep_merge(base_dict[k], v)
-        else:
-            base_dict[k] = v
-
-# Perform the merge
-deep_merge(prompt_templates, custom_prompts)
-
-###############################################################
-# Create the CodeAgent with the merged prompt templates
+    with open("prompts.yaml", "r") as f:
+        user_templates = yaml.safe_load(f) or {}
+except Exception as exc:
+    print("⚠️  Could not load prompts.yaml — using bare-bones defaults.", exc)
+    user_templates = {}
+
+# --------- repair / fill gaps ---------
+def ensure_block(mapping, key, default):
+    """If key missing or wrong type, replace with default."""
+    if key not in mapping or not isinstance(mapping[key], type(default)):
+        mapping[key] = copy.deepcopy(default)
+
+# Minimal viable defaults
+MIN_SYSTEM = "You are a helpful agent."
+MIN_FINAL = "Thought: I now know the answer.\nCode:\n```py\nfinal_answer({{answer}})\n```<end_code>"
+MIN_PLAN_DICT = {
+    "initial_facts": "|-\n  List any facts.",
+    "initial_plan": "|-\n  Write a high-level plan.",
+    "update_facts_pre_messages": "|-\n  Pre facts.",
+    "update_facts_post_messages": "|-\n  Post facts.",
+    "update_plan_pre_messages": "|-\n  Pre plan.",
+    "update_plan_post_messages": "|-\n  Post plan.",
+}
+MIN_AGENT_DICT = {
+    "task": "|-\n  You are {{name}}.",
+    "report": "|-\n  {{final_answer}}",
+}
+
+# Top-level fix-ups
+ensure_block(user_templates, "system_prompt", MIN_SYSTEM)
+ensure_block(user_templates, "final_answer", MIN_FINAL)
+ensure_block(user_templates, "planning", MIN_PLAN_DICT)
+ensure_block(user_templates, "managed_agent", MIN_AGENT_DICT)
+
+# If planning/managed_agent came in as strings, convert to dict with default
+if isinstance(user_templates["planning"], str):
+    user_templates["planning"] = copy.deepcopy(MIN_PLAN_DICT)
+if isinstance(user_templates["managed_agent"], str):
+    user_templates["managed_agent"] = copy.deepcopy(MIN_AGENT_DICT)
+
+# ------------- create agent -----------
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer,
-        search_tool,
-        visit_tool,
+        FinalAnswerTool(),
+        DuckDuckGoSearchTool(),
+        VisitWebpageTool(),
         get_current_time_in_timezone,
-        my_custom_tool,  # Example tool
+        my_custom_tool,
     ],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    # The merged dictionary—ensures all required keys & structure are present
-    prompt_templates=prompt_templates
+    prompt_templates=user_templates,
 )
 
-###############################################################
-# Launch Gradio UI
+# ------------- launch UI --------------
 GradioUI(agent).launch()