|
| 1 | +/** |
| 2 | + * 本地 ASR 模型下载引导卡片 |
| 3 | + * |
| 4 | + * 首次进入课堂助手时检测本地 ASR 模型是否已下载， |
| 5 | + * 未下载则显示引导卡片供用户选择下载（可选增强，不阻断使用）。 |
| 6 | + * 用户关闭后记录 localStorage 不再重复弹出。 |
| 7 | + * |
| 8 | + * @ai-context: 可选增强组件：检测 local_asr_check_available IPC， |
| 9 | + * 未下载时展示模型选择与下载进度；关闭后持久化到 localStorage。 |
| 10 | + * 下载进度经 local_asr_download_progress 事件推送。 |
| 11 | + */ |
| 12 | +import { useState, useEffect, useCallback } from 'react'; |
| 13 | +import { Download, X, HardDrive, CheckCircle2, Loader2 } from 'lucide-react'; |
| 14 | +import { cn } from '@/lib/utils'; |
| 15 | + |
| 16 | +const DISMISS_KEY = 'kb_asr_model_prompt_dismissed'; |
| 17 | + |
| 18 | +interface AsrModelInfo { |
| 19 | + engine: string; |
| 20 | + id: string; |
| 21 | + label: string; |
| 22 | + description: string; |
| 23 | + size: string; |
| 24 | + ready: boolean; |
| 25 | +} |
| 26 | + |
| 27 | +interface DownloadProgress { |
| 28 | + engine: string; |
| 29 | + progress: number; |
| 30 | +} |
| 31 | + |
| 32 | +export function AsrModelPrompt() { |
| 33 | + const [visible, setVisible] = useState(false); |
| 34 | + const [models, setModels] = useState<AsrModelInfo[]>([]); |
| 35 | + const [downloading, setDownloading] = useState<string | null>(null); |
| 36 | + const [progress, setProgress] = useState(0); |
| 37 | + |
| 38 | + // 初始检测：是否已关闭过 / 模型是否已下载 |
| 39 | + useEffect(() => { |
| 40 | + if (localStorage.getItem(DISMISS_KEY)) return; |
| 41 | + if (!window.electronAPI) return; |
| 42 | + |
| 43 | + window.electronAPI.invoke('local_asr_get_models').then((result: unknown) => { |
| 44 | + const data = result as { models: AsrModelInfo[] }; |
| 45 | + if (data?.models) { |
| 46 | + setModels(data.models); |
| 47 | + // 至少有一个模型已下载则不再提示 |
| 48 | + const anyDownloaded = data.models.some((m) => m.ready); |
| 49 | + if (!anyDownloaded) setVisible(true); |
| 50 | + } |
| 51 | + }).catch(() => { /* sherpa-onnx 未安装时静默跳过 */ }); |
| 52 | + }, []); |
| 53 | + |
| 54 | + // 监听下载进度事件 |
| 55 | + useEffect(() => { |
| 56 | + if (!window.electronAPI) return; |
| 57 | + const off = window.electronAPI.on('local_asr_download_progress', (...args: unknown[]) => { |
| 58 | + const data = args[0] as DownloadProgress; |
| 59 | + setProgress(data.progress); |
| 60 | + if (data.progress >= 100) { |
| 61 | + setDownloading(null); |
| 62 | + setProgress(0); |
| 63 | + // 刷新模型状态 |
| 64 | + window.electronAPI.invoke('local_asr_get_models').then((result: unknown) => { |
| 65 | + const d = result as { models: AsrModelInfo[] }; |
| 66 | + if (d?.models) setModels(d.models); |
| 67 | + }).catch(() => {}); |
| 68 | + } |
| 69 | + }); |
| 70 | + return off; |
| 71 | + }, []); |
| 72 | + |
| 73 | + const handleDownload = useCallback(async (engine: string) => { |
| 74 | + if (!window.electronAPI || downloading) return; |
| 75 | + setDownloading(engine); |
| 76 | + setProgress(0); |
| 77 | + try { |
| 78 | + await window.electronAPI.invoke('local_asr_download_model', { engine, useMirror: true }); |
| 79 | + } catch (err) { |
| 80 | + console.warn('[AsrModelPrompt] 下载失败:', err); |
| 81 | + setDownloading(null); |
| 82 | + } |
| 83 | + }, [downloading]); |
| 84 | + |
| 85 | + const handleDismiss = useCallback(() => { |
| 86 | + setVisible(false); |
| 87 | + localStorage.setItem(DISMISS_KEY, '1'); |
| 88 | + }, []); |
| 89 | + |
| 90 | + if (!visible) return null; |
| 91 | + |
| 92 | + return ( |
| 93 | + <div className="mx-4 mt-3 p-4 rounded-kb-lg bg-bg-elevated border border-border/40 shadow-kb-sm"> |
| 94 | + <div className="flex items-start justify-between"> |
| 95 | + <div className="flex items-center gap-2"> |
| 96 | + <HardDrive className="w-4 h-4 text-brand-500" strokeWidth={1.5} /> |
| 97 | + <span className="text-b2 font-medium text-text-primary">下载本地语音识别模型</span> |
| 98 | + </div> |
| 99 | + <button |
| 100 | + onClick={handleDismiss} |
| 101 | + className="p-1 rounded-kb-sm text-text-tertiary hover:text-text-secondary hover:bg-bg-secondary transition-colors" |
| 102 | + aria-label="关闭提示" |
| 103 | + > |
| 104 | + <X className="w-3.5 h-3.5" strokeWidth={1.5} /> |
| 105 | + </button> |
| 106 | + </div> |
| 107 | + |
| 108 | + <p className="mt-1.5 text-c1 text-text-tertiary"> |
| 109 | + 下载后可完全离线转写语音，无需联网、零 API 费用。不下载也可正常使用（云端转写）。 |
| 110 | + </p> |
| 111 | + |
| 112 | + <div className="mt-3 space-y-2"> |
| 113 | + {models.map((model) => ( |
| 114 | + <div key={model.id} className="flex items-center gap-3 p-2.5 rounded-kb-md bg-bg-secondary/50"> |
| 115 | + <div className="flex-1 min-w-0"> |
| 116 | + <p className="text-b3 font-medium text-text-primary truncate">{model.label}</p> |
| 117 | + <p className="text-c1 text-text-tertiary mt-0.5">{model.description} · {model.size}</p> |
| 118 | + </div> |
| 119 | + {model.ready ? ( |
| 120 | + <span className="flex items-center gap-1 text-c1 text-emerald-600 flex-shrink-0"> |
| 121 | + <CheckCircle2 className="w-3.5 h-3.5" strokeWidth={1.5} /> |
| 122 | + 已下载 |
| 123 | + </span> |
| 124 | + ) : downloading === model.engine ? ( |
| 125 | + <span className="flex items-center gap-1.5 text-c1 text-brand-600 flex-shrink-0"> |
| 126 | + <Loader2 className="w-3.5 h-3.5 animate-spin" strokeWidth={1.5} /> |
| 127 | + {progress}% |
| 128 | + </span> |
| 129 | + ) : ( |
| 130 | + <button |
| 131 | + onClick={() => handleDownload(model.engine)} |
| 132 | + disabled={!!downloading} |
| 133 | + className={cn( |
| 134 | + 'flex items-center gap-1 px-2.5 py-1.5 rounded-kb-sm text-c1 font-medium transition-colors flex-shrink-0', |
| 135 | + downloading |
| 136 | + ? 'bg-bg-secondary text-text-tertiary cursor-not-allowed' |
| 137 | + : 'bg-brand-600 text-white hover:bg-brand-700', |
| 138 | + )} |
| 139 | + > |
| 140 | + <Download className="w-3 h-3" strokeWidth={1.5} /> |
| 141 | + 下载 |
| 142 | + </button> |
| 143 | + )} |
| 144 | + </div> |
| 145 | + ))} |
| 146 | + </div> |
| 147 | + |
| 148 | + {/* 下载进度条 */} |
| 149 | + {downloading && ( |
| 150 | + <div className="mt-3 h-1.5 rounded-full bg-bg-secondary overflow-hidden"> |
| 151 | + <div |
| 152 | + className="h-full rounded-full bg-brand-500 transition-all duration-300" |
| 153 | + style={{ width: `${progress}%` }} |
| 154 | + /> |
| 155 | + </div> |
| 156 | + )} |
| 157 | + </div> |
| 158 | + ); |
| 159 | +} |
0 commit comments