Merge pull request #15 from gridfm/fix/interactive

add output cell

Author: etienne-v

gridfm_datakit/interactive.py

@@ -7,6 +7,9 @@
 from importlib import resources
 from ipyfilechooser import FileChooser
 
+# Run the generation function directly
+from gridfm_datakit.generate import generate_power_flow_data_distributed
+
 
 def get_available_load_profiles():
     """Get list of available aggregated load profiles."""
@@ -102,6 +105,17 @@ def interactive_interface():
     # Get available load profiles
     available_profiles = get_available_load_profiles()
 
+    # Dedicated output box
+    output_box = widgets.Output(
+        layout=widgets.Layout(
+            border="1px solid #bbb",
+            padding="10px",
+            margin="10px 0",
+            height="300px",
+            overflow="auto",
+        ),
+    )
+
     # Network Configuration
     global \
         network_source, \
@@ -303,7 +317,7 @@ def update_network_visibility(*args):
     )
 
     sigma = widgets.FloatSlider(
-        value=0.05,
+        value=0.2,
         min=0.0,
         max=0.3,
         step=0.01,
@@ -386,8 +400,8 @@ def update_network_visibility(*args):
         min=1,
         max=20,
         step=1,
-        description="Topology Variants per load scenario:",
-        style={"description_width": "150px"},
+        description="Top. Variants per load scenario:",
+        style={"description_width": "200px"},
         layout=widgets.Layout(width="550px"),
         readout_format="d",
     )
@@ -417,7 +431,7 @@ def update_network_visibility(*args):
         layout=widgets.Layout(width="550px", height="120px"),
     )
 
-    # Generation Perturbation Configuration
+    # Generation Configuration
 
     global gen_perturbation_type, gen_sigma
 
@@ -493,7 +507,7 @@ def update_network_visibility(*args):
     )
 
     large_chunk_size = widgets.IntSlider(
-        value=50,
+        value=200,
         min=10,
         max=500,
         step=10,
@@ -508,15 +522,15 @@ def update_network_visibility(*args):
     no_stats = widgets.Checkbox(
         value=False,
         description="Disable statistical calculations (faster)",
-        style={"description_width": "200px"},
-        layout=widgets.Layout(width="350px"),
+        style={"description_width": "100px"},
+        layout=widgets.Layout(width="500px"),
     )
 
     overwrite = widgets.Checkbox(
         value=True,
         description="Overwrite existing files (vs. append)",
-        style={"description_width": "200px"},
-        layout=widgets.Layout(width="350px"),
+        style={"description_width": "100px"},
+        layout=widgets.Layout(width="500px"),
     )
 
     # Set mode to "pf"
@@ -695,25 +709,48 @@ def update_advanced_load_scaling(*args):
 
     # Button to create YAML config only
     def save_config_only(b):
-        config = create_config()
-        config_path = Path(config_filename.value)
-        with open(config_path, "w") as f:
-            yaml.dump(config, f, default_flow_style=False)
-        print(f"YAML configuration saved to {config_path}")
+        with output_box:
+            output_box.clear_output()
+            try:
+                config = create_config()
+                config_path = Path(config_filename.value)
+                with open(config_path, "w") as f:
+                    yaml.dump(config, f, default_flow_style=False)
+                print(f"YAML configuration saved to {config_path.resolve()}")
+            except Exception as e:
+                print("Error while saving configuration:")
+                print(e)
 
     def save_and_run_config(b):
         """Save configuration and run the generation script."""
-        config = create_config()
-
-        # Save config to file
-        config_path = Path(config_filename.value)
-        with open(config_path, "w") as f:
-            yaml.dump(config, f, default_flow_style=False)
-
-        # Run the generation function directly
-        from gridfm_datakit.generate import generate_power_flow_data_distributed
-
-        generate_power_flow_data_distributed(str(config_path))
+        # Grey/disable both buttons immediately
+        run_button.description = "Running…"
+        run_button.disabled = True
+        save_config_button.disabled = True
+
+        with output_box:
+            output_box.clear_output()
+            try:
+                config = create_config()
+
+                # Save config to file
+                config_path = Path(config_filename.value)
+                with open(config_path, "w") as f:
+                    yaml.dump(config, f, default_flow_style=False)
+                print(f"\nConfiguration written to: {config_path.resolve()}\n")
+
+                # Run the generator (stdout/stderr will appear here)
+                print("Starting generation...")
+                generate_power_flow_data_distributed(str(config_path))
+                print("\nGeneration completed successfully.")
+            except Exception as e:
+                print("Error during generation:")
+                print(e)
+            finally:
+                # Re-enable buttons and restore label regardless of success/failure
+                run_button.description = "Generate and Run Configuration"
+                run_button.disabled = False
+                save_config_button.disabled = False
 
     save_config_button = widgets.Button(
         description="Create YAML Config",
@@ -739,5 +776,8 @@ def save_and_run_config(b):
     display(admittance_box)
     display(execution_box)
     display(config_filename)
-    display(save_config_button)
-    display(run_button)
+    display(save_config_button, run_button)
+
+    # Display the dedicated output area last so it is close to the buttons
+    display(widgets.HTML("<b>Output</b>"))
+    display(output_box)