studio: catch KeyboardInterrupt in samples and format imports.

* Adds exception handler block to simulator samples to allow clean Ctrl+C exits.
* Formats import statements inside code generator validator.

PiperOrigin-RevId: 950821976
Change-Id: I283533d87619d5d9b54d9bec970cdefe538fd752

python/mujoco/experimental/studio/sample/ghost.py

@@ -184,9 +184,13 @@ def main(argv: list[str]) -> None:
     handle.send_to_viewer(messages.ModelEvent(model=model))
 
     step_control = sim.StepControl()
-    while handle.is_running():
+    try:
-      step_control.advance(model, data)
+      while handle.is_running():
-      model, data, step_control = handle.sync(model, data, step_control)
+        step_control.advance(model, data)
+        model, data, step_control = handle.sync(model, data, step_control)
+    except KeyboardInterrupt:
+      # Ctrl+C is the documented way to quit; exit cleanly, no traceback.
+      print('\nShutting down.', flush=True)
 
 
 if __name__ == '__main__':

python/mujoco/experimental/studio/sample/implot.py

@@ -237,9 +237,13 @@ def main(argv: list[str]) -> None:
     handle.send_to_viewer(messages.ModelEvent(model=model))
 
     step_control = sim.StepControl()
-    while handle.is_running():
+    try:
-      step_control.advance(model, data)
+      while handle.is_running():
-      model, data, step_control = handle.sync(model, data, step_control)
+        step_control.advance(model, data)
+        model, data, step_control = handle.sync(model, data, step_control)
+    except KeyboardInterrupt:
+      # Ctrl+C is the documented way to quit; exit cleanly, no traceback.
+      print('\nShutting down.', flush=True)
 
 
 if __name__ == '__main__':

python/mujoco/experimental/studio/viewer.py

@@ -61,9 +61,13 @@ def main(argv: list[str]) -> None:
 
     # Run the simulation.
     step_control = sim.StepControl()
-    while handle.is_running():
+    try:
-      step_control.advance(model, data)
+      while handle.is_running():
-      model, data, step_control = handle.sync(model, data, step_control)
+        step_control.advance(model, data)
+        model, data, step_control = handle.sync(model, data, step_control)
+    except KeyboardInterrupt:
+      # Ctrl+C is the documented way to quit; exit cleanly, no traceback.
+      print('\nShutting down.', flush=True)
 
 
 _app.run(main)