Update app.py

app.py

@@ -22,7 +22,16 @@ class BasicAgent:
 
         messages = [HumanMessage(content=f"Can you answer this question please ? {question}")]
 
-        messages = self.agent_graph.invoke({"messages": messages, "question": question}, {"recursion_limit": 10})
+        messages = agent_graph.invoke(
+            input={"messages": messages, "question": question, "input_file": file_name}, 
+            config={
+                "recursion_limit": 10,
+                "callbacks": [langfuse_handler]
+            }
+        )
+
+        for m in messages['messages']:
+            m.pretty_print()
 
         try:
             regex_result = re.search(r"FINAL ANSWER:\s*(?P<answer>.*)$", messages['messages'][-1].content)
@@ -30,6 +39,8 @@ class BasicAgent:
         except:
             regex_result = re.search(r"\s*(?P<answer>.*)$", messages['messages'][-1].content)
             answer = regex_result.group("answer")
+
+        print(answer)
         
         return answer
 
@@ -99,6 +110,7 @@ def try_one_question( profile: gr.OAuthProfile | None):
             continue
         try:
             submitted_answer = agent(question_text)
+            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})
         except Exception as e: