Update app.py

app.py

@@ -4,7 +4,6 @@ import inspect
 import gradio as gr
 import requests
 import pandas as pd
-import time
 from langchain_core.messages import HumanMessage
 from agent import build_graph
 
@@ -16,8 +15,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-
-
 class BasicAgent:
     """A langgraph agent."""
     def __init__(self):
@@ -30,8 +27,7 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
-        return answer[14:]
-
+        return answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -93,9 +89,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
-        
-        # time.sleep(10)
-        
         try:
             submitted_answer = agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})