fix: persist Parakeet model selection across restarts

- Fix validation logic to check ParakeetManager for Parakeet models
- Previously only checked WhisperManager, causing valid Parakeet models to be cleared
- Remove redundant selectedModel state from useModelManagement hook
- Use settings.current_model as single source of truth for model selection
- Update OnboardingDesktop to use settings directly

Fixes model persistence issue where Parakeet models would reset to "none" on dev server restart

Author: moinulmoin

src-tauri/src/lib.rs

@@ -1628,23 +1628,45 @@ async fn perform_startup_checks(app: tauri::AppHandle) {
             }
         }
 
-        // Check current model is still available
+        // Check current model is still available based on engine type
         let mut _model_available = false;
         if let Some(current_model) = store
             .get("current_model")
             .and_then(|v| v.as_str().map(|s| s.to_string()))
         {
             if !current_model.is_empty() {
-                if let Some(whisper_manager) =
-                    app.try_state::<AsyncRwLock<whisper::manager::WhisperManager>>()
-                {
-                    let downloaded = whisper_manager.read().await.get_downloaded_model_names();
-                    _model_available = downloaded.contains(&current_model);
-                    if !_model_available {
-                        log::warn!("Current model '{}' no longer available", current_model);
-                        // Clear the selection
-                        store.set("current_model", serde_json::Value::String(String::new()));
-                        let _ = store.save();
+                // Get the engine type to determine which manager to check
+                let engine = store
+                    .get("current_model_engine")
+                    .and_then(|v| v.as_str().map(|s| s.to_string()))
+                    .unwrap_or_else(|| "whisper".to_string());
+
+                if engine == "parakeet" {
+                    // Check ParakeetManager for Parakeet models
+                    if let Some(parakeet_manager) = app.try_state::<parakeet::ParakeetManager>() {
+                        let models = parakeet_manager.list_models();
+                        _model_available = models.iter().any(|m| m.name == current_model && m.downloaded);
+                        if !_model_available {
+                            log::warn!("Current Parakeet model '{}' no longer available", current_model);
+                            // Clear the selection
+                            store.set("current_model", serde_json::Value::String(String::new()));
+                            store.set("current_model_engine", serde_json::Value::String("whisper".to_string()));
+                            let _ = store.save();
+                        }
+                    }
+                } else {
+                    // Check WhisperManager for Whisper models (default)
+                    if let Some(whisper_manager) =
+                        app.try_state::<AsyncRwLock<whisper::manager::WhisperManager>>()
+                    {
+                        let downloaded = whisper_manager.read().await.get_downloaded_model_names();
+                        _model_available = downloaded.contains(&current_model);
+                        if !_model_available {
+                            log::warn!("Current Whisper model '{}' no longer available", current_model);
+                            // Clear the selection
+                            store.set("current_model", serde_json::Value::String(String::new()));
+                            let _ = store.save();
+                        }
                     }
                 }
             }

src/components/onboarding/OnboardingDesktop.tsx

@@ -67,8 +67,6 @@ export const OnboardingDesktop = function OnboardingDesktop({
     modelOrder,
     downloadProgress,
     verifyingModels,
-    selectedModel,
-    setSelectedModel,
     loadModels,
     downloadModel,
     cancelDownload,
@@ -114,14 +112,14 @@ export const OnboardingDesktop = function OnboardingDesktop({
 
   useEffect(() => {
     // Only auto-select if no model is selected yet
-    if (!selectedModel) {
+    if (!settings?.current_model) {
       // Find a downloaded model
       const downloadedModel = Object.entries(models).find(([_, m]) => m.downloaded);
       if (downloadedModel) {
-        setSelectedModel(downloadedModel[0]);
+        updateSettings({ current_model: downloadedModel[0] });
       }
     }
-  }, [models]); // Only depend on models, not selectedModel to avoid loops
+  }, [models, settings?.current_model]); // Depend on both to react to changes
 
   const checkPermissions = async () => {
     // Use the hook methods to check permissions
@@ -184,7 +182,7 @@ export const OnboardingDesktop = function OnboardingDesktop({
 
       await updateSettings({
         hotkey: hotkey,
-        current_model: selectedModel || "",
+        current_model: settings?.current_model || "",
         current_model_engine: 'whisper',
         onboarding_completed: true
       });
@@ -242,7 +240,7 @@ export const OnboardingDesktop = function OnboardingDesktop({
       // automation check removed for now
       case "models":
         // User can proceed if they have selected a model that is downloaded
-        return selectedModel !== null && models[selectedModel]?.downloaded === true;
+        return settings?.current_model !== null && settings?.current_model !== "" && models[settings.current_model]?.downloaded === true;
       default:
         return true;
     }
@@ -448,9 +446,9 @@ export const OnboardingDesktop = function OnboardingDesktop({
                               model={model}
                               downloadProgress={progress}
                               isVerifying={verifyingModels.has(name)}
-                              isSelected={selectedModel === name}
+                              isSelected={settings?.current_model === name}
                               onDownload={downloadModel}
-                              onSelect={setSelectedModel}
+                              onSelect={(modelName) => updateSettings({ current_model: modelName })}
                               onCancelDownload={cancelDownload}
                               showSelectButton={model.downloaded}
                             />

src/hooks/useModelManagement.ts

@@ -48,7 +48,7 @@ export function useModelManagement(options: UseModelManagementOptions = {}) {
   const [downloadProgress, setDownloadProgress] = useState<Record<string, number>>({});
   const [verifyingModels, setVerifyingModels] = useState<Set<string>>(new Set());
 
-  const [selectedModel, setSelectedModel] = useState<string | null>(null);
+  // Removed selectedModel state - now using settings.current_model directly
   const [isLoading, setIsLoading] = useState(false);
 
   // Load models from backend
@@ -164,17 +164,14 @@ export function useModelManagement(options: UseModelManagementOptions = {}) {
       // Refresh model status
       await loadModels();
 
-      // If deleted model was the current one, clear selection
-      if (selectedModel === modelName) {
-        setSelectedModel(null);
-      }
+      // Model selection clearing is handled by the component via settings
     } catch (error) {
       console.error("Failed to delete model:", error);
       if (showToasts) {
         toast.error(`Failed to delete model: ${error}`);
       }
     }
-  }, [selectedModel, loadModels, showToasts]);
+  }, [loadModels, showToasts]);
 
   // Setup event listeners BEFORE any other effects
   useEffect(() => {
@@ -319,11 +316,9 @@ export function useModelManagement(options: UseModelManagementOptions = {}) {
     modelOrder,
     downloadProgress,
     verifyingModels,
-    selectedModel,
     isLoading,
 
     // Actions
-    setSelectedModel,
     loadModels,
     downloadModel,
     cancelDownload,