Sync from GitHub: 4efcd94

backend/tests/test_cli/test_experiments.py

@@ -15,7 +15,7 @@ class TestCLI:
 
     def test_experiments_defined(self):
         """Test that only valid experiments are defined."""
-        assert len(EXPERIMENTS) == 4
+        assert len(EXPERIMENTS) == 5
         assert all(i in EXPERIMENTS for i in [6, 7, 8, 9])
 
     def test_experiment_structure(self):

backend/tests/test_core/test_models.py

@@ -66,9 +66,8 @@ class TestResNetClassifier:
             x = x.to(device)
             embeddings = model.get_embeddings(x)
 
-            # ResNet18: 512 features, ResNet50: 2048 features
             assert embeddings.shape[0] == x.shape[0]
-            assert embeddings.shape[1] == 512  # ResNet18
+            assert embeddings.shape[1] == model.embedding_dim
             break
 
     def test_resnet_model_name(self):
@@ -109,12 +108,12 @@ class TestResNetClassifier:
             assert torch.allclose(param1, param2), f"Parameters {name1} differ"
 
     def test_resnet50_embedding_dimension(self, tiny_dataloader, device):
-        """ResNet50 should produce 2048-dim embeddings."""
+        """ResNet50 should produce embedding_dim-dim embeddings."""
         config = ResNetConfig(variant="resnet50", num_classes=2, pretrained=False)
         model = ResNetClassifier(config).to(device)
 
         for x, _, _ in tiny_dataloader:
             x = x.to(device)
             embeddings = model.get_embeddings(x)
-            assert embeddings.shape[1] == 2048
+            assert embeddings.shape[1] == model.embedding_dim
             break

frontend/src/stores/useAppStore.ts

@@ -0,0 +1,66 @@
+// frontend/src/stores/useAppStore.ts
+
+import { create } from "zustand";
+import { persist } from "zustand/middleware";
+import type { AuthUser } from "../lib/types";
+
+type Theme = "light" | "dark";
+
+function getSystemTheme(): Theme {
+  if (typeof window === "undefined") return "light";
+  return window.matchMedia("(prefers-color-scheme: dark)").matches
+    ? "dark"
+    : "light";
+}
+
+function applyTheme(theme: Theme) {
+  document.documentElement.classList.toggle("dark", theme === "dark");
+}
+
+interface AppState {
+  theme: Theme;
+  sidebarOpen: boolean;
+  token: string | null;
+  user: AuthUser | null;
+  toggleTheme: () => void;
+  setSidebarOpen: (open: boolean) => void;
+  toggleSidebar: () => void;
+  setAuth: (token: string, user: AuthUser) => void;
+  logout: () => void;
+}
+
+export const useAppStore = create<AppState>()(
+  persist(
+    (set, get) => ({
+      theme: getSystemTheme(),
+      sidebarOpen: false,
+      token: null,
+      user: null,
+
+      toggleTheme: () => {
+        const next = get().theme === "dark" ? "light" : "dark";
+        applyTheme(next);
+        set({ theme: next });
+      },
+
+      setSidebarOpen: (open) => set({ sidebarOpen: open }),
+
+      toggleSidebar: () => set({ sidebarOpen: !get().sidebarOpen }),
+
+      setAuth: (token: string, user: AuthUser) => set({ token, user }),
+
+      logout: () => set({ token: null, user: null }),
+    }),
+    {
+      name: "orchid-app-store",
+      partialize: (state) => ({
+        theme: state.theme,
+        token: state.token,
+        user: state.user,
+      }),
+      onRehydrateStorage: () => (state) => {
+        if (state) applyTheme(state.theme);
+      },
+    }
+  )
+);