Update Correctly run UI

Assets/Scripts/UI/UIManager.cs

@@ -30,14 +30,7 @@ namespace Hallucinate.UI
         [SerializeField] private VisualTreeAsset settingsTemplate;
         [SerializeField] private VisualTreeAsset hudTemplate;
 
-        [Header("Debug Settings")]
-        [SerializeField] private bool showDebugInfo = true;
-
         private MainMenuController _mainMenuController;
-        private LobbyController _lobbyController;
-        private ProfileController _profileController;
-        private SettingsController _settingsController;
-        private HUDController _hudController;
 
         private void Awake()
         {
@@ -74,28 +67,25 @@ namespace Hallucinate.UI
                 _mainMenuController.SetGameIcon(gameIcon);
             }
 
-            _lobbyController = RegisterController<LobbyController>(lobbyTemplate);
-            _profileController = RegisterController<ProfileController>(profileTemplate);
-            _settingsController = RegisterController<SettingsController>(settingsTemplate);
-            _hudController = RegisterController<HUDController>(hudTemplate);
+            RegisterController<LobbyController>(lobbyTemplate);
+            RegisterController<ProfileController>(profileTemplate);
+            RegisterController<SettingsController>(settingsTemplate);
+            RegisterController<HUDController>(hudTemplate);
 
-            // Start with Main Menu
+            // Khởi động màn hình đầu tiên
             _ = Push<MainMenuController>();
         }
 
         private T RegisterController<T>(VisualTreeAsset template) where T : BaseUIController, new()
         {
-            if (template == null)
-            {
-                Debug.LogWarning($"Template for {typeof(T).Name} is missing!");
-                return null;
-            }
+            if (template == null) return null;
 
             var instance = template.Instantiate();
             instance.style.flexGrow = 1;
             instance.style.position = Position.Absolute;
             instance.style.width = Length.Percent(100);
             instance.style.height = Length.Percent(100);
+            instance.style.display = DisplayStyle.None; // Ẩn mặc định
             _rootElement.Add(instance);
 
             var controller = new T();
@@ -108,16 +98,15 @@ namespace Hallucinate.UI
         private void Update()
         {
             _mainMenuController?.Update();
-            _hudController?.Update();
+            // Update các controller khác nếu cần
         }
 
         public async Task Push<T>() where T : BaseUIController
         {
-            if (!_controllers.TryGetValue(typeof(T), out var newScreen))
-            {
-                Debug.LogError($"Controller of type {typeof(T)} not registered!");
-                return;
-            }
+            if (!_controllers.TryGetValue(typeof(T), out var newScreen)) return;
+
+            // Nếu màn hình mới chính là màn hình đang hiện, không làm gì cả
+            if (_history.Count > 0 && _history.Peek() == newScreen) return;
 
             if (_history.Count > 0)
             {
@@ -136,8 +125,11 @@ namespace Hallucinate.UI
             var currentScreen = _history.Pop();
             await currentScreen.PlayTransitionOut();
 
-            var previousScreen = _history.Peek();
-            await previousScreen.PlayTransitionIn();
+            if (_history.Count > 0)
+            {
+                var previousScreen = _history.Peek();
+                await previousScreen.PlayTransitionIn();
+            }
         }
     }
 }