Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import requests
 import pandas as pd
 import gradio as gr
@@ -19,9 +18,7 @@ model = InferenceClientModel(
 agent = CodeAgent(
     tools=[DuckDuckGoSearchTool()],
     model=model,
-    additional_authorized_imports=[
-        "math", "re", "json", "datetime", "pandas"
-    ],
+    additional_authorized_imports=["math", "re", "json", "datetime", "pandas"],
     max_steps=6
 )
 
@@ -29,11 +26,8 @@ agent = CodeAgent(
 def get_attachment(task_id):
     """Download and read spreadsheet attachments, if available."""
     try:
-        response = requests.get(
-            f"{API_URL}/files/{task_id}", timeout=30
-        )
+        response = requests.get(f"{API_URL}/files/{task_id}", timeout=30)
         response.raise_for_status()
-
         data = BytesIO(response.content)
 
         try:
@@ -103,18 +97,12 @@ def run_evaluation(profile: gr.OAuthProfile | None):
         if item.get("file_name"):
             attachment = get_attachment(task_id)
 
-                answer = answer_question(question, attachment)
+        answer = answer_question(question, attachment)
 
+        # Stop if model inference fails
         if answer.startswith("Agent error:"):
             log += f"Model inference failed: {answer}\n"
-            return log + "\nNo answers submitted. Please fix model inference first.", answers
-
-        answers.append({
-            "task_id": task_id,
-            "submitted_answer": answer
-        })
-
-        log += f"Answer: {answer}\n"
+            return log + "\nNo answers submitted. Fix model inference first.", answers
 
         answers.append({
             "task_id": task_id,
@@ -123,7 +111,7 @@ def run_evaluation(profile: gr.OAuthProfile | None):
 
         log += f"Answer: {answer}\n"
 
-    # Submit answers to the official scoring API
+    # Submit answers to official scoring API
     payload = {
         "username": username,
         "agent_code": AGENT_CODE_URL,
@@ -178,4 +166,4 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()