feat: Add "Reset" and "Quit" buttons to the TUI, enabling game resets and application termination.

Author: ondrej-lukas

agents/attackers/interactive_tui/interactive_tui.py

@@ -217,9 +217,13 @@ def compose(self) -> ComposeResult:
         yield Horizontal(
             Button("Take Action", variant="primary", id="act"),
             Button("Assist", variant="primary", id="assist"),
-            Button("Assist & Play", variant="warning", id="play"),
+            Button("Play", variant="warning", id="play"),
+            Button("Hack the Future", variant="error", id="hack"),
+            Vertical(classes="spacer"),
+            Button("Reset", id="reset"),
+            Button("Quit", id="quit"),
+            classes="buttons-row"
         )
-        yield Button("Hack the Future", variant="error", id="hack")
 
     @on(Select.Changed)
     def select_changed(self, event: Select.Changed) -> None:
@@ -402,6 +406,22 @@ async def do_ask_llm():
                         log.write(msg)
 
                 asyncio.create_task(do_ask_llm())
+        elif event.button.id == "quit":
+             if self.agent.socket:
+                self.agent.terminate_connection()
+             self.exit()
+        elif event.button.id == "reset":
+            log.write("Resetting the episode...")
+            obs = self.agent.request_game_reset()
+            if obs:
+                self.current_obs = obs
+                self._clear_state()
+                tree_state = self.query_one(TreeState)
+                tree = tree_state.children[0]
+                self.update_tree(tree)
+                log.write("Episode reset successfully.")
+            else:
+                 log.write("[bold red]Reset failed![/bold red]")
         else:
             if self.model is not None:
                 log.write(":hourglass: Waiting for the LLM...")

agents/attackers/interactive_tui/layout.tcss

@@ -39,4 +39,33 @@ Screen {
 
 #hack {
     margin: 0 1;
+}
+
+#reset {
+    background: darkcyan;
+    color: white; 
+}
+#reset:hover {
+    background: cyan;
+    color: black;
+}
+
+#quit {
+    background: indigo;
+    color: white;
+}
+#quit:hover {
+    background: purple;
+    color: white;
+}
+
+.spacer {
+    width: 1fr;
+    height: 1;
+}
+
+.buttons-row {
+    height: auto;
+    width: 100%;
+    column-span: 3;
 }