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

machineconfig/profile/shell.py

@@ -1,16 +1,22 @@
 """shell
 """
 
-from crocodile.environment import PathVar
-from crocodile.core import randstr
-from crocodile.file_management import P
+from machineconfig.utils.utils2 import randstr
+from machineconfig.utils.path_reduced import P as PathExtended, modify_text
 from crocodile.meta import Terminal
 from machineconfig.utils.utils import LIBRARY_ROOT, REPO_ROOT, display_options
 import platform
-from typing import Optional
+import os
+from typing import Literal, Optional
+from rich.console import Console
+from rich.panel import Panel
 
 
 system = platform.system()
+sep = ";" if system == "Windows" else ":"  # PATH separator, this is special for PATH object, not to be confused with PathExtended.sep (normal paths), usually / or \
+PATH = os.environ["PATH"].split(sep)  # this is a list of paths in PATH variable, not a crocodile.file_management.P object.
+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
@@ -21,60 +27,65 @@ system = platform.system()
 def create_default_shell_profile():
     profile_path = get_shell_profile_path()
     profile = profile_path.read_text()
-    if system == "Windows": source = f". {LIBRARY_ROOT.joinpath('settings/shells/pwsh/init.ps1').collapseuser().to_str().replace('~', '$HOME')}"
-    else: source = f"source {LIBRARY_ROOT.joinpath('settings/shells/bash/init.sh').collapseuser().to_str().replace('~', '$HOME')}"
+    if system == "Windows": source = f". {str(LIBRARY_ROOT.joinpath('settings/shells/pwsh/init.ps1').collapseuser()).replace('~', '$HOME')}"
+    else: source = f"source {str(LIBRARY_ROOT.joinpath('settings/shells/bash/init.sh').collapseuser()).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")
-        profile_path.create(parents_only=True).write_text(profile)
-        print("✅ Profile updated successfully")
+                console.print("📌 WSL detected - adding 'cd ~' to profile to avoid Windows filesystem")
+        profile_path.parent.mkdir(parents=True, exist_ok=True)
+        profile_path.write_text(profile)
+        console.print(Panel("✅ Profile updated successfully", title="[bold blue]Profile[/bold blue]", border_style="blue"))
 
 
 def get_shell_profile_path():
     if system == "Windows":
         obj = Terminal().run("$PROFILE", shell="pwsh")
         res = obj.op2path()
-        if isinstance(res, P): profile_path = res
+        if isinstance(res, PathExtended): profile_path = res
         else:
             obj.print(capture=False)
             raise ValueError(f"Could not get profile path for Windows. Got {res}")
-    elif system == "Linux": profile_path = P("~/.bashrc").expanduser()
+    elif system == "Linux": profile_path = PathExtended("~/.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
 
 
+def append_temporarily(dirs: list[str], kind: Literal['append', 'prefix', 'replace'] = "append"):
+    dirs_ = []
+    for path in dirs:
+        path_rel = PathExtended(path).collapseuser(strict=False)
+        if path_rel.as_posix() in PATH or str(path_rel) in PATH or str(path_rel.expanduser()) in PATH or path_rel.expanduser().as_posix() in PATH:
+            print(f"Path passed `{path}` is already in PATH, skipping the appending.")
+        else:
+            dirs_.append(path_rel.as_posix() if system == "Linux" else str(path_rel))
+    dirs = dirs_
+    if len(dirs) == 0: return ""
+
+    if system == "Windows":
+        """Source: https://docs.microsoft.com/en-us/powershell/module/microsoft.powershell.core/about/about_environment_variables?view=powershell-7.2"""
+        if kind == "append": command = fr'$env:Path += ";{sep.join(dirs)}"'  # Append to the Path variable in the current window:
+        elif kind == "prefix": command = fr'$env:Path = "{sep.join(dirs)};" + $env:Path'  # Prefix the Path variable in the current window:
+        elif kind == "replace": command = fr'$env:Path = "{sep.join(dirs)}"'  # Replace the Path variable in the current window (use with caution!):
+        else: raise KeyError
+        return command  # if run is False else tm.run(command, shell="powershell")
+    elif system in ["Linux", "Darwin"]: result = f'export PATH="{sep.join(dirs)}:$PATH"'
+    else: raise ValueError
+    return result
+
 def main_env_path(choice: Optional[str] = None, profile_path: Optional[str] = None):
     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 +94,22 @@ 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}")
-    addition = PathVar.append_temporarily(dirs=dirs)
-    profile_path_obj = P(profile_path) if isinstance(profile_path, str) else get_shell_profile_path()
+    console.print(f"\n📌 Adding directories to PATH: {dirs}")
+    addition = append_temporarily(dirs=dirs)
+    profile_path_obj = PathExtended(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_*")
-    profile_path_obj.modify_text(addition, addition, replace_line=False, notfound_append=True)
-    print("✅ PATH variables added to profile successfully")
+    console.print(f"💾 Created backup of profile: {profile_path_obj.name}.orig_*")
+    # Inline deprecated P.modify_text: if file missing, seed with search text before modification
+    current = profile_path_obj.read_text() if profile_path_obj.exists() else addition
+    updated = modify_text(current, addition, addition, replace_line=False, notfound_append=True)
+    profile_path_obj.write_text(updated)
+    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)
@@ -112,37 +122,33 @@ def main_add_sources_to_shell_profile(profile_path: Optional[str] = None, choice
     elif choice == "none(EXIT)": return
 
     if isinstance(profile_path, str):
-        profile_path_obj = P(profile_path)
+        profile_path_obj = PathExtended(profile_path)
     else: profile_path_obj = get_shell_profile_path()
     profile = profile_path_obj.read_text()
 
     for a_file in sources:
         tmp = a_file.replace("REPO_ROOT", REPO_ROOT.as_posix()).replace("LIBRARY_ROOT", LIBRARY_ROOT.as_posix())
-        file = P(tmp).collapseuser()  # this makes the shell profile interuseable across machines.
+        file = PathExtended(tmp).collapseuser()  # this makes the shell profile interuseable across machines.
         file = file.as_posix() if system == "Linux" else str(file)
         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,31 +156,31 @@ 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_path_obj = PathExtended(profile_path) if isinstance(profile_path, str) else get_shell_profile_path()
     profile = profile_path_obj.read_text()
 
     for patch_path in patches:
-        patch_path_obj = P(patch_path)
+        patch_path_obj = PathExtended(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/fire_agents

@@ -0,0 +1,27 @@
+#!/usr/bin/env bash
+
+. $HOME/scripts/activate_ve 've'
+
+# 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
+
+
+python -m machineconfig.scripts.python.fire_agents "$@"
+
+if [ -f "$op_script" ]; then
+  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.
+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/linux/wifi_conn

@@ -0,0 +1,24 @@
+#!/usr/bin/env bash
+
+# check below if VIRTUAL_ENV is set
+# if [ -z "$VIRTUAL_ENV" ]; then
+source $HOME/scripts/activate_ve 've'
+# fi
+
+op_script=$HOME/tmp_results/shells/python_return_command.sh
+if [ -f "$op_script" ]; then
+  rm $op_script
+fi
+
+python machineconfig.scripts.python.wifi_conn
+
+if [ -f "$op_script" ]; then
+  chmod +x $op_script
+  source $op_script
+fi
+
+
+# check below if VIRTUAL_ENV enviroment variable exist, then deactivate it, otherwise skip
+if [ -n "$VIRTUAL_ENV" ]; then
+  deactivate
+fi

machineconfig/scripts/python/archive/im2text.py

@@ -1,36 +1,34 @@
-"""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 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, PathExtended.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__':