diff --git a/TUI/allowlistselectionscreen.py b/TUI/allowlistselectionscreen.py index 10cf011..72426f2 100644 --- a/TUI/allowlistselectionscreen.py +++ b/TUI/allowlistselectionscreen.py @@ -567,7 +567,7 @@ class AllowlistSelectionWidget(Static): # Success notification self.app.notify( - f"✅ Successfully added {len(self.hashes_to_add)} hashes to '{allowlist_name}'", + f"✅ Successfully added {len(self.hashes_to_add)} hashes to '{allowlist_name}'", title="Success", severity="information", timeout=5, @@ -575,7 +575,7 @@ class AllowlistSelectionWidget(Static): # Update preview to show success self.preview_area.text = ( - f"## ✅ SUCCESS\n\n" + f"## ✅ SUCCESS\n\n" f"Added **{len(self.hashes_to_add)} hashes** to allowlist:\n" f"**{allowlist_name}** (ID: {app_id})\n\n" f"### Operation Details:\n" @@ -586,16 +586,13 @@ class AllowlistSelectionWidget(Static): ) # Change button to "Done" - self.add_btn.label = "✅ Done - Close" - self.add_btn.disabled = False - - # When clicked again, close the screen - self.add_btn_success = True + self.add_btn.label = "✅ Done" + self.add_btn.disabled = True except Exception as exc: logger.exception(f"Failed to add hashes to allowlist: {exc}") self.app.notify( - f"❌ Failed to add hashes: {str(exc)}", + f"❌ Failed to add hashes: {str(exc)}", title="Error", severity="error", timeout=10, @@ -603,7 +600,7 @@ class AllowlistSelectionWidget(Static): # Re-enable button self.add_btn.disabled = False - self.add_btn.label = "➕ Retry Add to Allowlist" + self.add_btn.label = "⟳ Retry Add to Allowlist" class AllowlistSelectionScreen(Screen):