Fix widget UI issues and add custom filename support

Issues Fixed:
- #30: Add proper sizing/scrolling to status output (150px height, scrollable)
- #31: Remove inappropriate auto-import message
- #32: Add custom filename input for save/load operations

Key Changes:
- Status output now has fixed height with scroll and border for visibility
- Removed print message from load_ipython_extension since widget auto-displays
- Added save_filename_input text widget for custom file names
- Added file extension validation (.yml, .yaml, .json)
- Updated save logic to use custom filename instead of fixed clustrix.yml
- Added existing files dropdown for quick selection
- Fixed related tests for new behavior

Features:
- Custom filename: Users can specify any .yml/.yaml/.json filename
- File validation: Prevents invalid extensions
- Quick selection: Dropdown for existing files fills filename input
- Better visibility: Status messages now properly visible with scrolling

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: jeremymanning

clustrix/notebook_magic.py

@@ -363,8 +363,15 @@ def _create_widgets(self):
             layout=widgets.Layout(width="auto"),
         )
         self.delete_btn.on_click(self._on_delete_config)
-        # Status output
-        self.status_output = widgets.Output()
+        # Status output with proper sizing
+        self.status_output = widgets.Output(
+            layout=widgets.Layout(
+                height="150px",
+                width="100%",
+                overflow_y="scroll",
+                border="1px solid #ccc",
+            )
+        )
         # Update dropdown and load initial configuration
         self._update_config_dropdown()
         if self.configs:
@@ -554,17 +561,28 @@ def _create_advanced_options(self):
     def _create_save_section(self):
         """Create save configuration section."""
         style = {"description_width": "120px"}
-        # File selection dropdown
-        file_options = ["New file: clustrix.yml"]
+
+        # Custom filename input
+        self.save_filename_input = widgets.Text(
+            value="clustrix.yml",
+            description="Filename:",
+            placeholder="config.yml or config.json",
+            style=style,
+            layout=widgets.Layout(width="70%"),
+        )
+
+        # Existing files dropdown (optional)
+        file_options = ["(Create new file)"]
         for config_file in self.config_files:
-            file_options.append(f"Existing: {config_file}")
+            file_options.append(f"Overwrite: {config_file}")
         self.save_file_dropdown = widgets.Dropdown(
             options=file_options,
             value=file_options[0],
-            description="Save to:",
+            description="Or select:",
             style=style,
             layout=widgets.Layout(width="70%"),
         )
+        self.save_file_dropdown.observe(self._on_save_file_select, names="value")
         # Save button
         self.save_btn = widgets.Button(
             description="Save Configuration",
@@ -627,6 +645,14 @@ def _setup_change_tracking(self):
         for field in fields_to_track:
             field.observe(self._mark_unsaved_changes, names="value")
 
+    def _on_save_file_select(self, change):
+        """Handle selection from existing files dropdown."""
+        selected = change["new"]
+        if selected.startswith("Overwrite: "):
+            # Extract filename from "Overwrite: /path/to/file"
+            file_path = selected.replace("Overwrite: ", "")
+            self.save_filename_input.value = str(Path(file_path).name)
+
     def _create_section_containers(self):
         """Create the main UI section containers."""
         # Connection fields (dynamically shown/hidden)
@@ -909,13 +935,18 @@ def _on_save_config(self, button):
 
                 self.configs[new_config_name] = config
                 self.current_config_name = new_config_name
-                # Determine save file
-                save_option = self.save_file_dropdown.value
-                if save_option.startswith("New file:"):
-                    save_path = Path("clustrix.yml")
-                else:
-                    # Extract path from "Existing: /path/to/file"
-                    save_path = Path(save_option.split("Existing: ", 1)[1])
+                # Determine save file from custom filename input
+                filename = self.save_filename_input.value.strip()
+                if not filename:
+                    print("❌ Please enter a filename")
+                    return
+
+                # Validate file extension
+                if not filename.lower().endswith((".yml", ".yaml", ".json")):
+                    print("❌ Filename must end with .yml, .yaml, or .json")
+                    return
+
+                save_path = Path(filename)
                 # Load existing configs if updating a file
                 if save_path.exists():
                     existing_configs = load_config_from_file(save_path)
@@ -1071,6 +1102,8 @@ def display(self):
         save_section = widgets.VBox(
             [
                 widgets.HTML("<h5>Configuration Management</h5>"),
+                widgets.HTML("<h6>Save Configuration</h6>"),
+                self.save_filename_input,
                 widgets.HBox([self.save_file_dropdown, self.save_btn]),
                 widgets.HTML("<br><h6>Load Configuration</h6>"),
                 widgets.HBox([self.load_file_upload, self.load_btn]),
@@ -1163,9 +1196,7 @@ def load_ipython_extension(ipython):
         ipython.register_magic_function(
             ClusterfyMagics(ipython).clusterfy, "cell", "clusterfy"
         )
-        print(
-            "Clustrix notebook magic loaded. Use %%clusterfy to manage configurations."
-        )
+        # Note: No print message since widget displays automatically on import
 
 
 # Export the widget class for testing

tests/test_notebook_magic.py

@@ -507,12 +507,10 @@ def test_load_ipython_extension(self, mock_ipython_environment):
         with patch("clustrix.notebook_magic.ClusterfyMagics") as MockMagics:
             mock_magic_instance = MagicMock()
             MockMagics.return_value = mock_magic_instance
-            with patch("builtins.print") as mock_print:
-                clustrix.notebook_magic.load_ipython_extension(mock_ipython)
-                MockMagics.assert_called_once_with(mock_ipython)
-                mock_ipython.register_magic_function.assert_called_once()
-                mock_print.assert_called_once()
-                assert "Clustrix notebook magic loaded" in mock_print.call_args[0][0]
+            clustrix.notebook_magic.load_ipython_extension(mock_ipython)
+            MockMagics.assert_called_once_with(mock_ipython)
+            mock_ipython.register_magic_function.assert_called_once()
+            # Note: No print message expected since widget displays automatically
 
     def test_clusterfy_magic_without_ipython(self):
         """Test magic command fails gracefully without IPython."""
@@ -595,6 +593,9 @@ def test_save_load_cycle(self, mock_ipython_environment):
             widget.module_loads.value = ""
             widget.pre_exec_commands = MagicMock()
             widget.pre_exec_commands.value = ""
+            # Mock the new filename input field
+            widget.save_filename_input = MagicMock()
+            widget.save_filename_input.value = "clustrix.yml"
             # Change current directory for the test
             import os