Update app.py

app.py

@@ -9,7 +9,7 @@ import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from smolagents.models import OpenAIServerModel
 
-# System prompt required by GAIA
+# System prompt for GAIA evaluation
 SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
 Report your thoughts, and finish your answer with the following template:
 FINAL ANSWER: [YOUR FINAL ANSWER].
@@ -18,13 +18,11 @@ of numbers and/or strings. If you are asked for a number, don't use comma to wri
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# Extract FINAL ANSWER
-
 def extract_final_answer(response: str) -> str:
+    """Extract the FINAL ANSWER: portion from the model's response."""
     match = re.search(r"FINAL ANSWER:\s*(.*)", response, re.IGNORECASE)
     return match.group(1).strip() if match else response.strip()
 
-# MyAgent class
 class MyAgent:
     def __init__(self):
         self.model = OpenAIServerModel(model_id="gpt-4")
@@ -39,17 +37,17 @@ class MyAgent:
             {"role": "user", "content": question}
         ]
         try:
-            response = self.model.chat(messages)
+            # Correct usage: model is callable, pass messages directly
+            response = self.model(messages)
             return extract_final_answer(response)
         except Exception as e:
             import traceback
             traceback.print_exc()
             return f"AGENT ERROR: {e}"
 
-# Evaluation and submission
-
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
+
     if profile:
         username = profile.username
         print(f"User logged in: {username}")
@@ -79,6 +77,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     results_log = []
     answers_payload = []
+
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
@@ -121,20 +120,22 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"An unexpected error occurred during submission: {e}", pd.DataFrame(results_log)
 
-# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
-    gr.Markdown("""
-    **Instructions:**
-    1. Clone this space, modify code to define your agent's logic, tools, and packages.
-    2. Log in to your Hugging Face account using the button below.
-    3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see your score.
+    gr.Markdown(
+        """
+        **Instructions:**
+        1. Clone this space, modify code to define your agent's logic, tools, and packages.
+        2. Log in to your Hugging Face account using the button below.
+        3. Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see your score.
 
-    **Note:** Submitting can take some time.
-    """)
+        **Note:** Submitting can take some time.
+        """
+    )
 
     gr.LoginButton()
     run_button = gr.Button("Run Evaluation & Submit All Answers")
+
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
@@ -164,4 +165,3 @@ if __name__ == "__main__":
     demo.launch(debug=True, share=False)
 
 
-