Update app.py

app.py

@@ -8,12 +8,9 @@ import re
 from langchain_core.messages import HumanMessage
 from langgraph_agent import build_graph
 
-# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
 class BasicAgent:
-    """A langgraph agent."""
     def __init__(self):
         print("BasicAgent initialized.")
         self.graph = build_graph()
@@ -36,24 +33,21 @@ class BasicAgent:
 
         answer = answer.strip()
 
-        # Extract final answer after "FINAL ANSWER:"
         match = re.search(r'FINAL ANSWER:\s*(.*)', answer, re.IGNORECASE | re.DOTALL)
         if match:
             final_answer = match.group(1).strip()
-            # Remove surrounding quotes if present
             if (final_answer.startswith('"') and final_answer.endswith('"')) or \
                (final_answer.startswith("'") and final_answer.endswith("'")):
                 final_answer = final_answer[1:-1].strip()
             answer = final_answer
         else:
-            print("Warning: 'FINAL ANSWER:' not found in agent output; submitting full answer.")
+            print("Warning: 'FINAL ANSWER:' not found; submitting full answer.")
 
         if not answer:
             answer = "I am unable to determine the information using the available tools."
 
         return answer
 
-
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
@@ -102,11 +96,22 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         try:
             model_answer = agent(question_text, task_id=task_id)
+            model_answer = model_answer.strip()
+            # Clean trailing quotes/brackets if any
+            if model_answer.endswith("]'") or model_answer.endswith(']"'):
+                model_answer = model_answer[:-2].strip()
+            if (model_answer.startswith('"') and model_answer.endswith('"')) or \
+               (model_answer.startswith("'") and model_answer.endswith("'")):
+                model_answer = model_answer[1:-1].strip()
+
+            # Placeholder reasoning trace; replace with actual reasoning if available
+            reasoning_trace = f"Reasoning steps for task {task_id}."
+
             print(f"Answer for task {task_id}: '{model_answer}'")
             answers_payload.append({
                 "task_id": task_id,
                 "model_answer": model_answer,
-                # "reasoning_trace": None  # Add reasoning trace here if available
+                "reasoning_trace": reasoning_trace
             })
             results_log.append({
                 "Task ID": task_id,
@@ -118,7 +123,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             answers_payload.append({
                 "task_id": task_id,
                 "model_answer": f"AGENT ERROR: {e}",
-                # "reasoning_trace": None
+                "reasoning_trace": ""
             })
             results_log.append({
                 "Task ID": task_id,
@@ -130,7 +135,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # Serialize answers to JSON Lines format
+    # Serialize to JSON Lines string
     json_lines_str = "\n".join(json.dumps(ans, ensure_ascii=False) for ans in answers_payload)
     file_like = io.BytesIO(json_lines_str.encode("utf-8"))
     file_like.name = "submission.jsonl"
@@ -163,35 +168,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print("Submission successful.")
         results_df = pd.DataFrame(results_log)
         return final_status, results_df
-    except requests.exceptions.HTTPError as e:
-        error_detail = f"Server responded with status {e.response.status_code}."
-        try:
-            error_json = e.response.json()
-            error_detail += f" Detail: {error_json.get('detail', e.response.text)}"
-        except requests.exceptions.JSONDecodeError:
-            error_detail += f" Response: {e.response.text[:500]}"
-        status_message = f"Submission Failed: {error_detail}"
-        print(status_message)
-        results_df = pd.DataFrame(results_log)
-        return status_message, results_df
-    except requests.exceptions.Timeout:
-        status_message = "Submission Failed: The request timed out."
-        print(status_message)
-        results_df = pd.DataFrame(results_log)
-        return status_message, results_df
-    except requests.exceptions.RequestException as e:
-        status_message = f"Submission Failed: Network error - {e}"
-        print(status_message)
-        results_df = pd.DataFrame(results_log)
-        return status_message, results_df
     except Exception as e:
-        status_message = f"An unexpected error occurred during submission: {e}"
+        status_message = f"Submission Failed: {e}"
         print(status_message)
         results_df = pd.DataFrame(results_log)
         return status_message, results_df
 
 
-# --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown(
@@ -226,6 +209,7 @@ with gr.Blocks() as demo:
         outputs=[status_output, results_table]
     )
 
+
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)
     space_host_startup = os.getenv("SPACE_HOST")