gitarsenal-cli 1.4.7 → 1.4.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gitarsenal-cli",
-  "version": "1.4.7",
+  "version": "1.4.9",
   "description": "CLI tool for creating Modal sandboxes with GitHub repositories",
   "main": "index.js",
   "bin": {

package/python/test_modalSandboxScript.py

@@ -944,8 +944,8 @@ def create_modal_sandbox(gpu_type, repo_url=None, repo_name=None, setup_commands
         'T4': {'gpu': 'T4', 'memory': 16},
         'L4': {'gpu': 'L4', 'memory': 24},
         'A10G': {'gpu': 'A10G', 'memory': 24},
-        'A100-40GB': {'gpu': 'A100-SXM4-40GB', 'memory': 40},
-        'A100-80GB': {'gpu': 'A100-80GB', 'memory': 80},
+        'A100-40': {'gpu': 'A100-SXM4-40GB', 'memory': 40},
+        'A100-80': {'gpu': 'A100-80', 'memory': 80},
         'L40S': {'gpu': 'L40S', 'memory': 48},
         'H100': {'gpu': 'H100', 'memory': 80},
         'H200': {'gpu': 'H200', 'memory': 141},
@@ -4096,63 +4096,123 @@ def get_setup_commands_from_gitingest(repo_url):
 
 def prompt_for_gpu():
     """
-    Prompt the user to select a GPU type from available options.
+    Prompt the user to select a GPU type from available options using arrow keys.
     Returns the selected GPU type.
     """
+    import sys
+    import tty
+    import termios
+    
     # Define available GPU types and their specifications
     gpu_specs = {
         'T4': {'gpu': 'T4', 'memory': '16GB'},
         'L4': {'gpu': 'L4', 'memory': '24GB'},
         'A10G': {'gpu': 'A10G', 'memory': '24GB'},
-        'A100-40GB': {'gpu': 'A100-SXM4-40GB', 'memory': '40GB'},
-        'A100-80GB': {'gpu': 'A100-80GB', 'memory': '80GB'},
+        'A100-40': {'gpu': 'A100-40GB', 'memory': '40GB'},
+        'A100-80': {'gpu': 'A100-80GB', 'memory': '80GB'},
         'L40S': {'gpu': 'L40S', 'memory': '48GB'},
         'H100': {'gpu': 'H100', 'memory': '80GB'},
         'H200': {'gpu': 'H200', 'memory': '141GB'},
         'B200': {'gpu': 'B200', 'memory': '141GB'}
     }
     
-    print("\n📊 Available GPU Options:")
-    print("┌─────────────┬──────────┐")
-    print("│ GPU Type    │ Memory   │")
-    print("├─────────────┼──────────┤")
-    
-    # Create a list to keep track of valid options
-    options = []
+    # Create a list of options
+    options = list(gpu_specs.keys())
+    selected_index = 2  # Default to A10G (index 2)
     
-    # Display GPU options
-    for i, (gpu_type, specs) in enumerate(gpu_specs.items(), 1):
-        options.append(gpu_type)
-        print(f"│ {i}. {gpu_type:<8} │ {specs['memory']:<7} │")
+    def get_key():
+        """Get a single keypress from the user."""
+        fd = sys.stdin.fileno()
+        old_settings = termios.tcgetattr(fd)
+        try:
+            tty.setraw(sys.stdin.fileno())
+            ch = sys.stdin.read(1)
+            if ch == '\x1b':  # Escape sequence
+                ch2 = sys.stdin.read(1)
+                if ch2 == '[':
+                    ch3 = sys.stdin.read(1)
+                    if ch3 == 'A':
+                        return 'UP'
+                    elif ch3 == 'B':
+                        return 'DOWN'
+            elif ch == '\r' or ch == '\n':
+                return 'ENTER'
+            elif ch == '\x03':  # Ctrl+C
+                return 'CTRL_C'
+            return ch
+        finally:
+            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+    
+    def display_menu():
+        """Display the GPU selection menu with current selection highlighted."""
+        print("\n📊 Available GPU Options:")
+        print("┌──────────────┬─────────┐")
+        print("│ GPU Type     │ Memory  │")
+        print("├──────────────┼─────────┤")
+        
+        for i, gpu_type in enumerate(options):
+            specs = gpu_specs[gpu_type]
+            # Calculate proper spacing for alignment
+            number_part = f"{i+1}."
+            if i == selected_index:
+                prefix = "> "
+                suffix = " ←"
+            else:
+                prefix = "  "
+                suffix = ""
+            
+            # Ensure consistent width for GPU type column
+            gpu_display = f"{prefix}{number_part} {gpu_type}"
+            gpu_padded = f"{gpu_display:<12}"  # Fixed width for GPU column
+            
+            print(f"│ {gpu_padded} │ {specs['memory']:<7} │{suffix}")
+        
+        print("└──────────────┴─────────┘")
+        print("Use ↑/↓ arrows to select, Enter to confirm, Ctrl+C to cancel")
     
-    print("└─────────────┴──────────┘")
+    # Clear screen and show initial menu
+    print("\033[2J\033[H", end="")  # Clear screen and move cursor to top
+    display_menu()
     
-    # Prompt for selection
     while True:
         try:
-            choice = input("\n🔍 Select GPU type (number or name, default is A10G): ").strip()
-            
-            # Default to A10G if empty
-            if not choice:
-                return "A10G"
+            key = get_key()
             
-            # Check if input is a number
-            if choice.isdigit():
-                index = int(choice) - 1
-                if 0 <= index < len(options):
-                    return options[index]
-                else:
-                    print(f"❌ Invalid selection. Please enter a number between 1 and {len(options)}.")
-            # Check if input is a valid GPU type
-            elif choice in options:
-                return choice
-            else:
-                print(f"❌ Invalid GPU type. Please select from the available options.")
+            if key == 'UP':
+                selected_index = (selected_index - 1) % len(options)
+                print("\033[2J\033[H", end="")  # Clear screen
+                display_menu()
+            elif key == 'DOWN':
+                selected_index = (selected_index + 1) % len(options)
+                print("\033[2J\033[H", end="")  # Clear screen
+                display_menu()
+            elif key == 'ENTER':
+                selected_gpu = options[selected_index]
+                print(f"\n✅ Selected GPU: {selected_gpu}")
+                return selected_gpu
+            elif key == 'CTRL_C':
+                print("\n🛑 Selection cancelled.")
+                sys.exit(1)
+                
         except KeyboardInterrupt:
             print("\n🛑 Selection cancelled.")
             sys.exit(1)
         except Exception as e:
-            print(f"❌ Error: {e}")
+            print(f"\n❌ Error: {e}")
+            # Fall back to simple input method
+            try:
+                choice = input("\n🔍 Select GPU type (number or name, default is A10G): ").strip()
+                if not choice:
+                    return "A10G"
+                if choice.isdigit():
+                    index = int(choice) - 1
+                    if 0 <= index < len(options):
+                        return options[index]
+                elif choice in options:
+                    return choice
+                return "A10G"
+            except:
+                return "A10G"
 
 # Replace the existing GPU argument parsing in the main section
 if __name__ == "__main__":
@@ -4227,6 +4287,18 @@ if __name__ == "__main__":
     gpu_type = prompt_for_gpu()
     args.gpu = gpu_type
     
+    # Display configuration after GPU selection
+    print("\n📋 Container Configuration:")
+    print(f"Repository URL: {args.repo_url or 'Not specified'}")
+    print(f"GPU Type: {gpu_type}")
+    print(f"Volume: {args.volume_name or 'None'}")
+    if args.use_api:
+        print("Setup Commands: Auto-detect from repository")
+    elif args.setup_commands:
+        print(f"Setup Commands: {len(args.setup_commands)} custom commands")
+    else:
+        print("Setup Commands: Auto-detect from repository")
+    
     # Interactive mode or missing required arguments
     if args.interactive or not args.repo_url or not args.volume_name:
         # Get repository URL if not provided