From 7048cb765dc705c582fedd44a38691c05573020d Mon Sep 17 00:00:00 2001
From: rambros <rambrosteam@gmail.com>
Date: Mon, 16 Mar 2026 00:50:47 +0530
Subject: [PATCH] improve visual feedback after operation completion

---
 src/ui/modals.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/ui/modals.py b/src/ui/modals.py
index 56a9446..7317069 100644
--- a/src/ui/modals.py
+++ b/src/ui/modals.py
@@ -403,6 +403,15 @@ class ProgressScreen(Screen[None]):
             back_btn.display = show_back
         except Exception:
             pass
+            
+        try:
+            menu_btn = self.query_one("#btn_main_menu", Button)
+            if status == "complete":
+                menu_btn.variant = "success"
+            else:
+                menu_btn.variant = "warning"
+        except Exception:
+            pass
 
     def show_info(self, migration_status: str, items_status: str):
         try: