Create app/app.py

app/app.py

@@ -0,0 +1,73 @@
+import gradio as gr
+import json
+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 = []
+
+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]
+
+        answers = []
+
+        for idx, q in enumerate(qa_data, start=1):
+            task_id = q.get("task_id")
+            question = q.get("question")
+            file_text = None
+
+            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")
+
+        status_lines.append(f"\n🎯 Done. {len(answers)} questions answered.")
+        return "\n".join(status_lines)
+
+    except Exception as e:
+        return f"❌ Error:\n{traceback.format_exc()}"
+
+def handle_submit(username, code_link):
+    try:
+        result = submit_answers(username, code_link, answers)
+        return result
+    except Exception as e:
+        return f"❌ Submission error:\n{traceback.format_exc()}"
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🤖 GAIA Level 1 Agent (Local - Validation Set)")
+
+    with gr.Row():
+        username = gr.Textbox(label="Your Hugging Face Username")
+        code_link = gr.Textbox(label="Your Space Code Link (/tree/main)")
+
+    status = gr.Textbox(label="Status", lines=15)
+
+    run_btn = gr.Button("🧠 Run Agent")
+    submit_btn = gr.Button("🚀 Submit Answers")
+
+    run_btn.click(fn=run_agent, outputs=status)
+    submit_btn.click(fn=handle_submit, inputs=[username, code_link], outputs=status)
+
+if __name__ == "__main__":
+    demo.launch()