From b36f4e0584268aad350c5e440a344c4cf00775f2 Mon Sep 17 00:00:00 2001
From: killian <63927363+KillianLucas@users.noreply.github.com>
Date: Thu, 7 Nov 2024 13:37:34 -0800
Subject: [PATCH] GUI

---
 computer_use/cli.py          | 1 +
 computer_use/misc/welcome.py | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

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}