Update app.py

app.py

@@ -7,6 +7,7 @@ import re
 from agentic import build_graph, langfuse_handler
 from langchain_core.messages import HumanMessage
 from langfuse.callback import CallbackHandler
+from pprint import pprint
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -22,14 +23,16 @@ class BasicAgent:
         print(input)
         question = input.get("question", "No question provided")
         file_name = input.get("file_name", "")
+        task_id = input.get("task_id", "")
 
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         messages = [HumanMessage(content=f"Can you answer this question please ? {question}")]
 
+        user_input = {"messages": messages, "question": question, "input_file": file_name, "task_id": task_id}
 
         messages = self.agent_graph.invoke(
-            input={"messages": messages, "question": question, "input_file": file_name},
+            input=user_input,
             config={
                 "recursion_limit": 10,
                 "callbacks": [langfuse_handler]
@@ -84,6 +87,7 @@ def try_one_question( profile: gr.OAuthProfile | None):
         response = requests.get(one_question_url, timeout=15)
         response.raise_for_status()
         questions_data = response.json()
+        pprint(f"QUESTION_DATA : {questions_data}")
         if not questions_data:
              print("Fetched questions list is empty.")
              return "Fetched questions list is empty or invalid format.", None
@@ -117,7 +121,7 @@ def try_one_question( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(input={"question": question_text, "file_name": question_file})
+            submitted_answer = agent(input={"question": question_text, "file_name": question_file, "task_id": task_id})
             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})