Add Enter key to exit TUI after render completes

Update footer to show render-finished keybindings and allow
exiting with Enter once rendering is done.

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

Author: tjazerzen

tui/components.py

@@ -14,16 +14,24 @@ class CustomFooter(Horizontal):
     """A custom footer with keyboard shortcuts and render ID."""
 
     FOOTER_TEXT = "ctrl+c: copy  *  ctrl+d: quit  *  ctrl+l: toggle logs"
+    RENDER_FINISHED_TEXT = "enter: exit  *  ctrl+c: copy  *  ctrl+l: toggle logs"
 
     def __init__(self, render_id: str = "", **kwargs):
         super().__init__(**kwargs)
         self.render_id = render_id
+        self._footer_text_widget: Optional[Static] = None
 
     def compose(self):
-        yield Static(self.FOOTER_TEXT, classes="custom-footer-text")
+        self._footer_text_widget = Static(self.FOOTER_TEXT, classes="custom-footer-text")
+        yield self._footer_text_widget
         if self.render_id:
             yield Static(f"render id: {self.render_id}", classes="custom-footer-render-id")
 
+    def show_render_finished(self) -> None:
+        """Update footer text to show render-finished keybindings."""
+        if self._footer_text_widget is not None:
+            self._footer_text_widget.update(self.RENDER_FINISHED_TEXT)
+
 
 class ScriptOutputType(str, Enum):
     UNIT_TEST_OUTPUT_TEXT = "Unit tests output: "

tui/plain2code_tui.py

@@ -57,6 +57,7 @@ class Plain2CodeTUI(App):
     BINDINGS = [
         Binding("ctrl+c", "copy_selection", "Copy", show=False),
         Binding("ctrl+d", "quit", "Quit", show=False),
+        Binding("enter", "enter_exit", "Exit", show=False),
         ("ctrl+l", "toggle_logs", "Toggle Logs"),
     ]
 
@@ -80,6 +81,7 @@ def __init__(
         self.conformance_tests_script: Optional[str] = conformance_tests_script
         self.prepare_environment_script: Optional[str] = prepare_environment_script
         self.state_machine_version = state_machine_version
+        self._render_finished = False
 
         # Initialize state handlers
         self._state_handlers: dict[str, StateHandler] = {
@@ -278,6 +280,12 @@ def _handle_frid_state(
     def on_render_completed(self, event: RenderCompleted):
         """Handle successful render completion."""
         self._render_success_handler.handle(event.rendered_code_path)
+        self._render_finished = True
+        try:
+            footer = self.screen.query_one(CustomFooter)
+            footer.show_render_finished()
+        except NoMatches:
+            pass
 
     def on_render_failed(self, event: RenderFailed):
         """Handle render failure."""
@@ -313,6 +321,11 @@ async def action_copy_selection(self) -> None:
             self.screen.clear_selection()
             self.notify("Copied to clipboard", timeout=2)
 
+    def action_enter_exit(self) -> None:
+        """Handle enter: exit the TUI only after rendering has finished."""
+        if self._render_finished:
+            self.action_quit()
+
     def action_quit(self) -> None:
         """Quit the application immediately.
 

tui/widget_helpers.py

@@ -105,7 +105,7 @@ def display_success_message(tui, rendered_code_path: str):
         rendered_code_path: The path to the rendered code
     """
 
-    message = f"[#79FC96]✓ Rendering finished![/#79FC96] [#888888](ctrl+d to exit)[/#888888]\n[#888888]Generated code: {rendered_code_path}[/#888888] "
+    message = f"[#79FC96]✓ Rendering finished![/#79FC96] [#888888](enter to exit)[/#888888]\n[#888888]Generated code: {rendered_code_path}[/#888888] "
 
     widget: Static = tui.query_one(f"#{TUIComponents.RENDER_STATUS_WIDGET.value}", Static)
     widget.update(message)