Update app.py

app.py

@@ -18,12 +18,13 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent_graph = build_graph()
-    def __call__(self, question: str) -> str:
+    def __call__(self, input: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         messages = [HumanMessage(content=f"Can you answer this question please ? {question}")]
 
-        file_name = ""
+        question = input.get("question", "No question provided")
+        file_name = input.get("file_name", "")
         messages = self.agent_graph.invoke(
             input={"messages": messages, "question": question, "input_file": file_name}, 
             config={
@@ -106,12 +107,14 @@ def try_one_question( profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        question_file = item.get("file_name")
         print(f"question: {question_text}")
+        print(f"file_name: {question_file}")
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(input={"question": question_text, "file_name": question_file})
             print(f"SUBMITTED_ANSWER : {submitted_answer}")
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})