vladd19 commited on
Commit
9ae2f4e
·
verified ·
1 Parent(s): 4bb8d2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -9
app.py CHANGED
@@ -562,6 +562,7 @@ class BasicAgent:
562
  g.add_node("solve_code", self.solve_code)
563
  g.add_node("solve_direct", self.solve_direct)
564
  g.add_node("solve_research", self.solve_research)
 
565
 
566
  g.add_node("verify_answer", self.verify_answer)
567
  g.add_node("final_cleaner", self.final_cleaner)
@@ -579,6 +580,7 @@ class BasicAgent:
579
  "solve_code": "solve_code",
580
  "solve_direct": "solve_direct",
581
  "solve_research": "solve_research",
 
582
  },
583
  )
584
 
@@ -589,6 +591,7 @@ class BasicAgent:
589
  "solve_code",
590
  "solve_direct",
591
  "solve_research",
 
592
  ]:
593
  g.add_edge(node, "verify_answer")
594
 
@@ -636,6 +639,7 @@ class BasicAgent:
636
  "solve_code",
637
  "solve_direct",
638
  "solve_research",
 
639
  }
640
  return route if route in allowed else "solve_research"
641
 
@@ -755,8 +759,7 @@ class BasicAgent:
755
  print(f"[web preview] {repr(web_results[:500])}")
756
 
757
  context = truncate_text(
758
- f"Search query:\n{query}\n\n",
759
- f"Web search results:\n{web_results}\n\n",
760
  MAX_SEARCH_CONTEXT_CHARS,
761
  )
762
 
@@ -1006,6 +1009,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
1006
  return status, pd.DataFrame(results_log)
1007
 
1008
 
 
 
 
 
 
1009
  with gr.Blocks() as demo:
1010
  gr.Markdown("# Basic Agent Evaluation Runner — Routed LangGraph")
1011
  gr.Markdown(
@@ -1016,7 +1024,10 @@ with gr.Blocks() as demo:
1016
  """
1017
  )
1018
 
1019
- gr.LoginButton()
 
 
 
1020
  run_button = gr.Button("Run Evaluation & Submit All Answers")
1021
  status_output = gr.Textbox(label="Run Status / Submission Result", lines=6, interactive=False)
1022
  results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
@@ -1026,14 +1037,11 @@ with gr.Blocks() as demo:
1026
  outputs=[status_output, results_table],
1027
  )
1028
 
1029
-
1030
- space_host_startup = os.getenv("SPACE_HOST")
1031
- space_id_startup = os.getenv("SPACE_ID")
1032
-
1033
  if space_host_startup:
1034
  print(f"✅ SPACE_HOST found: {space_host_startup}")
1035
  if space_id_startup:
1036
  print(f"✅ SPACE_ID found: {space_id_startup}")
1037
 
1038
- print("Launching Gradio Interface for Routed LangGraph Agent Evaluation...")
1039
- demo.launch(debug=True, share=False)
 
 
562
  g.add_node("solve_code", self.solve_code)
563
  g.add_node("solve_direct", self.solve_direct)
564
  g.add_node("solve_research", self.solve_research)
565
+ g.add_node("solve_youtube", self.solve_youtube)
566
 
567
  g.add_node("verify_answer", self.verify_answer)
568
  g.add_node("final_cleaner", self.final_cleaner)
 
580
  "solve_code": "solve_code",
581
  "solve_direct": "solve_direct",
582
  "solve_research": "solve_research",
583
+ "solve_youtube": "solve_youtube",
584
  },
585
  )
586
 
 
591
  "solve_code",
592
  "solve_direct",
593
  "solve_research",
594
+ "solve_youtube",
595
  ]:
596
  g.add_edge(node, "verify_answer")
597
 
 
639
  "solve_code",
640
  "solve_direct",
641
  "solve_research",
642
+ "solve_youtube",
643
  }
644
  return route if route in allowed else "solve_research"
645
 
 
759
  print(f"[web preview] {repr(web_results[:500])}")
760
 
761
  context = truncate_text(
762
+ f"Search query:\n{query}\n\nWeb search results:\n{web_results}\n\n",
 
763
  MAX_SEARCH_CONTEXT_CHARS,
764
  )
765
 
 
1009
  return status, pd.DataFrame(results_log)
1010
 
1011
 
1012
+ space_host_startup = os.getenv("SPACE_HOST")
1013
+ space_id_startup = os.getenv("SPACE_ID")
1014
+ oauth_available = bool(space_host_startup or space_id_startup or os.getenv("HF_TOKEN"))
1015
+
1016
+
1017
  with gr.Blocks() as demo:
1018
  gr.Markdown("# Basic Agent Evaluation Runner — Routed LangGraph")
1019
  gr.Markdown(
 
1024
  """
1025
  )
1026
 
1027
+ if oauth_available:
1028
+ gr.LoginButton()
1029
+ else:
1030
+ gr.Markdown("Hugging Face OAuth is disabled locally. Run inside a Space or set `HF_TOKEN`.")
1031
  run_button = gr.Button("Run Evaluation & Submit All Answers")
1032
  status_output = gr.Textbox(label="Run Status / Submission Result", lines=6, interactive=False)
1033
  results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
1037
  outputs=[status_output, results_table],
1038
  )
1039
 
 
 
 
 
1040
  if space_host_startup:
1041
  print(f"✅ SPACE_HOST found: {space_host_startup}")
1042
  if space_id_startup:
1043
  print(f"✅ SPACE_ID found: {space_id_startup}")
1044
 
1045
+ if __name__ == "__main__":
1046
+ print("Launching Gradio Interface for Routed LangGraph Agent Evaluation...")
1047
+ demo.launch(debug=True, share=False)