feat: 添加模型配置页面，支持用户自定义模型路径
- 新增设置弹窗界面，支持添加/删除/选择模型 - 支持模型文件路径检查 - 支持设置默认模型和输出模式 - 配置文件保存到 Electron userData 目录 - 更新 preload.ts 添加 checkFileExists API - 更新 electron-main.ts 添加配置文件读写功能
This commit is contained in:
impressionyang
parent
83e3084233
commit
98105d67ed
@ -7,12 +7,51 @@ import { app, BrowserWindow, ipcMain, globalShortcut, clipboard } from 'electron
|
||||
import { join, dirname } from 'path';
|
||||
import { fileURLToPath } from 'url';
|
||||
import { logger } from './utils/logger.js';
|
||||
import { existsSync, readFileSync, writeFileSync, mkdirSync } from 'fs';
|
||||
|
||||
const __filename = fileURLToPath(import.meta.url);
|
||||
const __dirname = dirname(__filename);
|
||||
|
||||
let mainWindow: BrowserWindow | null = null;
|
||||
|
||||
// 配置文件路径
|
||||
const CONFIG_DIR = join(app.getPath('userData'), 'config');
|
||||
const SETTINGS_FILE = join(CONFIG_DIR, 'settings.json');
|
||||
|
||||
// 读取设置
|
||||
function readSettings(): Record<string, unknown> {
|
||||
try {
|
||||
if (existsSync(SETTINGS_FILE)) {
|
||||
const content = readFileSync(SETTINGS_FILE, 'utf-8');
|
||||
return JSON.parse(content);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error('读取设置失败', error);
|
||||
}
|
||||
return {
|
||||
language: 'zh',
|
||||
outputMode: 'clipboard',
|
||||
models: [],
|
||||
defaultModel: null as number | null,
|
||||
defaultOutputMode: 'clipboard',
|
||||
};
|
||||
}
|
||||
|
||||
// 保存设置
|
||||
function writeSettings(settings: Record<string, unknown>): boolean {
|
||||
try {
|
||||
if (!existsSync(CONFIG_DIR)) {
|
||||
mkdirSync(CONFIG_DIR, { recursive: true });
|
||||
}
|
||||
writeFileSync(SETTINGS_FILE, JSON.stringify(settings, null, 2), 'utf-8');
|
||||
logger.info('设置已保存', { file: SETTINGS_FILE });
|
||||
return true;
|
||||
} catch (error) {
|
||||
logger.error('保存设置失败', error);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
function createWindow() {
|
||||
logger.info('创建主窗口...');
|
||||
|
||||
@ -90,18 +129,26 @@ ipcMain.handle('copy-to-clipboard', async (_, text: string) => {
|
||||
return { success: true };
|
||||
});
|
||||
|
||||
ipcMain.handle('check-file-exists', async (_, filePath: string) => {
|
||||
try {
|
||||
const exists = existsSync(filePath);
|
||||
logger.info(`IPC: 检查文件存在`, { path: filePath, exists });
|
||||
return exists;
|
||||
} catch (error) {
|
||||
logger.error(`IPC: 检查文件失败`, { path: filePath, error });
|
||||
return false;
|
||||
}
|
||||
});
|
||||
|
||||
ipcMain.handle('get-settings', async () => {
|
||||
logger.info('IPC: 获取设置');
|
||||
return {
|
||||
language: 'zh',
|
||||
outputMode: 'clipboard',
|
||||
modelPath: './models/model.onnx',
|
||||
};
|
||||
return readSettings();
|
||||
});
|
||||
|
||||
ipcMain.handle('save-settings', async (_event: any, settings: Record<string, unknown>) => {
|
||||
logger.info('IPC: 保存设置', settings);
|
||||
return { success: true };
|
||||
const success = writeSettings(settings);
|
||||
return { success };
|
||||
});
|
||||
|
||||
// 所有窗口关闭时退出应用
|
||||
|
||||
@ -19,6 +19,9 @@ contextBridge.exposeInMainWorld('electronAPI', {
|
||||
saveSettings: (settings: Record<string, unknown>) =>
|
||||
ipcRenderer.invoke('save-settings', settings),
|
||||
|
||||
// 文件检查
|
||||
checkFileExists: (path: string) => ipcRenderer.invoke('check-file-exists', path),
|
||||
|
||||
// 事件监听
|
||||
onToggleRecording: (callback: () => void) => {
|
||||
ipcRenderer.on('toggle-recording', () => callback());
|
||||
|
||||
@ -36,6 +36,13 @@
|
||||
padding: 20px;
|
||||
text-align: center;
|
||||
border-bottom: 1px solid var(--border);
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
}
|
||||
|
||||
.header-left {
|
||||
text-align: left;
|
||||
}
|
||||
|
||||
.header h1 {
|
||||
@ -49,6 +56,21 @@
|
||||
margin-top: 4px;
|
||||
}
|
||||
|
||||
.settings-btn {
|
||||
background: var(--bg-secondary);
|
||||
border: 1px solid var(--border);
|
||||
border-radius: 8px;
|
||||
padding: 8px 12px;
|
||||
color: var(--text-primary);
|
||||
cursor: pointer;
|
||||
font-size: 18px;
|
||||
transition: all 0.2s ease;
|
||||
}
|
||||
|
||||
.settings-btn:hover {
|
||||
background: var(--border);
|
||||
}
|
||||
|
||||
.main-content {
|
||||
flex: 1;
|
||||
padding: 20px;
|
||||
@ -144,7 +166,7 @@
|
||||
font-style: italic;
|
||||
}
|
||||
|
||||
.settings-section {
|
||||
.quick-settings {
|
||||
background: var(--bg-secondary);
|
||||
border-radius: 12px;
|
||||
padding: 16px;
|
||||
@ -192,13 +214,236 @@
|
||||
border: 1px solid var(--border);
|
||||
font-family: monospace;
|
||||
}
|
||||
|
||||
/* 配置弹窗 */
|
||||
.modal-overlay {
|
||||
position: fixed;
|
||||
top: 0;
|
||||
left: 0;
|
||||
right: 0;
|
||||
bottom: 0;
|
||||
background: rgba(0, 0, 0, 0.7);
|
||||
display: none;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
z-index: 1000;
|
||||
}
|
||||
|
||||
.modal-overlay.active {
|
||||
display: flex;
|
||||
}
|
||||
|
||||
.modal {
|
||||
background: var(--bg-secondary);
|
||||
border-radius: 12px;
|
||||
padding: 24px;
|
||||
width: 90%;
|
||||
max-width: 500px;
|
||||
max-height: 80vh;
|
||||
overflow-y: auto;
|
||||
}
|
||||
|
||||
.modal-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-bottom: 20px;
|
||||
padding-bottom: 12px;
|
||||
border-bottom: 1px solid var(--border);
|
||||
}
|
||||
|
||||
.modal-header h2 {
|
||||
font-size: 18px;
|
||||
font-weight: 600;
|
||||
}
|
||||
|
||||
.modal-close {
|
||||
background: transparent;
|
||||
border: none;
|
||||
color: var(--text-secondary);
|
||||
font-size: 24px;
|
||||
cursor: pointer;
|
||||
padding: 0;
|
||||
width: 32px;
|
||||
height: 32px;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
border-radius: 4px;
|
||||
}
|
||||
|
||||
.modal-close:hover {
|
||||
background: var(--border);
|
||||
color: var(--text-primary);
|
||||
}
|
||||
|
||||
.modal-section {
|
||||
margin-bottom: 20px;
|
||||
}
|
||||
|
||||
.modal-section h3 {
|
||||
font-size: 14px;
|
||||
color: var(--text-secondary);
|
||||
margin-bottom: 12px;
|
||||
}
|
||||
|
||||
.model-item {
|
||||
background: var(--bg-primary);
|
||||
border-radius: 8px;
|
||||
padding: 12px;
|
||||
margin-bottom: 8px;
|
||||
}
|
||||
|
||||
.model-item-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-bottom: 8px;
|
||||
}
|
||||
|
||||
.model-item-name {
|
||||
font-weight: 600;
|
||||
font-size: 14px;
|
||||
}
|
||||
|
||||
.model-item-status {
|
||||
font-size: 12px;
|
||||
padding: 2px 8px;
|
||||
border-radius: 4px;
|
||||
}
|
||||
|
||||
.model-item-status.loaded {
|
||||
background: rgba(0, 217, 165, 0.2);
|
||||
color: var(--success);
|
||||
}
|
||||
|
||||
.model-item-status.missing {
|
||||
background: rgba(233, 69, 96, 0.2);
|
||||
color: var(--accent);
|
||||
}
|
||||
|
||||
.model-item-path {
|
||||
font-size: 12px;
|
||||
color: var(--text-secondary);
|
||||
word-break: break-all;
|
||||
margin-bottom: 8px;
|
||||
}
|
||||
|
||||
.model-item-actions {
|
||||
display: flex;
|
||||
gap: 8px;
|
||||
}
|
||||
|
||||
.btn {
|
||||
padding: 6px 12px;
|
||||
border-radius: 6px;
|
||||
font-size: 12px;
|
||||
cursor: pointer;
|
||||
border: none;
|
||||
transition: all 0.2s ease;
|
||||
}
|
||||
|
||||
.btn-primary {
|
||||
background: var(--accent);
|
||||
color: white;
|
||||
}
|
||||
|
||||
.btn-primary:hover {
|
||||
background: var(--accent-hover);
|
||||
}
|
||||
|
||||
.btn-secondary {
|
||||
background: var(--bg-primary);
|
||||
color: var(--text-primary);
|
||||
border: 1px solid var(--border);
|
||||
}
|
||||
|
||||
.btn-secondary:hover {
|
||||
background: var(--border);
|
||||
}
|
||||
|
||||
.btn-danger {
|
||||
background: rgba(233, 69, 96, 0.2);
|
||||
color: var(--accent);
|
||||
border: 1px solid var(--accent);
|
||||
}
|
||||
|
||||
.btn-danger:hover {
|
||||
background: var(--accent);
|
||||
color: white;
|
||||
}
|
||||
|
||||
.add-model-form {
|
||||
background: var(--bg-primary);
|
||||
border-radius: 8px;
|
||||
padding: 12px;
|
||||
margin-top: 12px;
|
||||
}
|
||||
|
||||
.form-group {
|
||||
margin-bottom: 12px;
|
||||
}
|
||||
|
||||
.form-group label {
|
||||
display: block;
|
||||
font-size: 12px;
|
||||
color: var(--text-secondary);
|
||||
margin-bottom: 4px;
|
||||
}
|
||||
|
||||
.form-group input,
|
||||
.form-group select {
|
||||
width: 100%;
|
||||
padding: 8px 12px;
|
||||
border-radius: 6px;
|
||||
border: 1px solid var(--border);
|
||||
background: var(--bg-secondary);
|
||||
color: var(--text-primary);
|
||||
font-size: 13px;
|
||||
}
|
||||
|
||||
.form-group input:focus,
|
||||
.form-group select:focus {
|
||||
outline: none;
|
||||
border-color: var(--accent);
|
||||
}
|
||||
|
||||
.toast {
|
||||
position: fixed;
|
||||
bottom: 20px;
|
||||
left: 50%;
|
||||
transform: translateX(-50%);
|
||||
background: var(--bg-secondary);
|
||||
border: 1px solid var(--border);
|
||||
border-radius: 8px;
|
||||
padding: 12px 24px;
|
||||
box-shadow: 0 4px 12px rgba(0, 0, 0, 0.3);
|
||||
z-index: 1001;
|
||||
opacity: 0;
|
||||
transition: opacity 0.3s ease;
|
||||
}
|
||||
|
||||
.toast.show {
|
||||
opacity: 1;
|
||||
}
|
||||
|
||||
.toast.success {
|
||||
border-color: var(--success);
|
||||
}
|
||||
|
||||
.toast.error {
|
||||
border-color: var(--accent);
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="header">
|
||||
<div class="header-left">
|
||||
<h1>🎤 Impress ASR Input</h1>
|
||||
<p>语音识别输入工具</p>
|
||||
</div>
|
||||
<button class="settings-btn" id="settingsBtn" title="配置">⚙️</button>
|
||||
</div>
|
||||
|
||||
<div class="main-content">
|
||||
<div class="status-card">
|
||||
@ -213,7 +458,7 @@
|
||||
<div class="result-text" id="resultText"></div>
|
||||
</div>
|
||||
|
||||
<div class="settings-section">
|
||||
<div class="quick-settings">
|
||||
<div class="setting-item">
|
||||
<label>识别语言</label>
|
||||
<select id="languageSelect">
|
||||
@ -238,6 +483,82 @@
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- 配置弹窗 -->
|
||||
<div class="modal-overlay" id="settingsModal">
|
||||
<div class="modal">
|
||||
<div class="modal-header">
|
||||
<h2>⚙️ 设置</h2>
|
||||
<button class="modal-close" id="modalClose">×</button>
|
||||
</div>
|
||||
|
||||
<div class="modal-section">
|
||||
<h3>📦 模型配置</h3>
|
||||
<div id="modelList">
|
||||
<!-- 模型列表动态生成 -->
|
||||
</div>
|
||||
<button class="btn btn-secondary" id="addModelBtn" style="width: 100%; margin-top: 8px;">
|
||||
+ 添加模型
|
||||
</button>
|
||||
<div class="add-model-form" id="addModelForm" style="display: none;">
|
||||
<div class="form-group">
|
||||
<label>模型名称</label>
|
||||
<input type="text" id="modelNameInput" placeholder="例如：SenseVoice 中文">
|
||||
</div>
|
||||
<div class="form-group">
|
||||
<label>模型类型</label>
|
||||
<select id="modelTypeSelect">
|
||||
<option value="sensevoice">SenseVoice</option>
|
||||
<option value="whisper">Whisper</option>
|
||||
<option value="paraformer">Paraformer</option>
|
||||
</select>
|
||||
</div>
|
||||
<div class="form-group">
|
||||
<label>模型文件路径</label>
|
||||
<input type="text" id="modelPathInput" placeholder="C:\models\sensevoice.onnx">
|
||||
</div>
|
||||
<div class="form-group">
|
||||
<label>识别语言</label>
|
||||
<select id="modelLanguageSelect">
|
||||
<option value="zh">中文</option>
|
||||
<option value="en">English</option>
|
||||
<option value="ja">日本語</option>
|
||||
<option value="ko">한국어</option>
|
||||
<option value="auto">自动检测</option>
|
||||
</select>
|
||||
</div>
|
||||
<div style="display: flex; gap: 8px;">
|
||||
<button class="btn btn-primary" id="confirmAddModelBtn" style="flex: 1;">添加</button>
|
||||
<button class="btn btn-secondary" id="cancelAddModelBtn" style="flex: 1;">取消</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div class="modal-section">
|
||||
<h3>🎯 默认设置</h3>
|
||||
<div class="form-group">
|
||||
<label>默认模型</label>
|
||||
<select id="defaultModelSelect">
|
||||
<option value="">-- 选择默认模型 --</option>
|
||||
</select>
|
||||
</div>
|
||||
<div class="form-group">
|
||||
<label>默认输出模式</label>
|
||||
<select id="defaultOutputModeSelect">
|
||||
<option value="clipboard">剪贴板</option>
|
||||
<option value="both">剪贴板 + 提示</option>
|
||||
</select>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<div style="display: flex; gap: 8px; margin-top: 20px;">
|
||||
<button class="btn btn-primary" id="saveSettingsBtn" style="flex: 1;">保存设置</button>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- 提示框 -->
|
||||
<div class="toast" id="toast"></div>
|
||||
|
||||
<script>
|
||||
const recordBtn = document.getElementById('recordBtn');
|
||||
const statusIndicator = document.getElementById('statusIndicator');
|
||||
@ -245,8 +566,30 @@
|
||||
const resultText = document.getElementById('resultText');
|
||||
const languageSelect = document.getElementById('languageSelect');
|
||||
const outputModeSelect = document.getElementById('outputModeSelect');
|
||||
const settingsBtn = document.getElementById('settingsBtn');
|
||||
const settingsModal = document.getElementById('settingsModal');
|
||||
const modalClose = document.getElementById('modalClose');
|
||||
const addModelBtn = document.getElementById('addModelBtn');
|
||||
const addModelForm = document.getElementById('addModelForm');
|
||||
const confirmAddModelBtn = document.getElementById('confirmAddModelBtn');
|
||||
const cancelAddModelBtn = document.getElementById('cancelAddModelBtn');
|
||||
const modelList = document.getElementById('modelList');
|
||||
const defaultModelSelect = document.getElementById('defaultModelSelect');
|
||||
const saveSettingsBtn = document.getElementById('saveSettingsBtn');
|
||||
const toast = document.getElementById('toast');
|
||||
|
||||
let isRecording = false;
|
||||
let models = [];
|
||||
let settings = {};
|
||||
|
||||
// 显示提示
|
||||
function showToast(message, type = 'info') {
|
||||
toast.textContent = message;
|
||||
toast.className = 'toast show ' + type;
|
||||
setTimeout(() => {
|
||||
toast.className = 'toast';
|
||||
}, 2000);
|
||||
}
|
||||
|
||||
// 更新 UI 状态
|
||||
function updateUI() {
|
||||
@ -263,6 +606,63 @@
|
||||
}
|
||||
}
|
||||
|
||||
// 渲染模型列表
|
||||
function renderModelList() {
|
||||
if (models.length === 0) {
|
||||
modelList.innerHTML = '<p style="color: var(--text-secondary); text-align: center; padding: 20px;">暂无模型配置</p>';
|
||||
defaultModelSelect.innerHTML = '<option value="">-- 选择默认模型 --</option>';
|
||||
return;
|
||||
}
|
||||
|
||||
modelList.innerHTML = models.map((model, index) => `
|
||||
<div class="model-item">
|
||||
<div class="model-item-header">
|
||||
<span class="model-item-name">${escapeHtml(model.name)}</span>
|
||||
<span class="model-item-status ${model.exists ? 'loaded' : 'missing'}">
|
||||
${model.exists ? '✓ 存在' : '✗ 未找到'}
|
||||
</span>
|
||||
</div>
|
||||
<div class="model-item-path">${escapeHtml(model.path)}</div>
|
||||
<div class="model-item-actions">
|
||||
<button class="btn btn-secondary" onclick="useModel(${index})">使用</button>
|
||||
<button class="btn btn-danger" onclick="deleteModel(${index})">删除</button>
|
||||
</div>
|
||||
</div>
|
||||
`).join('');
|
||||
|
||||
// 更新默认模型选择
|
||||
defaultModelSelect.innerHTML = '<option value="">-- 选择默认模型 --</option>' +
|
||||
models.map((model, index) => `
|
||||
<option value="${index}" ${settings.defaultModel === index ? 'selected' : ''}>
|
||||
${escapeHtml(model.name)}
|
||||
</option>
|
||||
`).join('');
|
||||
}
|
||||
|
||||
// HTML 转义
|
||||
function escapeHtml(text) {
|
||||
const div = document.createElement('div');
|
||||
div.textContent = text;
|
||||
return div.innerHTML;
|
||||
}
|
||||
|
||||
// 使用模型
|
||||
window.useModel = (index) => {
|
||||
settings.defaultModel = index;
|
||||
renderModelList();
|
||||
showToast('已选择模型', 'success');
|
||||
};
|
||||
|
||||
// 删除模型
|
||||
window.deleteModel = async (index) => {
|
||||
if (confirm(`确定要删除模型 "${models[index].name}" 吗？`)) {
|
||||
models.splice(index, 1);
|
||||
await window.electronAPI?.saveSettings({ models, defaultModel: settings.defaultModel });
|
||||
renderModelList();
|
||||
showToast('模型已删除', 'success');
|
||||
}
|
||||
};
|
||||
|
||||
// 点击录音按钮
|
||||
recordBtn.addEventListener('click', async () => {
|
||||
isRecording = !isRecording;
|
||||
@ -286,21 +686,93 @@
|
||||
updateUI();
|
||||
});
|
||||
|
||||
// 模拟识别结果（开发用）
|
||||
function simulateResult(text) {
|
||||
resultText.textContent = text;
|
||||
if (text) {
|
||||
window.electronAPI?.copyToClipboard(text);
|
||||
}
|
||||
}
|
||||
|
||||
// 设置保存
|
||||
languageSelect.addEventListener('change', () => {
|
||||
window.electronAPI?.saveSettings({ language: languageSelect.value });
|
||||
window.electronAPI?.saveSettings({ ...settings, language: languageSelect.value });
|
||||
});
|
||||
|
||||
outputModeSelect.addEventListener('change', () => {
|
||||
window.electronAPI?.saveSettings({ outputMode: outputModeSelect.value });
|
||||
window.electronAPI?.saveSettings({ ...settings, outputMode: outputModeSelect.value });
|
||||
});
|
||||
|
||||
// 打开设置弹窗
|
||||
settingsBtn.addEventListener('click', async () => {
|
||||
settings = await window.electronAPI?.getSettings() || {};
|
||||
models = settings.models || [];
|
||||
renderModelList();
|
||||
settingsModal.classList.add('active');
|
||||
});
|
||||
|
||||
// 关闭设置弹窗
|
||||
modalClose.addEventListener('click', () => {
|
||||
settingsModal.classList.remove('active');
|
||||
addModelForm.style.display = 'none';
|
||||
});
|
||||
|
||||
settingsModal.addEventListener('click', (e) => {
|
||||
if (e.target === settingsModal) {
|
||||
settingsModal.classList.remove('active');
|
||||
addModelForm.style.display = 'none';
|
||||
}
|
||||
});
|
||||
|
||||
// 显示添加模型表单
|
||||
addModelBtn.addEventListener('click', () => {
|
||||
addModelForm.style.display = 'block';
|
||||
});
|
||||
|
||||
// 取消添加模型
|
||||
cancelAddModelBtn.addEventListener('click', () => {
|
||||
addModelForm.style.display = 'none';
|
||||
modelNameInput.value = '';
|
||||
modelPathInput.value = '';
|
||||
});
|
||||
|
||||
// 确认添加模型
|
||||
confirmAddModelBtn.addEventListener('click', async () => {
|
||||
const name = document.getElementById('modelNameInput').value.trim();
|
||||
const path = document.getElementById('modelPathInput').value.trim();
|
||||
const type = document.getElementById('modelTypeSelect').value;
|
||||
const language = document.getElementById('modelLanguageSelect').value;
|
||||
|
||||
if (!name || !path) {
|
||||
showToast('请填写模型名称和路径', 'error');
|
||||
return;
|
||||
}
|
||||
|
||||
// 检查文件是否存在
|
||||
const exists = await window.electronAPI?.checkFileExists(path);
|
||||
|
||||
models.push({
|
||||
name,
|
||||
path,
|
||||
type,
|
||||
language,
|
||||
exists: exists || false
|
||||
});
|
||||
|
||||
await window.electronAPI?.saveSettings({ ...settings, models });
|
||||
renderModelList();
|
||||
addModelForm.style.display = 'none';
|
||||
document.getElementById('modelNameInput').value = '';
|
||||
document.getElementById('modelPathInput').value = '';
|
||||
showToast('模型已添加', 'success');
|
||||
});
|
||||
|
||||
// 保存设置
|
||||
saveSettingsBtn.addEventListener('click', async () => {
|
||||
const defaultModel = defaultModelSelect.value;
|
||||
const defaultOutputMode = document.getElementById('defaultOutputModeSelect').value;
|
||||
|
||||
await window.electronAPI?.saveSettings({
|
||||
...settings,
|
||||
models,
|
||||
defaultModel: defaultModel ? parseInt(defaultModel) : null,
|
||||
defaultOutputMode
|
||||
});
|
||||
|
||||
showToast('设置已保存', 'success');
|
||||
settingsModal.classList.remove('active');
|
||||
});
|
||||
|
||||
// 加载设置
|
||||
|
||||
Loading…
Reference in New Issue
Block a user