import { useCallback, useEffect, useRef, useState } from "react"; import { useTranslation } from "react-i18next"; import { Button } from "@/shared/ui/button"; import { cancelDictationLocalModelDownload, deleteDictationLocalModel, downloadDictationLocalModel, getDictationLocalModelDownloadProgress, listDictationLocalModels, } from "@/shared/api/dictation"; type LocalModel = { id: string; description: string; sizeMb: number; downloaded: boolean; downloadInProgress: boolean; }; type DownloadProgress = { bytesDownloaded: number; totalBytes: number; progressPercent: number; status: string; error?: string | null; }; const POLL_INTERVAL_MS = 750; interface LocalWhisperModelsProps { selectedModelId: string; onSelectModel: (modelId: string) => void | Promise; onModelsChanged: () => void | Promise; } export function LocalWhisperModels({ selectedModelId, onSelectModel, onModelsChanged, }: LocalWhisperModelsProps) { const { t } = useTranslation(["settings", "common"]); const [models, setModels] = useState([]); const [loading, setLoading] = useState(true); const [error, setError] = useState(null); const [downloadingIds, setDownloadingIds] = useState>(new Set()); const [progresses, setProgresses] = useState>( new Map(), ); const onModelsChangedRef = useRef(onModelsChanged); onModelsChangedRef.current = onModelsChanged; const refresh = useCallback(async () => { try { const list = (await listDictationLocalModels()) as unknown as LocalModel[]; setModels(list); setDownloadingIds((prev) => { const next = new Set(prev); for (const m of list) { if (m.downloadInProgress) next.add(m.id); } return next; }); } catch (err) { setError( err instanceof Error ? err.message : t("general.voiceInput.loadError"), ); } }, [t]); useEffect(() => { const load = async () => { setLoading(true); setError(null); await refresh(); setLoading(false); }; void load(); }, [refresh]); useEffect(() => { if (downloadingIds.size === 0) return; let cancelled = false; const tick = async () => { const next = new Map(); const stillActive = new Set(); const finishedIds: string[] = []; for (const id of downloadingIds) { try { const progress = (await getDictationLocalModelDownloadProgress( id, )) as unknown as DownloadProgress | null; if (!progress) { finishedIds.push(id); continue; } next.set(id, progress); if (progress.status === "downloading") { stillActive.add(id); } else { finishedIds.push(id); } } catch { stillActive.add(id); } } if (cancelled) return; setProgresses(next); if (finishedIds.length > 0) { await refresh(); await onModelsChangedRef.current(); } setDownloadingIds(stillActive); }; const interval = window.setInterval(() => { void tick(); }, POLL_INTERVAL_MS); return () => { cancelled = true; window.clearInterval(interval); }; }, [downloadingIds, refresh]); const startDownload = useCallback( async (modelId: string) => { setError(null); try { await downloadDictationLocalModel(modelId); setDownloadingIds((prev) => new Set(prev).add(modelId)); } catch (err) { setError( err instanceof Error ? err.message : t("general.voiceInput.saveError"), ); } }, [t], ); const cancelDownload = useCallback( async (modelId: string) => { setError(null); try { await cancelDictationLocalModelDownload(modelId); } catch (err) { setError( err instanceof Error ? err.message : t("general.voiceInput.saveError"), ); } finally { setProgresses((prev) => { const next = new Map(prev); next.delete(modelId); return next; }); setDownloadingIds((prev) => { const next = new Set(prev); next.delete(modelId); return next; }); await refresh(); } }, [refresh, t], ); const deleteModel = useCallback( async (modelId: string) => { setError(null); try { await deleteDictationLocalModel(modelId); await refresh(); await onModelsChanged(); } catch (err) { setError( err instanceof Error ? err.message : t("general.voiceInput.deleteError"), ); } }, [onModelsChanged, refresh, t], ); if (loading) { return (

{t("common:labels.loading")}

); } if (models.length === 0) { return (

{t("general.voiceInput.noLocalModels")}

); } return (

{t("general.voiceInput.localModelLabel")}

{t("general.voiceInput.localModelDescription")}

{model.id}
{model.sizeMb} MB {isSelected ? ( {t("general.voiceInput.selectedModel")} ) : null}

{model.description}
{isDownloading && progress ? (

{t("general.voiceInput.downloadProgress", { percent: Math.round(progress.progressPercent), })}

) : null} {progress?.status === "failed" && progress.error ? (
{progress.error}
) : null}

{isDownloading ? ( ) : model.downloaded ? ( <> {!isSelected ? ( ) : null} ) : ( )}

{error ?

{error}

: null}

); }