Update app.py

app.py

@@ -5,6 +5,7 @@ import traceback
 
 from agent import answer_question
 from submit import submit_answers
+from tools.file_loader import read_pdf, read_csv, read_txt
 
 qa_data = []
 answers = []
@@ -21,14 +22,24 @@ def run_agent():
         answers = []
 
         for idx, q in enumerate(qa_data, start=1):
-            task_id = q.get("task_id", "")
-            file_name = q.get("file_name", "")
-
-            # Construct full path and ensure required fields
-            q["question"] = q.get("question", "")
-            q["file_path"] = os.path.join("data", file_name) if file_name else ""
+            task_id = q.get("task_id")
+            question = q.get("question")
+            file_text = None
 
-            response = answer_question(q)
+            file_name = q.get("file_name", "")
+            if file_name:
+                file_path = os.path.join("data", file_name)
+                try:
+                    if file_name.endswith(".pdf"):
+                        file_text = read_pdf(file_path)
+                    elif file_name.endswith(".csv"):
+                        file_text = read_csv(file_path)
+                    elif file_name.endswith(".txt"):
+                        file_text = read_txt(file_path)
+                except Exception as e:
+                    status_lines.append(f"⚠️ Failed to read file {file_name}: {e}")
+
+            response = answer_question(question, file_context=file_text, do_search=False)
             answers.append({"task_id": task_id, "submitted_answer": response})
             status_lines.append(f"✅ Q{idx}: Answered")
 
@@ -47,7 +58,7 @@ def handle_submit(username, code_link):
         return f"❌ Submission error:\n{traceback.format_exc()}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 GAIA Level 1 Agent (Local - Validation Set)")
+    gr.Markdown("# 🤖 GAIA Level 1 Agent (Local - Hugging Face CPU Inference)")
 
     with gr.Row():
         username = gr.Textbox(label="Your Hugging Face Username")