Umer797 commited on
Commit
f06b73f
·
verified ·
1 Parent(s): 4a62492

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,17 +2,17 @@ import os
2
  import gradio as gr
3
  import pandas as pd
4
  import traceback
5
- from langchain_core.messages import HumanMessage
6
  from graph_builder import build_graph
7
  from api_client import fetch_questions, submit_answers
8
 
9
  class GaiaAgent:
10
  def __init__(self):
11
  self.graph = build_graph()
 
12
 
13
  def __call__(self, question):
14
  state = {"question": question}
15
- result_state = self.graph.invoke(state)
16
  return result_state["final_answer"]
17
 
18
  def run_and_submit_all(profile):
@@ -71,7 +71,7 @@ with gr.Blocks() as demo:
71
 
72
  run_button.click(
73
  fn=run_and_submit_all,
74
- inputs=[login_button], # wire login profile as input
75
  outputs=[status_output, results_table]
76
  )
77
 
 
2
  import gradio as gr
3
  import pandas as pd
4
  import traceback
 
5
  from graph_builder import build_graph
6
  from api_client import fetch_questions, submit_answers
7
 
8
  class GaiaAgent:
9
  def __init__(self):
10
  self.graph = build_graph()
11
+ self.executor = self.graph.compile() # ✅ compile the graph for execution
12
 
13
  def __call__(self, question):
14
  state = {"question": question}
15
+ result_state = self.executor.invoke(state) # ✅ run using executor
16
  return result_state["final_answer"]
17
 
18
  def run_and_submit_all(profile):
 
71
 
72
  run_button.click(
73
  fn=run_and_submit_all,
74
+ inputs=[login_button],
75
  outputs=[status_output, results_table]
76
  )
77