Update app.py

app.py

@@ -1,17 +1,17 @@
 import gradio as gr
-from agent_code import agent
+from submit_gaia import run_submission  # This is the function that does the whole GAIA submission
 
-def ask_question(q):
-    state = {
-        "question": q,
-        "thoughts": [],
-        "tool_results": [],
-        "answer": None
-    }
-    result = agent.invoke(state)
-    return result["answer"]
+def on_submit_click():
+    result = run_submission()  # Run the big GAIA submission
+    return f"Leaderboard submission result:\n{result}"
 
-gr.Interface(fn=ask_question,
-             inputs=gr.Textbox(label="Ask a question"),
-             outputs="text",
-             title="GIAI LangGraph Agent").launch()
+with gr.Blocks() as demo:
+    gr.Markdown("# GIAI LangGraph Agent Demo")
+
+    submit_button = gr.Button("Submit All GAIA Level 1 Questions")
+    output = gr.Textbox(label="Submission Result", lines=10)
+
+    # When the button is clicked, call on_submit_click, show output in textbox
+    submit_button.click(fn=on_submit_click, inputs=[], outputs=[output])
+
+demo.launch()