Fixed Breaking Legacy change

widgets/agentmoveoperations.py

@@ -506,6 +506,7 @@ class AgentMoveOperations(Widget):
         unsuccessful = []
         status_label = self.query_one("#status_label", Static)
         status_label.update("Exporting CSV...")
+        self.app.refresh_data()
         agents = self.agents
         policies = self.app.policies
         path = self.app.working_dir
@@ -604,6 +605,8 @@ class AgentMoveOperations(Widget):
 
                     successful.append((agent, f"Moved to {mode}: {result}"))
                     logger.info(f"Successfully toggled {agent.hostname} to {mode}")
+                    self.app.refresh_data()
+
                 except Exception as e:
                     unsuccessful.append((agent, str(e)))
                     logger.error(f"Failed to toggle {agent.hostname}: {e}")
@@ -728,7 +731,7 @@ class AgentMoveOperations(Widget):
             status_label.update(f"Error: {str(e)}")
             self.operation_in_progress = False
             return
-
+        self.app.refresh_data()
         self.operation_in_progress = False
         status_label.update("Operation complete!")
 

widgets/multiagentselector.py

@@ -80,11 +80,11 @@ class MultiAgentSelector(Widget):
                     select_buttons.styles.margin = (0, 0, 0, 0)
 
                     select_none_button = Button("🚫 Select None", id="select_none")
-                    select_none_button.styles.margin = (1, 0, 0, 1)
+                    select_none_button.styles.margin = (1, 1, 0, 1)
                     yield select_none_button
 
                     select_all_button = Button("✅ Select All", id="select_all")
-                    select_all_button.styles.margin = (1, 1, 0, 1)
+                    select_all_button.styles.margin = (1, 0, 0, 1)
                     yield select_all_button
 
                 with Horizontal() as button_row:
@@ -93,6 +93,7 @@ class MultiAgentSelector(Widget):
 
                     back_button = Button("← Back", id="back_button")
                     back_button.styles.width = "1fr"
+                    back_button.styles.margin = (0, 0, 0, 1)
                     yield back_button
 
                     submit_button = Button(