refactor: 分离前端代码到 ui.js

This commit is contained in:
zwbcc
2026-03-28 20:59:27 +08:00
parent fc02c5100b
commit f4556a69d4
3 changed files with 317 additions and 321 deletions

View File

@@ -67,11 +67,12 @@ systemctl --user enable --now image-generator.service
```
image-generator/
├── index.html # 前端
├── style.css # 样式
├── index.html # HTML 页面
├── style.css # 样式(深色主题)
├── ui.js # 前端交互逻辑
├── app.js # 服务端Express
├── start.sh # 启动脚本
├── config.json # API Key 存储(自动生成)
├── config.json # API Key 存储(自动生成,不提交
└── CHANGELOG.md
```

View File

@@ -219,323 +219,6 @@
</div>
</div>
<script>
'use strict';
// ── State ──────────────────────────────────────────────────────
let currentModel = 'image-01';
let currentImages = []; // [{url, base64}]
let loading = false;
// ── DOM refs ───────────────────────────────────────────────────
const $ = id => document.getElementById(id);
const promptInput = $('promptInput');
const charCount = $('charCount');
const generateForm = $('generateForm');
const generateBtn = $('generateBtn');
const btnIcon = generateBtn.querySelector('.btn-icon');
const btnText = generateBtn.querySelector('.btn-text');
const spinner = generateBtn.querySelector('.spinner');
const errorMsg = $('errorMsg');
const resultArea = $('resultArea');
const resultMeta = $('resultMeta');
const imageGrid = $('imageGrid');
const apiKeyAlert = $('apiKeyAlert');
const alertSettings = $('alertSettingsBtn');
const aspectRatio = $('aspectRatio');
const responseFormat = $('responseFormat');
const nInput = $('nInput');
const nMinus = $('nMinus');
const nPlus = $('nPlus');
const seedInput = $('seedInput');
const widthInput = $('widthInput');
const heightInput = $('heightInput');
const styleGroup = $('styleGroup');
const styleType = $('styleType');
const styleWeight = $('styleWeight');
const weightHint = $('weightHint');
const promptOptimizer= $('promptOptimizer');
const watermark = $('watermark');
const settingsModal = $('settingsModal');
const openSettings = $('openSettings');
const closeSettings = $('closeSettings');
const cancelSettings = $('cancelSettings');
const saveSettingsBtn= $('saveSettingsBtn');
const apiKeyInput = $('apiKeyInput');
const toggleKey = $('toggleKey');
const baseUrlInput = $('baseUrlInput');
// ── Model tabs ─────────────────────────────────────────────────
document.querySelectorAll('.model-tab').forEach(tab => {
tab.addEventListener('click', () => {
const model = tab.dataset.model;
setModel(model);
});
});
function setModel(model) {
currentModel = model;
document.querySelectorAll('.model-tab').forEach(t =>
t.classList.toggle('active', t.dataset.model === model)
);
const isLive = model === 'image-01-live';
styleGroup.style.display = isLive ? '' : 'none';
// 21:9 only for image-01
const opt21_9 = aspectRatio.querySelector('option[value="21:9"]');
if (opt21_9) opt21_9.disabled = isLive;
if (isLive && aspectRatio.value === '21:9') aspectRatio.value = '16:9';
}
// ── Stepper (n) ────────────────────────────────────────────────
nMinus.addEventListener('click', () => {
const v = parseInt(nInput.value) - 1;
if (v >= 1) nInput.value = v;
});
nPlus.addEventListener('click', () => {
const v = parseInt(nInput.value) + 1;
if (v <= 9) nInput.value = v;
});
// ── Weight slider ──────────────────────────────────────────────
styleWeight.addEventListener('input', () => {
weightHint.textContent = (styleWeight.value / 10).toFixed(1);
});
// ── Helpers ────────────────────────────────────────────────────
function setLoading(on) {
loading = on;
generateBtn.disabled = on;
btnIcon.style.display = on ? 'none' : '';
btnText.style.display = on ? 'none' : '';
spinner.style.display = on ? 'inline' : 'none';
generateBtn.classList.toggle('loading', on);
}
function showError(msg) {
errorMsg.textContent = msg;
errorMsg.style.display = 'flex';
errorMsg.scrollIntoView({ behavior: 'smooth', block: 'nearest' });
}
function clearError() {
errorMsg.style.display = 'none';
}
async function api(path, opts = {}) {
const res = await fetch(path, opts);
const json = await res.json();
if (!res.ok) throw new Error(json.error || `HTTP ${res.status}`);
return json;
}
function buildPayload() {
const n = parseInt(nInput.value);
const seed = seedInput.value ? parseInt(seedInput.value) : undefined;
const width = widthInput.value ? parseInt(widthInput.value) : undefined;
const height = heightInput.value ? parseInt(heightInput.value) : undefined;
const payload = {
model: currentModel,
prompt: promptInput.value.trim(),
aspect_ratio: aspectRatio.value,
response_format: responseFormat.value,
n: n,
};
if (seed) payload.seed = seed;
if (width && height) { payload.width = width; payload.height = height; }
if (promptOptimizer.checked) payload.prompt_optimizer = true;
if (watermark.checked) payload.aigc_watermark = true;
if (currentModel === 'image-01-live') {
payload.style = {
style_type: styleType.value,
style_weight: parseFloat((styleWeight.value / 10).toFixed(1)),
};
}
return payload;
}
function renderResults(images, id, successCount, failedCount) {
currentImages = images;
imageGrid.innerHTML = '';
if (id) {
resultMeta.innerHTML = `
<span>任务 ID<code>${id}</code></span>
${successCount != null ? `<span>成功 ${successCount}${failedCount ? `,失败 ${failedCount}` : ''}</span>` : ''}
`;
resultMeta.style.display = 'flex';
} else {
resultMeta.style.display = 'none';
}
const fmt = responseFormat.value;
images.forEach((src, i) => {
const card = document.createElement('div');
card.className = 'image-card';
const img = document.createElement('img');
img.src = fmt === 'url' ? src : `data:image/jpeg;base64,${src}`;
img.alt = `生成结果 ${i + 1}`;
img.loading = 'lazy';
const actions = document.createElement('div');
actions.className = 'image-actions';
const downloadBtn = document.createElement('button');
downloadBtn.className = 'action-btn';
downloadBtn.innerHTML = `<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4"/><polyline points="7,10 12,15 17,10"/><line x1="12" y1="15" x2="12" y2="3"/></svg>下载`;
downloadBtn.addEventListener('click', () => downloadImage(src, fmt, i));
const copyBtn = document.createElement('button');
copyBtn.className = 'action-btn';
copyBtn.innerHTML = `<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect x="9" y="9" width="13" height="13" rx="2" ry="2"/><path d="M5 15H4a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h9a2 2 0 0 1 2 2v1"/></svg>复制`;
copyBtn.addEventListener('click', () => copySrc(src, fmt, copyBtn));
actions.appendChild(downloadBtn);
actions.appendChild(copyBtn);
card.appendChild(img);
card.appendChild(actions);
imageGrid.appendChild(card);
});
resultArea.style.display = 'block';
resultArea.scrollIntoView({ behavior: 'smooth', block: 'start' });
}
function downloadImage(src, fmt, index) {
const a = document.createElement('a');
if (fmt === 'url') {
a.href = src;
} else {
a.href = `data:image/jpeg;base64,${src}`;
}
a.download = `生成-${Date.now()}-${index + 1}.jpg`;
a.target = '_blank';
a.click();
}
async function copySrc(src, fmt, btn) {
try {
if (fmt === 'url') {
await navigator.clipboard.writeText(src);
} else {
await navigator.clipboard.writeText(src);
}
const orig = btn.innerHTML;
btn.innerHTML = `<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><polyline points="20,6 9,17 4,12"/></svg>已复制`;
setTimeout(() => { btn.innerHTML = orig; }, 1500);
} catch {
showError('复制失败');
}
}
// ── Init ────────────────────────────────────────────────────────
async function init() {
try {
const cfg = await api('/api/config');
if (!cfg.hasApiKey) apiKeyAlert.style.display = 'flex';
baseUrlInput.value = cfg.baseUrl || 'https://api.minimaxi.com';
} catch { /* non-fatal */ }
}
// ── Generate ────────────────────────────────────────────────────
generateForm.addEventListener('submit', async (e) => {
e.preventDefault();
if (loading) return;
const prompt = promptInput.value.trim();
if (!prompt) { showError('请输入图片描述'); return; }
clearError();
resultArea.style.display = 'none';
setLoading(true);
try {
const payload = buildPayload();
const data = await api('/api/generate', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(payload),
});
const images = data.images || [];
if (images.length === 0) {
throw new Error('未返回任何图片');
}
renderResults(images, data.id, data.success_count, data.failed_count);
} catch (err) {
showError(err.message);
} finally {
setLoading(false);
}
});
// ── Prompt char count ──────────────────────────────────────────
promptInput.addEventListener('input', () => {
charCount.textContent = `${promptInput.value.length} / 1500`;
});
// ── Settings modal ──────────────────────────────────────────────
function openModal() {
settingsModal.classList.add('open');
apiKeyInput.focus();
}
function closeModal() {
settingsModal.classList.remove('open');
clearError();
}
openSettings.addEventListener('click', openModal);
alertSettings.addEventListener('click', openModal);
closeSettings.addEventListener('click', closeModal);
cancelSettings.addEventListener('click', closeModal);
settingsModal.addEventListener('click', e => { if (e.target === settingsModal) closeModal(); });
toggleKey.addEventListener('click', () => {
apiKeyInput.type = apiKeyInput.type === 'password' ? 'text' : 'password';
});
saveSettingsBtn.addEventListener('click', async () => {
clearError();
const apiKey = apiKeyInput.value.trim();
const baseUrl = baseUrlInput.value.trim() || 'https://api.minimaxi.com';
try {
await api('/api/config', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ apiKey, baseUrl }),
});
if (apiKey) apiKeyAlert.style.display = 'none';
closeModal();
} catch (err) {
showError(err.message);
}
});
// ── Start ──────────────────────────────────────────────────────
init();
</script>
<script src="ui.js" defer></script>
</body>
</html>

