Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 import json
+import os
+import traceback
+
 from agent import answer_question
 from submit import submit_answers
 from tools.file_loader import read_pdf, read_csv, read_txt
-import os
-import traceback
 
 qa_data = []
 answers = []
@@ -13,6 +14,7 @@ def run_agent():
     global qa_data, answers
     try:
         status_lines = []
+
         with open("data/metadata.jsonl", "r", encoding="utf-8") as f:
             lines = f.readlines()[:20]
             qa_data = [json.loads(line) for line in lines]
@@ -28,19 +30,19 @@ def run_agent():
             if file_name:
                 file_path = os.path.join("data", file_name)
                 try:
-                    if file_name.endswith(".pdf"):
-                        with open(file_path, "rb") as f:
+                    with open(file_path, "rb") as f:
+                        if file_name.endswith(".pdf"):
                             file_text = read_pdf(f.read())
-                    elif file_name.endswith(".csv"):
-                        with open(file_path, "rb") as f:
+                        elif file_name.endswith(".csv"):
                             file_text = read_csv(f.read())
-                    elif file_name.endswith(".txt"):
-                        with open(file_path, "rb") as f:
+                        elif file_name.endswith(".txt"):
                             file_text = read_txt(f.read())
                 except Exception as e:
                     status_lines.append(f"⚠️ Failed to read file {file_name}: {e}")
 
-            response = answer_question(question, file_context=file_text or "", do_search=False)
+            # Pass file path to agent
+            q["file_path"] = os.path.join("data", file_name) if file_name else ""
+            response = answer_question(q)
             answers.append({"task_id": task_id, "submitted_answer": response})
             status_lines.append(f"✅ Q{idx}: Answered")