Update app.py

app.py

@@ -2,38 +2,72 @@ import gradio as gr
 from api import get_all_questions, submit_answers, get_file
 from agent import answer_question
 from tools.file_loader import read_pdf, read_csv, read_txt
+import traceback
 
 qa_data = []
 answers = []
 
 def run_agent():
     global qa_data, answers
-    qa_data = get_all_questions()
-    answers = []
-    for q in qa_data:
-        task_id = q['task_id']
-        question = q['question']
-        file_text = None
-        if q.get("files"):
-            for fname in q["files"]:
-                raw = get_file(task_id, fname)
-                if fname.endswith(".pdf"):
-                    file_text = read_pdf(raw)
-                elif fname.endswith(".csv"):
-                    file_text = read_csv(raw)
-                elif fname.endswith(".txt"):
-                    file_text = read_txt(raw)
-                break
-
-        response = answer_question(question, file_context=file_text, do_search=True)
-        answers.append({"task_id": task_id, "submitted_answer": response})
-    return f"✅ Agent answered {len(answers)} questions. Ready to submit!"
+    try:
+        print("[INFO] Fetching questions from GAIA API...")
+        qa_data = get_all_questions()
+        answers = []
+
+        print(f"[INFO] Total questions retrieved: {len(qa_data)}")
+
+        for q in qa_data:
+            task_id = q.get('task_id')
+            question = q.get('question')
+            file_text = None
+
+            if q.get("files"):
+                for fname in q["files"]:
+                    try:
+                        raw = get_file(task_id, fname)
+                        print(f"[INFO] Loaded file: {fname}")
+
+                        if fname.endswith(".pdf"):
+                            file_text = read_pdf(raw)
+                        elif fname.endswith(".csv"):
+                            file_text = read_csv(raw)
+                        elif fname.endswith(".txt"):
+                            file_text = read_txt(raw)
+                        else:
+                            print(f"[WARN] Unsupported file type: {fname}")
+                    except Exception as fe:
+                        print(f"[ERROR] Failed to load file {fname}: {str(fe)}")
+                    break  # Load only the first file
+
+            print(f"[INFO] Answering question: {question}")
+            response = answer_question(question, file_context=file_text, do_search=True)
+            print(f"[INFO] Answer: {response}")
+
+            answers.append({
+                "task_id": task_id,
+                "submitted_answer": response
+            })
+
+        return f"✅ Agent answered {len(answers)} questions. Ready to submit!"
+
+    except Exception as e:
+        error_log = traceback.format_exc()
+        print("[ERROR]", error_log)
+        return f"❌ Error:\n{error_log}"
 
 def handle_submit(username, code_link):
-    return submit_answers(username, code_link, answers)
+    try:
+        print(f"[INFO] Submitting answers for user: {username}")
+        result = submit_answers(username, code_link, answers)
+        print(f"[INFO] Submission response: {result}")
+        return result
+    except Exception as e:
+        error_log = traceback.format_exc()
+        print("[ERROR]", error_log)
+        return f" Submission failed:\n{error_log}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 GAIA Level 1 Agent")
+    gr.Markdown("# GAIA Level 1 Agent")
 
     with gr.Row():
         username = gr.Textbox(label="Hugging Face Username")
@@ -41,11 +75,11 @@ with gr.Blocks() as demo:
 
     status = gr.Textbox(label="Status", lines=4)
 
-    run_btn = gr.Button("🔍 Run Agent")
-    submit_btn = gr.Button("📤 Submit Answers")
+    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()
+# Make sure this launches in Spaces
+demo.launch()