trying to extract file

app copy.py

@@ -5,6 +5,8 @@ import inspect
 import pandas as pd
 from langchain_core.messages import HumanMessage
 from agent import build_graph 
+from huggingface_hub import HfApi, hf_hub_download
+from logging import logger
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -22,8 +24,28 @@ class BasicAgent:
         messages = [HumanMessage(content=question)]
         result = self.graph.invoke({"messages": messages})
         answer = result['messages'][-1].content
-        print(f"Agent returning answer: {answer}")
         return answer
+
+def file_extract(local_file_path,task_file_paths):
+    if local_file_path:
+        try:
+            # GAIA's file_path is relative to the dataset repo root.
+            # Download the file into the allowed cache and get its local path.
+            resolved_path = hf_hub_download(
+                repo_id="gaia-benchmark/GAIA",
+                filename=local_file_path,  # e.g. "2023/test/<attachment-id>.pdf"
+                repo_type="dataset",
+            )
+
+            task_file_paths[str(task_id)] = resolved_path
+            logger.debug(
+                f"Stored file path mapping for task_id {task_id}: {resolved_path}"
+            )
+        except Exception as e:
+            logger.warning(
+                f"Could not download file '{local_file_path}' for task_id {task_id}: {e}. "
+                "Mapping skipped."
+            )
         
 agent = BasicAgent()
 questions_url = f"{DEFAULT_API_URL}/questions"
@@ -34,12 +56,18 @@ for item in questions_data[:5]:
     question_text = item.get("question")
     if question_text is None:
         continue
+    files_text = item.get("files")
+    task_id = item.get("task_id")
+    # file = file_extract(,task_id)
+    print(files_text,task_id)
     output = agent(question_text)
     print("Q:", question_text)
     print("A:", output)
     print("-" * 40)
 
 
+
+
 # def run_and_submit_all( profile: gr.OAuthProfile | None):
 #     """
 #     Fetches all questions, runs the BasicAgent on them, submits all answers,