From 3fd0aab6bafa99fb804582b3104c5f8a932bcda2 Mon Sep 17 00:00:00 2001 From: JetSquirrel Date: Sat, 14 Oct 2023 01:46:55 +0800 Subject: [PATCH 1/2] add gradio params --- agentverse/gui.py | 5 +++-- agentverse_command/main_simulation_gui.py | 19 ++++++++++++++++++- 2 files changed, 21 insertions(+), 3 deletions(-) diff --git a/agentverse/gui.py b/agentverse/gui.py index 911207f48..8bdd8fe24 100644 --- a/agentverse/gui.py +++ b/agentverse/gui.py @@ -30,13 +30,14 @@ class GUI: the UI of frontend """ - def __init__(self, task: str, tasks_dir: str): + def __init__(self, task: str, tasks_dir: str,ui_kwargs: Dict[str, str]): """ init a UI. default number of students is 0 """ self.messages = [] self.task = task + self.ui_kwargs = ui_kwargs if task == "pipeline_brainstorming": self.backend = TaskSolving.from_task(task, tasks_dir) else: @@ -502,5 +503,5 @@ def respond(message, chat_history): show_progress=False, ) - demo.queue(concurrency_count=5, max_size=20).launch() + demo.queue(concurrency_count=5, max_size=20).launch(**self.ui_kwargs) # demo.launch() diff --git a/agentverse_command/main_simulation_gui.py b/agentverse_command/main_simulation_gui.py index b91df38c0..a16eb9907 100644 --- a/agentverse_command/main_simulation_gui.py +++ b/agentverse_command/main_simulation_gui.py @@ -2,6 +2,17 @@ from agentverse.gui import GUI from argparse import ArgumentParser +def is_share(v): + if isinstance(v, bool): + return v + if v.lower() in ('yes', 'true', 't', 'y', '1'): + return True + elif v.lower() in ('no', 'false', 'f', 'n', '0'): + return False + else: + import argparse + raise argparse.ArgumentTypeError('Boolean value expected.') + parser = ArgumentParser() parser.add_argument("--task", type=str, default="simulation/nlp_classroom_9players") parser.add_argument( @@ -9,11 +20,17 @@ type=str, default=os.path.join(os.path.dirname(__file__), "..", "agentverse", "tasks"), ) +parser.add_argument("--share", + type=is_share, + default=False, + help="Create a publicly shareable link") +parser.add_argument("--server_name",type=str,default="127.0.0.1",help="Server name") + args = parser.parse_args() def cli_main(): - ui = GUI(args.task, args.tasks_dir) + ui = GUI(args.task, args.tasks_dir,ui_kwargs={"share":args.share,"server_name":args.server_name}) ui.launch() From c63bc28e56674cefddd544b818e4013dfa5a6cb3 Mon Sep 17 00:00:00 2001 From: JetSquirrel Date: Sun, 15 Oct 2023 13:51:07 +0800 Subject: [PATCH 2/2] remove is_share --- agentverse_command/main_simulation_gui.py | 24 +++++++---------------- 1 file changed, 7 insertions(+), 17 deletions(-) diff --git a/agentverse_command/main_simulation_gui.py b/agentverse_command/main_simulation_gui.py index a16eb9907..75d7fbd3f 100644 --- a/agentverse_command/main_simulation_gui.py +++ b/agentverse_command/main_simulation_gui.py @@ -2,17 +2,6 @@ from agentverse.gui import GUI from argparse import ArgumentParser -def is_share(v): - if isinstance(v, bool): - return v - if v.lower() in ('yes', 'true', 't', 'y', '1'): - return True - elif v.lower() in ('no', 'false', 'f', 'n', '0'): - return False - else: - import argparse - raise argparse.ArgumentTypeError('Boolean value expected.') - parser = ArgumentParser() parser.add_argument("--task", type=str, default="simulation/nlp_classroom_9players") parser.add_argument( @@ -21,18 +10,19 @@ def is_share(v): default=os.path.join(os.path.dirname(__file__), "..", "agentverse", "tasks"), ) parser.add_argument("--share", - type=is_share, - default=False, - help="Create a publicly shareable link") -parser.add_argument("--server_name",type=str,default="127.0.0.1",help="Server name") + action='store_true', + default=False, + help="Create a publicly shareable link") +parser.add_argument("--server_name", + type=str, + default="127.0.0.1", + help="Server name") args = parser.parse_args() - def cli_main(): ui = GUI(args.task, args.tasks_dir,ui_kwargs={"share":args.share,"server_name":args.server_name}) ui.launch() - if __name__ == "__main__": cli_main()