fix code

app.py

@@ -8,9 +8,12 @@ import gradio as gr
 import pandas as pd
 import requests
 
-from smolagents import CodeAgent, tool
-from smolagents import DuckDuckGoSearchTool, VisitWebpageTool
-from smolagents import OpenAIServerModel
+from smolagents import CodeAgent, tool, DuckDuckGoSearchTool, VisitWebpageTool
+
+try:
+    from smolagents import OpenAIServerModel
+except ImportError:
+    from smolagents import OpenAIModel as OpenAIServerModel
 
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -18,29 +21,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 @tool
 def get_task_file_url(task_id: str) -> str:
-    """
-    Returns the download URL for the file attached to a GAIA task.
-
-    Args:
-        task_id: The GAIA task id.
-
-    Returns:
-        Direct URL to the attached file for this task.
-    """
     return f"{DEFAULT_API_URL}/files/{task_id}"
 
 
 @tool
 def download_file(url: str) -> str:
-    """
-    Downloads a file from a URL to a temporary local path and returns that path.
-
-    Args:
-        url: The file URL.
-
-    Returns:
-        Local file path as a string.
-    """
     suffix = ""
     match = re.search(r"\.([a-zA-Z0-9]{1,6})(?:\?|$)", url)
     if match:
@@ -60,16 +45,6 @@ def download_file(url: str) -> str:
 
 @tool
 def inspect_local_text_file(path: str, max_chars: int = 12000) -> str:
-    """
-    Reads a local text-like file and returns a preview.
-
-    Args:
-        path: Local file path.
-        max_chars: Max number of characters to return.
-
-    Returns:
-        Text preview.
-    """
     p = Path(path)
     try:
         return p.read_text(encoding="utf-8", errors="ignore")[:max_chars]
@@ -79,17 +54,6 @@ def inspect_local_text_file(path: str, max_chars: int = 12000) -> str:
 
 @tool
 def analyze_spreadsheet(path: str, instruction: str) -> str:
-    """
-    Loads an Excel/CSV spreadsheet with pandas and executes a constrained analysis
-    instruction by giving the LLM a schema + preview.
-
-    Args:
-        path: Local spreadsheet path.
-        instruction: What to compute.
-
-    Returns:
-        JSON string with columns, preview, dtypes, shape.
-    """
     p = Path(path)
     if p.suffix.lower() in [".csv", ".tsv"]:
         sep = "\t" if p.suffix.lower() == ".tsv" else ","
@@ -113,15 +77,6 @@ def analyze_spreadsheet(path: str, instruction: str) -> str:
 
 @tool
 def python_compute(code: str) -> str:
-    """
-    Executes trusted short Python snippets for local analysis.
-
-    Args:
-        code: Python code. Must assign the final value to a variable named `result`.
-
-    Returns:
-        Stringified result.
-    """
     local_vars = {}
     global_vars = {
         "pd": pd,
@@ -147,7 +102,6 @@ def clean_final_answer(text: str) -> str:
     answer = re.sub(r"^(FINAL ANSWER\s*:\s*)", "", answer, flags=re.IGNORECASE)
     answer = re.sub(r"^(Answer\s*:\s*)", "", answer, flags=re.IGNORECASE)
     answer = re.sub(r"[ \t]+", " ", answer).strip()
-
     return answer
 
 
@@ -162,8 +116,7 @@ Rules:
 - For exact-match scoring, your final output must be ONLY the answer and nothing else.
 - Do not include explanations in your final answer.
 - Do not include the words FINAL ANSWER.
-- When a question requests a format (comma-separated list, USD with two decimals, single word, date, etc.),
-  obey that exact format.
+- When a question requests a format, obey that exact format.
 - If the answer is numeric and the prompt requests decimals, format correctly.
 """
 
@@ -184,22 +137,37 @@ class GaiaAgent:
             max_tokens=1200,
         )
 
-self.agent = CodeAgent(
-    model=self.model,
-    tools=[
-        DuckDuckGoSearchTool(),
-        VisitWebpageTool(),
-        get_task_file_url,
-        download_file,
-        inspect_local_text_file,
-        analyze_spreadsheet,
-        python_compute,
-    ],
-    add_base_tools=False,
-    max_steps=10,
-    instructions=SYSTEM_PROMPT,
-)
-    
+        try:
+            self.agent = CodeAgent(
+                model=self.model,
+                tools=[
+                    DuckDuckGoSearchTool(),
+                    VisitWebpageTool(),
+                    get_task_file_url,
+                    download_file,
+                    inspect_local_text_file,
+                    analyze_spreadsheet,
+                    python_compute,
+                ],
+                add_base_tools=False,
+                max_steps=10,
+                instructions=SYSTEM_PROMPT,
+            )
+        except TypeError:
+            self.agent = CodeAgent(
+                model=self.model,
+                tools=[
+                    DuckDuckGoSearchTool(),
+                    VisitWebpageTool(),
+                    get_task_file_url,
+                    download_file,
+                    inspect_local_text_file,
+                    analyze_spreadsheet,
+                    python_compute,
+                ],
+                add_base_tools=False,
+                max_steps=10,
+            )
 
     def __call__(self, question: str, task_id: str) -> str:
         full_prompt = f"""
@@ -316,6 +284,5 @@ with gr.Blocks() as demo:
         outputs=[status_output, results_table],
     )
 
-
 if __name__ == "__main__":
     demo.launch(debug=True)