From 197ea1faad36aa3fecb084fcd8dcf4831570b545 Mon Sep 17 00:00:00 2001
From: "gpt-engineer-app[bot]"
 <159125892+gpt-engineer-app[bot]@users.noreply.github.com>
Date: Thu, 21 May 2026 14:50:19 +0000
Subject: [PATCH] Changes

Co-authored-by: Reifonas <211114984+Reifonas@users.noreply.github.com>
---
 src/stores/useModelStore.ts | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/stores/useModelStore.ts b/src/stores/useModelStore.ts
index 5589f7f..8eeae17 100644
--- a/src/stores/useModelStore.ts
+++ b/src/stores/useModelStore.ts
@@ -348,7 +348,16 @@ export const useModelStore = create<ModelStore>((set, get) => ({
 
   setActiveModel: (id) => {
     const state = get();
-    set({ activeModelId: id, ...syncActive({ models: state.models, activeModelId: id }) });
+    const active = state.models.find(m => m.id === id);
+    const vis = active ? loadVisibility(active.fileName) : { hidden: new Set<string>(), isolated: null };
+    set({
+      activeModelId: id,
+      ...syncActive({ models: state.models, activeModelId: id }),
+      hiddenElementKeys: vis.hidden,
+      isolatedElementKeys: vis.isolated,
+      selectedElementKeys: new Set(),
+      visibilityNonce: state.visibilityNonce + 1,
+    });
   },
 
   toggleModelVisible: (id) => set((state) => ({