Update app.py

app.py

@@ -1,8 +1,8 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
+import requests
 import pytz
 import yaml
-import requests
 import pandas as pd
 import json
 
@@ -13,17 +13,16 @@ from Gradio_UI import GradioUI
 # -----------------------------
 # TIME TOOL
 # -----------------------------
-
 @tool
 def get_time(timezone: str) -> str:
     """
-    Get current time in a specific timezone.
+    Get current time in a timezone.
 
     Args:
-        timezone (str): Name of timezone like 'Asia/Kathmandu', 'America/New_York'
+        timezone: Timezone name like Asia/Kathmandu
 
     Returns:
-        str: Current time in the given timezone
+        Current time string
     """
     try:
         tz = pytz.timezone(timezone)
@@ -37,7 +36,15 @@ def get_time(timezone: str) -> str:
 # -----------------------------
 @tool
 def calc(expression: str) -> str:
-    """Safe math calculator"""
+    """
+    Evaluate math expression.
+
+    Args:
+        expression: Math expression like 2+2*5
+
+    Returns:
+        Result of calculation
+    """
     try:
         return str(eval(expression))
     except:
@@ -45,11 +52,19 @@ def calc(expression: str) -> str:
 
 
 # -----------------------------
-# CSV / FILE TOOL (GAIA CRITICAL)
+# CSV TOOL
 # -----------------------------
 @tool
 def read_csv(url: str) -> str:
-    """Read CSV from URL and return structured summary"""
+    """
+    Read CSV file from URL.
+
+    Args:
+        url: CSV file URL
+
+    Returns:
+        CSV content
+    """
     try:
         df = pd.read_csv(url)
         return df.to_string()
@@ -58,11 +73,19 @@ def read_csv(url: str) -> str:
 
 
 # -----------------------------
-# JSON TOOL (VERY USEFUL FOR GAIA)
+# JSON TOOL
 # -----------------------------
 @tool
 def read_json(url: str) -> str:
-    """Read JSON from URL"""
+    """
+    Read JSON file from URL.
+
+    Args:
+        url: JSON file URL
+
+    Returns:
+        JSON content
+    """
     try:
         r = requests.get(url)
         return json.dumps(r.json(), indent=2)[:3000]
@@ -70,44 +93,31 @@ def read_json(url: str) -> str:
         return str(e)
 
 
-# -----------------------------
-# FINAL ANSWER TOOL
-# -----------------------------
 final_answer = FinalAnswerTool()
 
 
-# -----------------------------
-# MODEL (lower temperature = better GAIA)
-# -----------------------------
+# If model overloads use endpoint:
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
+
 model = HfApiModel(
-    max_tokens=2048,
+    max_tokens=2096,
     temperature=0.1,
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
 )
 
 
-# -----------------------------
-# PROMPT IMPROVEMENT (VERY IMPORTANT)
-# -----------------------------
-with open("prompts.yaml", "r") as f:
-    prompt_templates = yaml.safe_load(f)
+# optional image tool
+image_generation_tool = load_tool(
+    "agents-course/text-to-image",
+    trust_remote_code=True
+)
 
-prompt_templates["system_prompt"] = """
-You are a high-precision GAIA agent.
 
-STRICT RULES:
-1. Use tools whenever helpful.
-2. Break problem into steps internally.
-3. Always verify with tools if possible.
-4. FINAL OUTPUT = ONLY FINAL ANSWER.
-5. No explanation, no extra words.
-6. If multiple options exist, choose most likely correct.
-"""
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
 
-# -----------------------------
-# AGENT (UPGRADED STRATEGY)
-# -----------------------------
 agent = CodeAgent(
     model=model,
     tools=[
@@ -118,13 +128,14 @@ agent = CodeAgent(
         read_csv,
         read_json
     ],
-    max_steps=15,
+    max_steps=12,
     verbosity_level=1,
+    grammar=None,
+    planning_interval=None,
+    name=None,
+    description=None,
     prompt_templates=prompt_templates
 )
 
 
-# -----------------------------
-# LAUNCH
-# -----------------------------
 GradioUI(agent).launch()