Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
-
|
| 1039 |
-
|
|
|
|
|
|
| 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)
|