Update app.py

app.py

@@ -2,17 +2,17 @@ import os
 import gradio as gr
 import pandas as pd
 import traceback
-from graph_builder import build_graph
+from graph_builder import build_graph, AgentState
 from api_client import fetch_questions, submit_answers
 
 class GaiaAgent:
     def __init__(self):
         self.graph = build_graph()
-        self.executor = self.graph.compile()  # ✅ compile the graph for execution
+        self.executor = self.graph.compile()
 
     def __call__(self, question):
-        state = {"question": question}
-        result_state = self.executor.invoke(state)  # ✅ run using executor
+        initial_state = AgentState({"question": question})  # ✅ validated state init
+        result_state = self.executor.invoke(initial_state)
         return result_state["final_answer"]
 
 def run_and_submit_all(profile):