From 245a4e9ba221cb89be1e9882928175f619692a2c Mon Sep 17 00:00:00 2001
From: "gpt-engineer-app[bot]"
 <159125892+gpt-engineer-app[bot]@users.noreply.github.com>
Date: Fri, 15 May 2026 13:06:02 +0000
Subject: [PATCH] Changes

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

diff --git a/src/stores/useModelStore.ts b/src/stores/useModelStore.ts
index a5ac3f5..328ba54 100644
--- a/src/stores/useModelStore.ts
+++ b/src/stores/useModelStore.ts
@@ -244,6 +244,10 @@ export const useModelStore = create<ModelStore>((set, get) => ({
     models: state.models.map(m => m.id === id ? { ...m, locked: !m.locked } : m),
   })),
 
+  setModelColor: (id, color) => set((state) => ({
+    models: state.models.map(m => m.id === id ? { ...m, color } : m),
+  })),
+
   duplicateModel: (id) => {
     const state = get();
     if (state.models.length >= MAX_MODELS) return;