diff --git a/computer_use/cli.py b/computer_use/cli.py index 5cce68e08..030f9a01f 100644 --- a/computer_use/cli.py +++ b/computer_use/cli.py @@ -15,6 +15,7 @@ def parse_args(): parser.add_argument("--api-base", "-b", help="Specify the API base URL") parser.add_argument("--api-key", "-k", help="Specify the API key") parser.add_argument("--debug", "-d", action="store_true", help="Run in debug mode") + parser.add_argument("--gui", "-g", action="store_true", help="Enable GUI control") parser.add_argument( "--yes", "-y", action="store_true", help="Automatically approve tools" ) diff --git a/computer_use/misc/welcome.py b/computer_use/misc/welcome.py index 491ccf045..8a277e7a7 100644 --- a/computer_use/misc/welcome.py +++ b/computer_use/misc/welcome.py @@ -32,6 +32,11 @@ def welcome_message(args): else: model = "` ✳ CLAUDE-3.5-SONNET `" # {"-" * (terminal_width - len(model))} # ⎇ + if args["gui"]: + gui = "` ✳ GUI CONTROL `" + else: + gui = " " * len(" ✳ GUI CONTROL ") + second_column_from_left = 20 markdown_text = f"""● @@ -42,7 +47,7 @@ def welcome_message(args): **TOOLS**{" "*(second_column_from_left-len("TOOLS"))}**MODEL** ` ❯ INTERPRETER `{" "*(second_column_from_left-len(" ❯ INTERPRETER "))}{model} ` ❚ FILE EDITOR `{" "*(second_column_from_left-len(" ❚ FILE EDITOR "))}**MODE** -` ⎋ GUI CONTROL `{" "*(second_column_from_left-len(" ⎋ GUI CONTROL "))}` ? ASK CONFIRMATION ` +{gui}{" "*(second_column_from_left-len(" ⎋ GUI CONTROL "))}` ? ASK CONFIRMATION ` {random_tip}