312
ui.js Normal file
View File

@@ -0,0 +1,312 @@
'use strict';
// ── State ──────────────────────────────────────────────────────
let currentModel = 'image-01';
let currentImages = []; // [{url, base64}]
let loading = false;
// ── DOM refs ───────────────────────────────────────────────────
const $ = id => document.getElementById(id);
const promptInput = $('promptInput');
const charCount = $('charCount');
const generateForm = $('generateForm');
const generateBtn = $('generateBtn');
const btnIcon = generateBtn.querySelector('.btn-icon');
const btnText = generateBtn.querySelector('.btn-text');
const spinner = generateBtn.querySelector('.spinner');
const errorMsg = $('errorMsg');
const resultArea = $('resultArea');
const resultMeta = $('resultMeta');
const imageGrid = $('imageGrid');
const apiKeyAlert = $('apiKeyAlert');
const alertSettings = $('alertSettingsBtn');
const aspectRatio = $('aspectRatio');
const responseFormat = $('responseFormat');
const nInput = $('nInput');
const nMinus = $('nMinus');
const nPlus = $('nPlus');
const seedInput = $('seedInput');
const widthInput = $('widthInput');
const heightInput = $('heightInput');
const styleGroup = $('styleGroup');
const styleType = $('styleType');
const styleWeight = $('styleWeight');
const weightHint = $('weightHint');
const promptOptimizer= $('promptOptimizer');
const watermark = $('watermark');
const settingsModal = $('settingsModal');
const openSettings = $('openSettings');
const closeSettings = $('closeSettings');
const cancelSettings = $('cancelSettings');
const saveSettingsBtn= $('saveSettingsBtn');
const apiKeyInput = $('apiKeyInput');
const toggleKey = $('toggleKey');
const baseUrlInput = $('baseUrlInput');
// ── Model tabs ─────────────────────────────────────────────────
document.querySelectorAll('.model-tab').forEach(tab => {
tab.addEventListener('click', () => {
const model = tab.dataset.model;
setModel(model);
});
});
function setModel(model) {
currentModel = model;
document.querySelectorAll('.model-tab').forEach(t =>
t.classList.toggle('active', t.dataset.model === model)
);
const isLive = model === 'image-01-live';
styleGroup.style.display = isLive ? '' : 'none';
// 21:9 only for image-01
const opt21_9 = aspectRatio.querySelector('option[value="21:9"]');
if (opt21_9) opt21_9.disabled = isLive;
if (isLive && aspectRatio.value === '21:9') aspectRatio.value = '16:9';
}
// ── Stepper (n) ────────────────────────────────────────────────
nMinus.addEventListener('click', () => {
const v = parseInt(nInput.value) - 1;
if (v >= 1) nInput.value = v;
});
nPlus.addEventListener('click', () => {
const v = parseInt(nInput.value) + 1;
if (v <= 9) nInput.value = v;
});
// ── Weight slider ──────────────────────────────────────────────
styleWeight.addEventListener('input', () => {
weightHint.textContent = (styleWeight.value / 10).toFixed(1);
});
// ── Helpers ────────────────────────────────────────────────────
function setLoading(on) {
loading = on;
generateBtn.disabled = on;
btnIcon.style.display = on ? 'none' : '';
btnText.style.display = on ? 'none' : '';
spinner.style.display = on ? 'inline' : 'none';
generateBtn.classList.toggle('loading', on);
}
function showError(msg) {
errorMsg.textContent = msg;
errorMsg.style.display = 'flex';
errorMsg.scrollIntoView({ behavior: 'smooth', block: 'nearest' });
}
function clearError() {
errorMsg.style.display = 'none';
}
async function api(path, opts = {}) {
const res = await fetch(path, opts);
const json = await res.json();
if (!res.ok) throw new Error(json.error || `HTTP ${res.status}`);
return json;
}
function buildPayload() {
const n = parseInt(nInput.value);
const seed = seedInput.value ? parseInt(seedInput.value) : undefined;
const width = widthInput.value ? parseInt(widthInput.value) : undefined;
const height = heightInput.value ? parseInt(heightInput.value) : undefined;
const payload = {
model: currentModel,
prompt: promptInput.value.trim(),
aspect_ratio: aspectRatio.value,
response_format: responseFormat.value,
n: n,
};
if (seed) payload.seed = seed;
if (width && height) { payload.width = width; payload.height = height; }
if (promptOptimizer.checked) payload.prompt_optimizer = true;
if (watermark.checked) payload.aigc_watermark = true;
if (currentModel === 'image-01-live') {
payload.style = {
style_type: styleType.value,
style_weight: parseFloat((styleWeight.value / 10).toFixed(1)),
};
}
return payload;
}
function renderResults(images, id, successCount, failedCount) {
currentImages = images;
imageGrid.innerHTML = '';
if (id) {
resultMeta.innerHTML = `
<span>任务 ID<code>${id}</code></span>
${successCount != null ? `<span>成功 ${successCount}${failedCount ? `,失败 ${failedCount}` : ''}</span>` : ''}
`;
resultMeta.style.display = 'flex';
} else {
resultMeta.style.display = 'none';
}
const fmt = responseFormat.value;
images.forEach((src, i) => {
const card = document.createElement('div');
card.className = 'image-card';
const img = document.createElement('img');
img.src = fmt === 'url' ? src : `data:image/jpeg;base64,${src}`;
img.alt = `生成结果 ${i + 1}`;
img.loading = 'lazy';
const actions = document.createElement('div');
actions.className = 'image-actions';
const downloadBtn = document.createElement('button');
downloadBtn.className = 'action-btn';
downloadBtn.innerHTML = `<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4"/><polyline points="7,10 12,15 17,10"/><line x1="12" y1="15" x2="12" y2="3"/></svg>下载`;
downloadBtn.addEventListener('click', () => downloadImage(src, fmt, i));
const copyBtn = document.createElement('button');
copyBtn.className = 'action-btn';
copyBtn.innerHTML = `<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><rect x="9" y="9" width="13" height="13" rx="2" ry="2"/><path d="M5 15H4a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h9a2 2 0 0 1 2 2v1"/></svg>复制`;
copyBtn.addEventListener('click', () => copySrc(src, fmt, copyBtn));
actions.appendChild(downloadBtn);
actions.appendChild(copyBtn);
card.appendChild(img);
card.appendChild(actions);
imageGrid.appendChild(card);
});
resultArea.style.display = 'block';
resultArea.scrollIntoView({ behavior: 'smooth', block: 'start' });
}
function downloadImage(src, fmt, index) {
const a = document.createElement('a');
if (fmt === 'url') {
a.href = src;
} else {
a.href = `data:image/jpeg;base64,${src}`;
}
a.download = `生成-${Date.now()}-${index + 1}.jpg`;
a.target = '_blank';
a.click();
}
async function copySrc(src, fmt, btn) {
try {
await navigator.clipboard.writeText(src);
const orig = btn.innerHTML;
btn.innerHTML = `<svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><polyline points="20,6 9,17 4,12"/></svg>已复制`;
setTimeout(() => { btn.innerHTML = orig; }, 1500);
} catch {
showError('复制失败');
}
}
// ── Init ────────────────────────────────────────────────────────
async function init() {
try {
const cfg = await api('/api/config');
if (!cfg.hasApiKey) apiKeyAlert.style.display = 'flex';
baseUrlInput.value = cfg.baseUrl || 'https://api.minimaxi.com';
} catch { /* non-fatal */ }
}
// ── Generate ────────────────────────────────────────────────────
generateForm.addEventListener('submit', async (e) => {
e.preventDefault();
if (loading) return;
const prompt = promptInput.value.trim();
if (!prompt) { showError('请输入图片描述'); return; }
clearError();
resultArea.style.display = 'none';
setLoading(true);
try {
const payload = buildPayload();
const data = await api('/api/generate', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(payload),
});
const images = data.images || [];
if (images.length === 0) {
throw new Error('未返回任何图片');
}
renderResults(images, data.id, data.success_count, data.failed_count);
} catch (err) {
showError(err.message);
} finally {
setLoading(false);
}
});
// ── Prompt char count ──────────────────────────────────────────
promptInput.addEventListener('input', () => {
charCount.textContent = `${promptInput.value.length} / 1500`;
});
// ── Settings modal ──────────────────────────────────────────────
function openModal() {
settingsModal.classList.add('open');
apiKeyInput.focus();
}
function closeModal() {
settingsModal.classList.remove('open');
clearError();
}
openSettings.addEventListener('click', openModal);
alertSettings.addEventListener('click', openModal);
closeSettings.addEventListener('click', closeModal);
cancelSettings.addEventListener('click', closeModal);
settingsModal.addEventListener('click', e => { if (e.target === settingsModal) closeModal(); });
toggleKey.addEventListener('click', () => {
apiKeyInput.type = apiKeyInput.type === 'password' ? 'text' : 'password';
});
saveSettingsBtn.addEventListener('click', async () => {
clearError();
const apiKey = apiKeyInput.value.trim();
const baseUrl = baseUrlInput.value.trim() || 'https://api.minimaxi.com';
try {
await api('/api/config', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ apiKey, baseUrl }),
});
if (apiKey) apiKeyAlert.style.display = 'none';
closeModal();
} catch (err) {
showError(err.message);
}
});
// ── Start ──────────────────────────────────────────────────────
init();