Update app.py

app.py

@@ -2,17 +2,18 @@ import os
 import gradio as gr
 import pandas as pd
 import traceback
-from graph_builder import build_graph, AgentState
+from graph_builder import build_graph
 from api_client import fetch_questions, submit_answers
 
 class GaiaAgent:
     def __init__(self):
         self.graph = build_graph()
-        self.executor = self.graph.compile()  # ✅ compile graph into executable
+        self.executor = self.graph.compile()
 
     def __call__(self, question):
-        result_state = self.executor.invoke({"question": question})  # ✅ pass as dict
-        return result_state["final_answer"]
+        # Explicitly pass the state dictionary
+        result_state = self.executor.invoke({"question": question})
+        return result_state.get("final_answer", "No answer generated.")
 
 def run_and_submit_all(profile):
     try: