machineconfig 1.94__py3-none-any.whl → 1.95__py3-none-any.whl

machineconfig/jobs/python_custom_installers/scripts/linux/vscode.sh

@@ -5,8 +5,12 @@
 # This script installs Visual Studio Code on Ubuntu/Debian-based Linux distributions
 # Reference: https://code.visualstudio.com/docs/setup/linux
 
-echo """
-#=======================================================================
+# sudo rm /etc/apt/sources.list.d/vscode.sources
+# sudo apt-get update
+
+
+
+echo """#=======================================================================
 🔑 ADDING REPOSITORY KEYS | Setting up Microsoft repository
 #=======================================================================
 """
@@ -29,8 +33,7 @@ else
     echo "✅ VS Code repository already configured"
 fi
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 📦 INSTALLING VS CODE | Updating and installing packages
 #=======================================================================
 """
@@ -43,8 +46,7 @@ sudo nala update -o Dir::Etc::sourcelist="sources.list.d/vscode.list" -o Dir::Et
 echo "📥 Installing Visual Studio Code..."
 sudo nala install code -y
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 🔄 CHECKING FOR VS CODE INSIDERS | Updating if installed
 #=======================================================================
 """
@@ -57,8 +59,7 @@ else
     echo "ℹ️ VS Code Insiders not installed, skipping"
 fi
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ✅ INSTALLATION COMPLETE | VS Code has been installed successfully
 #=======================================================================
 """

machineconfig/jobs/python_custom_installers/scripts/linux/warp-cli.sh

@@ -5,8 +5,7 @@
 # This script installs Cloudflare WARP client on Ubuntu/Debian-based Linux distributions
 # Reference: https://pkg.cloudflareclient.com/
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 🔍 DETECTING SYSTEM | Identifying OS distribution version
 #=======================================================================
 """
@@ -28,8 +27,7 @@ get_ubuntu_base_version() {
 ubuntu_version=$(get_ubuntu_base_version)
 echo "📋 Detected distribution: $ubuntu_version"
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 🔑 ADDING REPOSITORY KEYS | Setting up Cloudflare repository
 #=======================================================================
 """
@@ -42,8 +40,7 @@ curl -fsSL https://pkg.cloudflareclient.com/pubkey.gpg | sudo gpg --yes --dearmo
 echo "📝 Adding Cloudflare WARP repository to sources list..."
 echo "deb [signed-by=/usr/share/keyrings/cloudflare-warp-archive-keyring.gpg] https://pkg.cloudflareclient.com/ $ubuntu_version main" | sudo tee /etc/apt/sources.list.d/cloudflare-client.list
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 📦 INSTALLING CLOUDFLARE WARP | Updating and installing packages
 #=======================================================================
 """
@@ -56,8 +53,7 @@ sudo nala update
 echo "📥 Installing Cloudflare WARP..."
 sudo nala install cloudflare-warp -y
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 🔧 REGISTERING WARP CLIENT | Setting up new registration
 #=======================================================================
 """
@@ -66,8 +62,7 @@ echo """
 echo "📡 Registering WARP client..."
 warp-cli registration new
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ✅ INSTALLATION COMPLETE | Cloudflare WARP has been installed successfully
 #=======================================================================
 """

machineconfig/jobs/python_custom_installers/scripts/linux/wezterm.sh

@@ -5,8 +5,7 @@
 # This script installs WezTerm terminal emulator on Ubuntu/Debian-based Linux distributions
 # Reference: https://wezfurlong.org/wezterm/install/linux.html
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 🔑 ADDING REPOSITORY KEYS | Setting up WezTerm repository
 #=======================================================================
 """
@@ -19,8 +18,7 @@ curl -fsSL https://apt.fury.io/wez/gpg.key | sudo gpg --yes --dearmor -o /usr/sh
 echo "📝 Adding WezTerm repository to sources list..."
 echo 'deb [signed-by=/usr/share/keyrings/wezterm-fury.gpg] https://apt.fury.io/wez/ * *' | sudo tee /etc/apt/sources.list.d/wezterm.list
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 📦 INSTALLING WEZTERM | Updating and installing packages
 #=======================================================================
 """
@@ -33,8 +31,7 @@ sudo nala update
 echo "📥 Installing WezTerm terminal emulator..."
 sudo nala install wezterm -y
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ✅ INSTALLATION COMPLETE | WezTerm has been installed successfully
 #=======================================================================
 """

machineconfig/profile/shell.py

@@ -8,10 +8,15 @@ from crocodile.meta import Terminal
 from machineconfig.utils.utils import LIBRARY_ROOT, REPO_ROOT, display_options
 import platform
 from typing import Optional
+from rich.console import Console
+from rich.panel import Panel
 
 
 system = platform.system()
 
+console = Console()
+BOX_WIDTH = 100  # Define BOX_WIDTH or get it from a config
+
 # --------------------------------------- SHELL PROFILE --------------------------------------------------------
 # modification of shell profile by additing dirs to PATH
 # Shell profile is either in dotfiles and is synced (as in Windows), hence no need for update, or is updated on the fly (for Linux)
@@ -25,25 +30,17 @@ def create_default_shell_profile():
     else: source = f"source {LIBRARY_ROOT.joinpath('settings/shells/bash/init.sh').collapseuser().to_str().replace('~', '$HOME')}"
 
     if source in profile: 
-        print(f"""
-╭{'─' * 78}╮
-│ 🔄 PROFILE | Skipping init script sourcing - already present in profile      │
-╰{'─' * 78}╯
-""")
+        console.print(Panel("🔄 PROFILE | Skipping init script sourcing - already present in profile", title="[bold blue]Profile[/bold blue]", border_style="blue"))
     else:
-        print(f"""
-╭{'─' * 78}╮
-│ 📝 PROFILE | Adding init script sourcing to profile                          │
-╰{'─' * 78}╯
-""")
+        console.print(Panel("📝 PROFILE | Adding init script sourcing to profile", title="[bold blue]Profile[/bold blue]", border_style="blue"))
         profile += "\n" + source + "\n"
         if system == "Linux":
             res = Terminal().run("cat /proc/version").op
             if "microsoft" in res.lower() or "wsl" in res.lower():
                 profile += "\ncd ~"  # this is to make sure that the current dir is not in the windows file system, which is terribly slow and its a bad idea to be there anyway.
-                print("📌 WSL detected - adding 'cd ~' to profile to avoid Windows filesystem")
+                console.print("📌 WSL detected - adding 'cd ~' to profile to avoid Windows filesystem")
         profile_path.create(parents_only=True).write_text(profile)
-        print("✅ Profile updated successfully")
+        console.print(Panel("✅ Profile updated successfully", title="[bold blue]Profile[/bold blue]", border_style="blue"))
 
 
 def get_shell_profile_path():
@@ -56,12 +53,7 @@ def get_shell_profile_path():
             raise ValueError(f"Could not get profile path for Windows. Got {res}")
     elif system == "Linux": profile_path = P("~/.bashrc").expanduser()
     else: raise ValueError(f"Not implemented for this system {system}")
-    
-    print(f"""
-╔{'═' * 78}╗
-║ 🐚 SHELL PROFILE | Working with path: `{profile_path}`
-╚{'═' * 78}╝
-""")
+    console.print(Panel(f"🐚 SHELL PROFILE | Working with path: `{profile_path}`", title="[bold blue]Shell Profile[/bold blue]", border_style="blue"))
     return profile_path
 
 
@@ -69,12 +61,7 @@ def main_env_path(choice: Optional[str] = None, profile_path: Optional[str] = No
     env_path = LIBRARY_ROOT.joinpath("profile/env_path.toml").readit()
     dirs = env_path[f'path_{system.lower()}']['extension']
 
-    print(f"""
-╔{'═' * 78}╗
-║ 🔍 ENVIRONMENT | Current PATH variables:
-╚{'═' * 78}╝
-""")
-    P.get_env().PATH.print()
+    console.print(Panel("🔍 ENVIRONMENT | Current PATH variables:", title="[bold blue]Environment[/bold blue]", border_style="blue"))
 
     if choice is None:
         tmp = display_options(msg="Which directory to add?", options=dirs + ["all", "none(EXIT)"], default="none(EXIT)")
@@ -83,23 +70,19 @@ def main_env_path(choice: Optional[str] = None, profile_path: Optional[str] = No
         if str(choice) != "all": dirs = [choice]
     if choice == "none(EXIT)": return
 
-    print(f"\n📌 Adding directories to PATH: {dirs}")
+    console.print(f"\n📌 Adding directories to PATH: {dirs}")
     addition = PathVar.append_temporarily(dirs=dirs)
     profile_path_obj = P(profile_path) if isinstance(profile_path, str) else get_shell_profile_path()
     profile_path_obj.copy(name=profile_path_obj.name + ".orig_" + randstr())
-    print(f"💾 Created backup of profile: {profile_path_obj.name}.orig_*")
+    console.print(f"💾 Created backup of profile: {profile_path_obj.name}.orig_*")
     profile_path_obj.modify_text(addition, addition, replace_line=False, notfound_append=True)
-    print("✅ PATH variables added to profile successfully")
+    console.print(Panel("✅ PATH variables added to profile successfully", title="[bold blue]Environment[/bold blue]", border_style="blue"))
 
 
 def main_add_sources_to_shell_profile(profile_path: Optional[str] = None, choice: Optional[str] = None):
     sources: list[str] = LIBRARY_ROOT.joinpath("profile/sources.toml").readit()[system.lower()]['files']
 
-    print(f"""
-╭{'─' * 78}╮
-│ 🔄 Adding sources to shell profile                                          │
-╰{'─' * 78}╯
-""")
+    console.print(Panel("🔄 Adding sources to shell profile", title="[bold blue]Sources[/bold blue]", border_style="blue"))
 
     if choice is None:
         choice_obj = display_options(msg="Which patch to add?", options=sources + ["all", "none(EXIT)"], default="none(EXIT)", multi=True)
@@ -123,26 +106,22 @@ def main_add_sources_to_shell_profile(profile_path: Optional[str] = None, choice
         if file not in profile:
             if system == "Windows": 
                 profile += f"\n. {file}"
-                print(f"➕ Added PowerShell source: {file}")
+                console.print(f"➕ Added PowerShell source: {file}")
             elif system == "Linux": 
                 profile += f"\nsource {file}"
-                print(f"➕ Added Bash source: {file}")
+                console.print(f"➕ Added Bash source: {file}")
             else: raise ValueError(f"Not implemented for this system {system}")
         else: 
-            print(f"⏭️  Source already present: {file}")
+            console.print(f"⏭️  Source already present: {file}")
     
     profile_path_obj.write_text(profile)
-    print("✅ Shell profile updated with sources")
+    console.print(Panel("✅ Shell profile updated with sources", title="[bold blue]Sources[/bold blue]", border_style="blue"))
 
 
 def main_add_patches_to_shell_profile(profile_path: Optional[str] = None, choice: Optional[str] = None):
     patches: list[str] = list(LIBRARY_ROOT.joinpath(f"profile/patches/{system.lower()}").search().apply(lambda x: x.as_posix()))
     
-    print(f"""
-╭{'─' * 78}╮
-│ 🩹 Adding patches to shell profile                                          │
-╰{'─' * 78}╯
-""")
+    console.print(Panel("🩹 Adding patches to shell profile", title="[bold blue]Patches[/bold blue]", border_style="blue"))
     
     if choice is None:
         choice_chosen = display_options(msg="Which patch to add?", options=list(patches) + ["all", "none(EXIT)"], default="none(EXIT)", multi=False)
@@ -150,10 +129,10 @@ def main_add_patches_to_shell_profile(profile_path: Optional[str] = None, choice
         choice = choice_chosen
     if choice == "none(EXIT)": return None
     elif str(choice) == "all": 
-        print("📌 Adding all patches to profile")
+        console.print("📌 Adding all patches to profile")
     else: 
         patches = [choice]
-        print(f"📌 Adding selected patch: {choice}")
+        console.print(f"📌 Adding selected patch: {choice}")
 
     profile_path_obj = P(profile_path) if isinstance(profile_path, str) else get_shell_profile_path()
     profile = profile_path_obj.read_text()
@@ -162,19 +141,19 @@ def main_add_patches_to_shell_profile(profile_path: Optional[str] = None, choice
         patch_path_obj = P(patch_path)
         patch = patch_path_obj.read_text()
         if patch in profile: 
-            print(f"⏭️  Patch already present: {patch_path_obj.name}")
+            console.print(f"⏭️  Patch already present: {patch_path_obj.name}")
         else: 
             profile += "\n" + patch
-            print(f"➕ Added patch: {patch_path_obj.name}")
+            console.print(f"➕ Added patch: {patch_path_obj.name}")
 
     if system == "Linux":
         res = Terminal().run("cat /proc/version").op
         if "microsoft" in res.lower() or "wsl" in res.lower():
             profile += "\ncd ~"  # this is to make sure that the current dir is not in the windows file system, which is terribly slow and its a bad idea to be there anyway.
-            print("📌 WSL detected - adding 'cd ~' to profile to avoid Windows filesystem")
+            console.print("📌 WSL detected - adding 'cd ~' to profile to avoid Windows filesystem")
 
     profile_path_obj.write_text(profile)
-    print("✅ Shell profile updated with patches")
+    console.print(Panel("✅ Shell profile updated with patches", title="[bold blue]Patches[/bold blue]", border_style="blue"))
 
 
 if __name__ == '__main__':

machineconfig/scripts/cloud/init.sh

@@ -4,16 +4,14 @@
 #=======================================================================
 # This script initializes cloud configuration settings and sets up the environment
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 🔑 COLLECTING CONFIGURATION PARAMETERS | Setting up cloud environment
 #=======================================================================
 """
 
 # Check for required environment variables and prompt if not set
 if [ -z "$CLOUD_CONFIG_NAME" ]; then
-    echo """
-    #-----------------------------------------------------------------------
+    echo """    #-----------------------------------------------------------------------
     📋 CONFIG NAME | Specify cloud configuration to run
     #-----------------------------------------------------------------------
     """
@@ -23,8 +21,7 @@ if [ -z "$CLOUD_CONFIG_NAME" ]; then
 fi
 
 if [ -z "$SHARE_URL" ]; then
-    echo """
-    #-----------------------------------------------------------------------
+    echo """    #-----------------------------------------------------------------------
     🔗 SHARE URL | Provide cloud share URL for configuration files
     #-----------------------------------------------------------------------
     ℹ️ To get share_url, go to dotfiles and run: cloud_copy . :^ --config ss
@@ -35,8 +32,7 @@ if [ -z "$SHARE_URL" ]; then
 fi
 
 if [ -z "$DECRYPTION_PASSWORD" ]; then
-    echo """
-    #-----------------------------------------------------------------------
+    echo """    #-----------------------------------------------------------------------
     🔒 SECURITY | Enter decryption password
     #-----------------------------------------------------------------------
     """
@@ -45,8 +41,7 @@ if [ -z "$DECRYPTION_PASSWORD" ]; then
     echo ""
 fi
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 📦 INSTALLING ESSENTIALS | Setting up core dependencies
 #=======================================================================
 """
@@ -70,8 +65,7 @@ echo "🚀 Activating Python virtual environment..."
 echo "📋 Setting up code repositories..."
 curl bit.ly/cfgreposlinux -L | bash
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ⚙️ CONFIGURING ENVIRONMENT | Setting up dotfiles
 #=======================================================================
 """
@@ -93,8 +87,7 @@ source ~/code/machineconfig/src/machineconfig/setup_linux/symlinks.sh
 echo "🔄 Reloading shell configuration..."
 . ~/.bashrc
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 📦 INSTALLING DEVELOPMENT TOOLS | Setting up development environment
 #=======================================================================
 """
@@ -111,8 +104,7 @@ python -m fire machineconfig.scripts.python.devops_devapps_install main --which=
 echo "🔄 Reloading shell configuration..."
 . ~/.bashrc
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ✅ FINALIZING CONFIGURATION | Running cloud-specific initialization
 #=======================================================================
 """
@@ -121,8 +113,7 @@ echo """
 echo "⚙️ Running cloud-specific configuration: $CLOUD_CONFIG_NAME"
 . $HOME/dotfiles/config/cloud/$CLOUD_CONFIG_NAME/init.sh
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ✅ INITIALIZATION COMPLETE | Cloud environment has been set up successfully
 #=======================================================================
 """

machineconfig/scripts/linux/fire

@@ -2,38 +2,20 @@
 
 . $HOME/scripts/activate_ve 've'
 
-op_script=$HOME/tmp_results/shells/python_return_command.sh
-
-# echo "May be removing $op_script"
-# sleep 1
+# Generate random string of length 5
+random_str=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 10 | head -n 1)
+export op_script=$HOME/tmp_results/shells/$random_str/python_return_command.sh
+# op_script=$HOME/tmp_results/shells/$random_str/python_return_command.sh
 
 if [ -f "$op_script" ]; then
   rm $op_script
 fi
 
 
-# if [ "$TERM" = "xterm-256color" ] || [ "$TERM" = "alacritty"  ]; then
-#   {
-#     chafa "$HOME/code/machineconfig/assets/aafire.webp" --speed 1 --duration 1 --symbols ascii
-#   } || {
-#     echo "An error occurred while executing the script."
-#   }
-# fi
-
-# timeout 0.5s aafire  #  -driver slang
-# clear
-
-# python $(dirname $(dirname $0))/python/fire_jobs.py "$@"
-# echo "Firing script preprocessor"
-# sleep 1
-
 python -m machineconfig.scripts.python.fire_jobs "$@"
-# ~/venvs/ve/bin/python -m machineconfig.scripts.python.fire_jobs "$@"
 
 if [ -f "$op_script" ]; then
-  # echo "Executing $op_script"
-  # sleep 1
-
+  echo "Running $op_script"
   chmod +x $op_script
   set --  # clear all positional parameters
   source $op_script  # sourcing the script means the args passed to the this script are passed to the sourced script which cofuses the sources script as it contains activate_ve and it doesn't understand first.sh arguments.
@@ -42,5 +24,4 @@ fi
 
 if [ -n "$VIRTUAL_ENV" ]; then
   deactivate || true
-
 fi

machineconfig/scripts/linux/share_cloud.sh

@@ -7,8 +7,7 @@
 
 # Check if arguments are provided
 if [ $# -eq 0 ]; then
-    echo """
-    #=======================================================================
+    echo """    #=======================================================================
     ❌ ERROR | No arguments specified
     #=======================================================================
     
@@ -27,8 +26,7 @@ if tty -s; then
     
     # Check if the file exists
     if [ ! -e "$file" ]; then
-        echo """
-        #=======================================================================
+        echo """        #=======================================================================
         ❌ ERROR | File not found
         #=======================================================================
         
@@ -37,8 +35,7 @@ if tty -s; then
         return 1
     fi
     
-    echo """
-    #=======================================================================
+    echo """    #=======================================================================
     📤 UPLOADING | Sharing file to transfer.sh
     #=======================================================================
     """
@@ -56,8 +53,7 @@ if tty -s; then
 else
     # Pipe mode - reading from stdin
     file_name=$1
-    echo """
-    #=======================================================================
+    echo """    #=======================================================================
     📤 UPLOADING | Sharing from stdin to transfer.sh
     #=======================================================================
     """
@@ -65,16 +61,14 @@ else
     curl --progress-bar --upload-file "-" "https://transfer.sh/$file_name" | tee /dev/null
     
     # Display QR code for the URL
-    echo """
-    #=======================================================================
+    echo """    #=======================================================================
     📱 QR CODE | Scan with mobile device to access file
     #=======================================================================
     """
     qr "https://transfer.sh/$file_name"
 fi
 
-echo """
-#=======================================================================
+echo """#=======================================================================
 ✅ UPLOAD COMPLETE | File is available at the URL above
 #=======================================================================
 """

machineconfig/scripts/python/archive/im2text.py

@@ -1,36 +1,36 @@
-"""Convert image to text.
-"""
+# """Convert image to text.
+# """
 
 
-# import sys
-from crocodile.msc.odds import capture_from_webcam
-from crocodile.meta import Terminal
+# # import sys
+# from crocodile.msc.odds import capture_from_webcam
+# from crocodile.meta import Terminal
+# from rich.console import Console
+# from rich.panel import Panel
+# import pytesseract
+# from PIL import Image
 
-print(f"""
-╔{'═' * 60}╗
-║ 📸 Image to Text Converter
-╚{'═' * 60}╝
-""")
+# console = Console()
 
-print("📷 Capturing image from webcam...")
-img_path = capture_from_webcam(show=False, wait=False, save=True)
-print(f"✅ Image captured and saved to: {img_path}")
+# console.print(Panel("📸 Image to Text Converter", title="Status", expand=False))
 
-# img_rgb = cv2.cvtColor(img_cv, cv2.COLOR_BGR2RGB)
-# import cv2
-# img_cv = cv2.imread(r'C:\Users\alex\Downloads\name.jpg')
+# print("📷 Capturing image from webcam...")
+# img_path = capture_from_webcam(show=False, wait=False, save=True)
+# print(f"✅ Image captured and saved to: {img_path}")
 
-# sys.path.insert(0, P.home().joinpath("AppData/Local/Tesseract-OCR").str)
-# import pytesseract
-# print(pytesseract.image_to_string(img_cv))
-
-print("\n🔍 Processing image with Tesseract OCR...")
-q = Terminal().run(f"cd ~/AppData/Local/Tesseract-OCR; pytesseract '{img_path}'", shell="pwsh").capture().op
-
-print(f"""
-╔{'═' * 60}╗
-║ 📄 Extracted Text Result:
-╠{'═' * 60}╣
-{q}
-╚{'═' * 60}╝
-""")
+# # img_rgb = cv2.cvtColor(img_cv, cv2.COLOR_BGR2RGB)
+# # import cv2
+# # img_cv = cv2.imread(r'C:\Users\alex\Downloads\name.jpg')
+
+# # sys.path.insert(0, P.home().joinpath("AppData/Local/Tesseract-OCR").str)
+# # import pytesseract
+# # print(pytesseract.image_to_string(img_cv))
+
+# print("\n🔍 Processing image with Tesseract OCR...")
+# q = Terminal().run(f"cd ~/AppData/Local/Tesseract-OCR; pytesseract '{img_path}'", shell="pwsh").capture().op
+
+# try:
+#     text = pytesseract.image_to_string(Image.open(img_path))
+#     console.print(Panel(text, title="📄 Extracted Text Result:", title_align="left", expand=False))
+# except FileNotFoundError:
+#     print(f"Error: Image file not found at {img_path}")

machineconfig/scripts/python/archive/tmate_conn.py

@@ -4,6 +4,11 @@ from pathlib import Path
 import random
 import string
 import os
+from rich.console import Console
+from rich.panel import Panel
+
+
+console = Console()
 
 
 def get_conn_string(sess_name: str) -> str:
@@ -15,29 +20,21 @@ def get_conn_string(sess_name: str) -> str:
 
 
 def main():
-    print(f"""
-╔{'═' * 60}╗
-║ 🔌 Tmate Connection Manager
-╚{'═' * 60}╝
-""")
+    console.print(Panel("🔌 Tmate Connection Manager", title="[bold]Welcome[/bold]"))
     
     parser = argparse.ArgumentParser(description='Tmate launcher')
     parser.add_argument("sess_name", help="session name", default=random.choices(list(string.digits + string.ascii_letters), k=20))
     args = parser.parse_args()
     
-    print(f"🔍 Looking up session: {args.sess_name}")
+    console.print(f"🔍 Looking up session: {args.sess_name}")
     conn_string = get_conn_string(args.sess_name)
     
-    print(f"""
-╭{'─' * 60}╮
-│ 🔗 Connection String: ssh {conn_string}
-╰{'─' * 60}╯
-""")
+    console.print(Panel(f"SSH Connection String: ssh {conn_string}", title="[bold green]SSH Connection[/bold green]"))
     
-    print("🚀 Connecting to tmate session...")
+    console.print("🚀 Connecting to tmate session...")
     os.system(f"ssh {conn_string}")
     
-    print("✅ Connection closed")
+    console.print("✅ Connection closed")
 
 
 if __name__ == '__main__':