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
 
@@ -26,11 +25,11 @@ class BasicAgent:
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # Wrap the question in a HumanMessage from langchain_core
         messages = [HumanMessage(content=question)]
-        messages = self.graph.invoke({"messages": messages})
-        answer = messages['messages'][-1].content
-        return answer[14:]
+        result = self.graph.invoke({"messages": messages})
+        answer = result['messages'][-1].content
+        return answer  # kein [14:] mehr nötig!
+
 
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
@@ -93,9 +92,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})
@@ -208,4 +204,4 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)