ai_v/static/js/main.js

lucide.createIcons();

async function checkAuth() {
    const r = await fetch('/api/auth/me');
    const d = await r.json();
    const profile = document.getElementById('userProfile');
    const entry = document.getElementById('loginEntryBtn');
    const loginHint = document.getElementById('loginHint');
    const submitBtn = document.getElementById('submitBtn');

    if (d.logged_in) {
        if (profile) profile.classList.remove('hidden');
        if (entry) entry.classList.add('hidden');
        if (loginHint) loginHint.classList.add('hidden');
        submitBtn.disabled = false;
        submitBtn.classList.remove('opacity-50', 'cursor-not-allowed');
        const phoneDisp = document.getElementById('userPhoneDisplay');
        if (phoneDisp) phoneDisp.innerText = d.phone;

        // 处理积分显示
        const pointsBadge = document.getElementById('pointsBadge');
        const pointsDisplay = document.getElementById('pointsDisplay');
        if (pointsBadge && pointsDisplay) {
            pointsBadge.classList.remove('hidden');
            pointsDisplay.innerText = d.points;
        }
        const headerPoints = document.getElementById('headerPoints');
        if (headerPoints) headerPoints.innerText = d.points;

        // 处理自定义 Key 显示逻辑
        // 处理自定义 Key 显示逻辑
        const keyBtn = document.getElementById('modeKeyBtn');
        const modeButtonsContainer = keyBtn ? keyBtn.parentElement : null;

        if (keyBtn && modeButtonsContainer) {
            if (d.hide_custom_key) {
                // 如果后端说要隐藏（用过积分生成），则隐藏整个按钮组
                modeButtonsContainer.classList.add('hidden');
                // 修改标题为“账户状态”
                const authTitle = document.querySelector('#authSection h3');
                if (authTitle) authTitle.innerText = '账户状态';
            } else {
                // 否则显示按钮组，让用户可以选择
                modeButtonsContainer.classList.remove('hidden');

                // 如果有 Key，默认帮忙切过去，方便老用户
                if (d.api_key) {
                    switchMode('key');
                    const keyInput = document.getElementById('apiKey');
                    if (keyInput && !keyInput.value) keyInput.value = d.api_key;
                    return;
                }
            }
        }

        // 如果用户已经有绑定的 Key，且当前没手动输入，则默认切到 Key 模式
        // 强制使用积分模式
        switchMode('trial');
    } else {
        if (profile) profile.classList.add('hidden');
        if (entry) entry.classList.remove('hidden');
        if (loginHint) loginHint.classList.remove('hidden');
        submitBtn.disabled = true;
        submitBtn.classList.add('opacity-50', 'cursor-not-allowed');
    }
    lucide.createIcons();
}

// 移除 redundant logout 监听，因为 base.html 已处理全局登出

// 历史记录分页状态
let currentHistoryPage = 1;
let hasMoreHistory = true;
let isHistoryLoading = false;

// 存储当前生成的所有图片 URL
let currentGeneratedUrls = [];
let currentMode = 'trial'; // 'trial' 或 'key'
let uploadedFiles = []; // 存储当前待上传的参考图
let isSetterActive = false; // 是否激活了拍摄角度设置器模式

function switchMode(mode) {
    currentMode = mode;
    const trialBtn = document.getElementById('modeTrialBtn');
    const keyBtn = document.getElementById('modeKeyBtn');
    const keyInputGroup = document.getElementById('keyInputGroup');
    const premiumToggle = document.getElementById('premiumToggle');

    if (mode === 'trial') {
        trialBtn.classList.add('border-indigo-500', 'bg-indigo-50', 'text-indigo-600', 'border-2');
        trialBtn.classList.remove('border-slate-200', 'text-slate-400');
        keyBtn.classList.remove('border-indigo-500', 'bg-indigo-50', 'text-indigo-600', 'border-2');
        keyBtn.classList.add('border-slate-200', 'text-slate-400');
        keyInputGroup.classList.add('hidden');
        if (premiumToggle) premiumToggle.classList.remove('hidden');
    } else {
        keyBtn.classList.add('border-indigo-500', 'bg-indigo-50', 'text-indigo-600', 'border-2');
        keyBtn.classList.remove('border-slate-200', 'text-slate-400');
        trialBtn.classList.remove('border-indigo-500', 'bg-indigo-50', 'text-indigo-600', 'border-2');
        trialBtn.classList.add('border-slate-200', 'text-slate-400');
        keyInputGroup.classList.remove('hidden');
        if (premiumToggle) premiumToggle.classList.add('hidden');
    }
    updateCostPreview(); // 切换模式时同步计费预览
}

async function downloadImage(url) {
    try {
        const response = await fetch(url);
        const blob = await response.blob();
        const blobUrl = window.URL.createObjectURL(blob);
        const a = document.createElement('a');
        a.href = blobUrl;
        // 从 URL 提取文件名
        const filename = url.split('/').pop().split('?')[0] || 'ai-vision-image.png';
        a.download = filename;
        document.body.appendChild(a);
        a.click();
        document.body.removeChild(a);
        window.URL.revokeObjectURL(blobUrl);
    } catch (e) {
        console.error('下载失败:', e);
        showToast('下载失败，请尝试右键保存', 'error');
    }
}

async function loadHistory(isLoadMore = false) {
    if (isHistoryLoading || (!hasMoreHistory && isLoadMore)) return;

    isHistoryLoading = true;
    if (!isLoadMore) {
        currentHistoryPage = 1;
        document.getElementById('historyList').innerHTML = '';
    }

    const footer = document.getElementById('historyFooter');
    footer.classList.remove('hidden');

    try {
        const r = await fetch(`/api/history?page=${currentHistoryPage}&per_page=10&filter_type=image`);
        const d = await r.json();

        const list = document.getElementById('historyList');

        if (d.history && d.history.length > 0) {
            const html = d.history.map(item => `
                <div class="bg-white border border-slate-100 rounded-2xl p-4 space-y-3 hover:border-indigo-100 transition-all shadow-sm group">
                    <div class="flex items-center justify-between">
                        <span class="text-[10px] font-black text-slate-400 bg-slate-50 px-2 py-0.5 rounded-md uppercase tracking-widest">${item.created_at}</span>
                        <span class="text-[10px] font-bold text-indigo-500">${item.model}</span>
                    </div>
                    <div class="grid grid-cols-3 gap-2">
                        ${item.urls.map(u => `
                            <div class="aspect-square rounded-lg overflow-hidden border border-slate-100 cursor-pointer transition-transform hover:scale-105" onclick="window.open('${u.url}')">
                                <img src="${u.thumb}" class="w-full h-full object-cover" loading="lazy">
                            </div>
                        `).join('')}
                    </div>
                </div>
            `).join('');

            if (isLoadMore) {
                list.insertAdjacentHTML('beforeend', html);
            } else {
                list.innerHTML = html;
            }

            hasMoreHistory = d.has_next;
            currentHistoryPage++;
        } else if (!isLoadMore) {
            list.innerHTML = `<div class="flex flex-col items-center justify-center h-64 text-slate-300">
                <i data-lucide="inbox" class="w-12 h-12 mb-4"></i>
                <span class="text-xs font-bold">暂无生成记录</span>
            </div>`;
        }
        lucide.createIcons();
    } catch (e) {
        console.error('加载历史失败:', e);
        if (!isLoadMore) {
            document.getElementById('historyList').innerHTML = `<div class="text-center text-rose-400 text-xs font-bold py-10">加载失败: ${e.message}</div>`;
        }
    } finally {
        isHistoryLoading = false;
        footer.classList.add('hidden');
    }
}

async function init() {
    checkAuth();

    // 模式切换监听
    const modeTrialBtn = document.getElementById('modeTrialBtn');
    const modeKeyBtn = document.getElementById('modeKeyBtn');
    const isPremiumCheckbox = document.getElementById('isPremium');
    if (modeTrialBtn) modeTrialBtn.onclick = () => switchMode('trial');
    if (modeKeyBtn) modeKeyBtn.onclick = () => switchMode('key');
    if (isPremiumCheckbox) isPremiumCheckbox.onchange = () => updateCostPreview();
    const numSelect = document.getElementById('numSelect');
    if (numSelect) numSelect.onchange = () => updateCostPreview();

    // 历史记录控制
    const historyDrawer = document.getElementById('historyDrawer');
    const showHistoryBtn = document.getElementById('showHistoryBtn');
    const closeHistoryBtn = document.getElementById('closeHistoryBtn');
    const historyList = document.getElementById('historyList');

    // 3D 构图辅助控制
    const openVisualizerBtn = document.getElementById('openVisualizerBtn');
    const closeVisualizerBtn = document.getElementById('closeVisualizerBtn');
    if (openVisualizerBtn) openVisualizerBtn.onclick = openVisualizerModal;
    if (closeVisualizerBtn) closeVisualizerBtn.onclick = closeVisualizerModal;
    const openSavePromptBtn = document.getElementById('openSavePromptBtn');
    if (openSavePromptBtn) {
        openSavePromptBtn.onclick = (e) => {
            e.preventDefault();
            e.stopPropagation();
            openSavePromptModal();
        };
    }

    if (showHistoryBtn) {
        showHistoryBtn.onclick = () => {
            historyDrawer.classList.remove('translate-x-full');
            loadHistory(false);
        };
    }
    if (closeHistoryBtn) {
        closeHistoryBtn.onclick = () => {
            historyDrawer.classList.add('translate-x-full');
        };
    }

    // 瀑布流滚动加载
    if (historyList) {
        historyList.onscroll = () => {
            const threshold = 100;
            if (historyList.scrollTop + historyList.clientHeight >= historyList.scrollHeight - threshold) {
                loadHistory(true);
            }
        };
    }

    // 全部下载按钮逻辑
    const downloadAllBtn = document.getElementById('downloadAllBtn');
    if (downloadAllBtn) {
        downloadAllBtn.onclick = async () => {
            if (currentGeneratedUrls.length === 0) return;
            showToast(`正在准备下载 ${currentGeneratedUrls.length} 张作品...`, 'info');
            for (const url of currentGeneratedUrls) {
                await downloadImage(url);
                // 稍微延迟一下，防止浏览器拦截
                await new Promise(r => setTimeout(r, 300));
            }
        };
    }

    // 重新生成按钮逻辑
    const regenBtn = document.getElementById('regenBtn');
    if (regenBtn) {
        regenBtn.onclick = () => {
            const submitBtn = document.getElementById('submitBtn');
            if (submitBtn) submitBtn.click();
        };
    }

    // 检查是否有来自 URL 的错误提示
    const urlParams = new URLSearchParams(window.location.search);
    if (urlParams.has('error')) {
        showToast(urlParams.get('error'), 'error');
        // 清理 URL 参数以防刷新时重复提示
        window.history.replaceState({}, document.title, window.location.pathname);
    }

    try {
        await refreshPromptsList();
        updateCostPreview(); // 初始化时显示默认模型的积分
    } catch (e) { console.error(e); }

    // 初始化拖拽排序
    const prev = document.getElementById('imagePreview');
    if (prev) {
        new Sortable(prev, {
            animation: 150,
            ghostClass: 'opacity-50',
            onEnd: function (evt) {
                // 拖拽结束，重新对 uploadedFiles 排序
                const oldIndex = evt.oldIndex;
                const newIndex = evt.newIndex;
                const movedFile = uploadedFiles.splice(oldIndex, 1)[0];
                uploadedFiles.splice(newIndex, 0, movedFile);
                // 重新渲染以更新“图1/图2/图3”标签
                renderImagePreviews();
            }
        });
    }
}

function fillSelect(id, list) {
    const el = document.getElementById(id);
    if (!el) return;
    if (id === 'modelSelect') {
        el.innerHTML = list.map(i => `<option value="${i.value}" data-cost="${i.cost || 0}">${i.label} (${i.cost || 0}积分)</option>`).join('');
    } else {
        el.innerHTML = list.map(i => {
            // 处理 data-id
            const dataIdAttr = i.id ? ` data-id="${i.id}"` : '';
            const disabledAttr = i.disabled ? ' disabled' : '';
            return `<option value="${i.value}"${dataIdAttr}${disabledAttr}>${i.label}</option>`;
        }).join('');
    }
}

// 更新计费预览显示
function updateCostPreview() {
    const modelSelect = document.getElementById('modelSelect');
    const costPreview = document.getElementById('costPreview');
    const numSelect = document.getElementById('numSelect');
    const isPremium = document.getElementById('isPremium')?.checked || false;
    const selectedOption = modelSelect.options[modelSelect.selectedIndex];

    if (currentMode === 'trial' && selectedOption) {
        let baseCost = parseInt(selectedOption.getAttribute('data-cost') || 0);
        let num = parseInt(numSelect?.value || 1);
        let totalCost = baseCost * num;
        if (isPremium) totalCost *= 2; // 优质模式 2 倍积分
        costPreview.innerText = `本次生成将消耗 ${totalCost} 积分`;
        costPreview.classList.remove('hidden');
    } else {
        costPreview.classList.add('hidden');
    }
}

// 渲染参考图预览
function renderImagePreviews() {
    const prev = document.getElementById('imagePreview');
    if (!prev) return;
    prev.innerHTML = '';

    uploadedFiles.forEach((file, index) => {
        // 同步创建容器，确保编号顺序永远正确
        const d = document.createElement('div');
        d.className = 'relative w-20 h-20 rounded-2xl overflow-hidden flex-shrink-0 border-2 border-white shadow-md group cursor-move';
        d.setAttribute('data-index', index);
        d.innerHTML = `
            <div class="w-full h-full bg-slate-100 animate-pulse flex items-center justify-center">
                <i data-lucide="image" class="w-5 h-5 text-slate-300"></i>
            </div>
            <div class="absolute bottom-1 left-1 bg-black/60 text-white text-[8px] px-1.5 py-0.5 rounded-md font-black backdrop-blur-sm z-10">图${index + 1}</div>
            <button onclick="removeUploadedFile(${index})" class="absolute top-1 right-1 bg-rose-500 text-white p-1 rounded-full opacity-0 group-hover:opacity-100 transition-opacity flex items-center justify-center z-20">
                <i data-lucide="x" class="w-3 h-3"></i>
            </button>
        `;
        prev.appendChild(d);

        // 异步加载图片内容
        const reader = new FileReader();
        reader.onload = (ev) => {
            d.innerHTML = `
                <img src="${ev.target.result}" class="w-full h-full object-cover">
                <div class="absolute bottom-1 left-1 bg-black/60 text-white text-[8px] px-1.5 py-0.5 rounded-md font-black backdrop-blur-sm z-10">图${index + 1}</div>
                <button onclick="removeUploadedFile(${index})" class="absolute top-1 right-1 bg-rose-500 text-white p-1 rounded-full opacity-0 group-hover:opacity-100 transition-opacity flex items-center justify-center z-20">
                    <i data-lucide="x" class="w-3 h-3"></i>
                </button>
            `;
            lucide.createIcons();
        };
        reader.readAsDataURL(file);
    });
    lucide.createIcons();
}

function removeUploadedFile(index) {
    uploadedFiles.splice(index, 1);
    renderImagePreviews();
}

// 统一处理新上传/粘贴/拖入的文件
function handleNewFiles(files) {
    const newFiles = Array.from(files).filter(f => f.type.startsWith('image/'));
    if (newFiles.length === 0) return;

    // 检查分辨率并提示压缩
    let largeFound = false;
    Promise.all(newFiles.map(file => new Promise(resolve => {
        const url = URL.createObjectURL(file);
        const img = new Image();
        img.onload = () => {
            if (img.width > 2048 || img.height > 2048) largeFound = true;
            URL.revokeObjectURL(url);
            resolve();
        };
        img.onerror = resolve;
        img.src = url;
    }))).then(() => {
        if (largeFound) showToast('检测到图片分辨率大于 2K，将为您自动压缩至 2K 分辨率', 'info');
    });

    // 如果处于设置器模式，严格限制为 1 张
    if (isSetterActive) {
        if (newFiles.length > 0) {
            uploadedFiles = [newFiles[0]];
            showToast('设置器模式已开启，仅保留第一张参考图', 'info');
        }
    } else {
        if (uploadedFiles.length + newFiles.length > 3) {
            showToast('最多只能上传 3 张参考图', 'warning');
            const remaining = 3 - uploadedFiles.length;
            if (remaining > 0) {
                uploadedFiles = uploadedFiles.concat(newFiles.slice(0, remaining));
            }
        } else {
            uploadedFiles = uploadedFiles.concat(newFiles);
        }
    }
    renderImagePreviews();
}

document.getElementById('fileInput').onchange = (e) => {
    handleNewFiles(e.target.files);
    e.target.value = ''; // 重置以允许重复选择同一文件
};

// 拖拽上传逻辑
const dropZone = document.getElementById('dropZone');
if (dropZone) {
    ['dragenter', 'dragover', 'dragleave', 'drop'].forEach(name => {
        dropZone.addEventListener(name, (e) => {
            e.preventDefault();
            e.stopPropagation();
        }, false);
    });

    ['dragenter', 'dragover'].forEach(name => {
        dropZone.addEventListener(name, () => dropZone.classList.add('border-indigo-500', 'bg-indigo-50/50'), false);
    });

    ['dragleave', 'drop'].forEach(name => {
        dropZone.addEventListener(name, () => dropZone.classList.remove('border-indigo-500', 'bg-indigo-50/50'), false);
    });

    dropZone.addEventListener('drop', (e) => {
        handleNewFiles(e.dataTransfer.files);
    }, false);
}

// 粘贴上传逻辑
document.addEventListener('paste', (e) => {
    const items = (e.clipboardData || e.originalEvent.clipboardData).items;
    const files = [];
    for (let item of items) {
        if (item.type.indexOf('image') !== -1) {
            files.push(item.getAsFile());
        }
    }
    if (files.length > 0) {
        handleNewFiles(files);
    }
});

const SIZE_MODELS = ['nano-banana-2', 'gemini-3.1-flash-image-preview'];

document.getElementById('modelSelect').onchange = (e) => {
    document.getElementById('sizeGroup').classList.toggle('hidden', !SIZE_MODELS.includes(e.target.value));
    updateCostPreview(); // 切换模型时更新计费预览
};

document.getElementById('promptTpl').onchange = (e) => {
    const area = document.getElementById('manualPrompt');
    const deleteContainer = document.getElementById('deletePromptContainer');
    const selectedOption = e.target.options[e.target.selectedIndex];

    // 判断是否是用户的收藏项 (通过 data-id 属性)
    const promptId = selectedOption.getAttribute('data-id');

    if (promptId) {
        deleteContainer.classList.remove('hidden');
        document.getElementById('deletePromptBtn').onclick = () => openDeleteConfirmModal(promptId);
    } else {
        deleteContainer.classList.add('hidden');
    }

    if (e.target.value !== 'manual') {
        area.value = e.target.value;
        area.classList.add('hidden');
    } else {
        area.value = '';
        area.classList.remove('hidden');
    }
};

document.getElementById('submitBtn').onclick = async () => {
    const btn = document.getElementById('submitBtn');
    const prompt = document.getElementById('manualPrompt').value;
    const apiKey = document.getElementById('apiKey').value;
    const num = parseInt(document.getElementById('numSelect').value);

    // 检查登录状态并获取积分
    const authCheck = await fetch('/api/auth/me');
    const authData = await authCheck.json();
    if (!authData.logged_in) {
        showToast('请先登录后再生成作品', 'warning');
        return;
    }

    // 根据模式验证
    if (currentMode === 'key') {
        if (!apiKey) return showToast('请输入您的 API 密钥', 'warning');
    } else {
        if (authData.points <= 0) return showToast('可用积分已耗尽，请充值或切换至自定义 Key 模式', 'warning');
    }

    // 允许文生图（不强制要求图片），但至少得有提示词或图片
    if (!prompt && uploadedFiles.length === 0) {
        return showToast('请至少输入提示词或上传参考图', 'warning');
    }

    // UI 锁定
    btn.disabled = true;
    const btnText = btn.querySelector('span');
    btnText.innerText = uploadedFiles.length > 0 ? "正在同步参考图..." : "正在开启 AI 引擎...";

    document.getElementById('statusInfo').classList.remove('hidden');
    document.getElementById('placeholder').classList.add('hidden');
    document.getElementById('finalWrapper').classList.remove('hidden');
    const grid = document.getElementById('imageGrid');
    grid.innerHTML = ''; // 清空
    currentGeneratedUrls = []; // 重置当前生成列表

    try {
        let image_data = [];

        // 1. 将图片转换为 Base64 (并压缩过大图片)
        if (uploadedFiles.length > 0) {
            btnText.innerText = "正在准备图片数据...";
            const processImageFile = (file) => new Promise((resolve, reject) => {
                const reader = new FileReader();
                reader.onload = (e) => {
                    const img = new Image();
                    img.onload = () => {
                        const maxDim = 2048;
                        let w = img.width;
                        let h = img.height;

                        if (w <= maxDim && h <= maxDim) {
                            resolve(e.target.result);
                            return;
                        }

                        if (w > h) {
                            if (w > maxDim) {
                                h = Math.round(h * (maxDim / w));
                                w = maxDim;
                            }
                        } else {
                            if (h > maxDim) {
                                w = Math.round(w * (maxDim / h));
                                h = maxDim;
                            }
                        }

                        const canvas = document.createElement('canvas');
                        canvas.width = w;
                        canvas.height = h;
                        const ctx = canvas.getContext('2d');
                        ctx.drawImage(img, 0, 0, w, h);
                        // 保持原格式，如果不是 png 则默认 jpeg (0.9 质量)
                        const outType = file.type === 'image/png' ? 'image/png' : 'image/jpeg';
                        resolve(canvas.toDataURL(outType, 0.9));
                    };
                    img.onerror = reject;
                    img.src = e.target.result;
                };
                reader.onerror = reject;
                reader.readAsDataURL(file);
            });
            image_data = await Promise.all(uploadedFiles.map(f => processImageFile(f)));
        }

        // 2. 并行启动多个生成任务
        btnText.innerText = `AI 构思中 (0/${num})...`;
        let finishedCount = 0;

        const startTask = async (index) => {
            const slot = document.createElement('div');
            slot.className = 'image-frame relative bg-white/50 animate-pulse min-h-[200px] flex items-center justify-center rounded-[2.5rem] border border-slate-100 shadow-sm';
            slot.innerHTML = `<div class="text-slate-400 text-[10px] font-bold italic">正在排队中...</div>`;
            grid.appendChild(slot);

            try {
                // 1. 发起生成请求，获取任务 ID
                const r = await fetch('/api/generate', {
                    method: 'POST',
                    headers: { 'Content-Type': 'application/json' },
                    body: JSON.stringify({
                        mode: currentMode,
                        is_premium: document.getElementById('isPremium')?.checked || false,
                        apiKey: currentMode === 'key' ? apiKey : '',
                        prompt: document.getElementById('manualPrompt').value,
                        model: document.getElementById('modelSelect').value,
                        ratio: document.getElementById('ratioSelect').value,
                        size: document.getElementById('sizeSelect').value,
                        image_data // 发送 Base64 数组
                    })
                });
                const res = await r.json();
                if (res.error) throw new Error(res.error);

                // 如果直接返回了 data (比如聊天模型)，直接显示
                if (res.data) {
                    displayResult(slot, res.data[0]);
                    return;
                }

                // 2. 轮询任务状态
                const taskId = res.task_id;
                let pollCount = 0;
                const maxPolls = 500; // 最多轮询约 16 分钟 (2s * 500 = 1000s)

                while (pollCount < maxPolls) {
                    await new Promise(resolve => setTimeout(resolve, 2000));
                    pollCount++;

                    const statusR = await fetch(`/api/task_status/${taskId}`);
                    const statusRes = await statusR.json();

                    if (statusRes.status === 'complete') {
                        const imgUrl = statusRes.urls[0];
                        currentGeneratedUrls.push(imgUrl);
                        displayResult(slot, { url: imgUrl });
                        finishedCount++;
                        btnText.innerText = `AI 构思中 (${finishedCount}/${num})...`;
                        if (currentMode === 'trial') checkAuth();
                        return; // 任务正常结束
                    } else if (statusRes.status === 'error') {
                        throw new Error(statusRes.message || "生成失败");
                    } else {
                        // 继续轮询状态显示
                        slot.innerHTML = `<div class="text-slate-400 text-[10px] font-bold italic">AI 正在努力创作中 (${pollCount * 2}s)...</div>`;
                    }
                }
                throw new Error("生成超时，请稍后在历史记录中查看");

            } catch (e) {
                slot.className = 'image-frame relative bg-red-50/50 flex items-center justify-center rounded-[2.5rem] border border-red-100 p-6';
                if (e.message.includes('401') || e.message.includes('请先登录')) {
                    slot.innerHTML = `<div class="text-red-400 text-[10px] font-bold text-center">登录已过期，请重新登录</div>`;
                } else {
                    slot.innerHTML = `<div class="text-red-400 text-[10px] font-bold text-center">生成异常: ${e.message}</div>`;
                }
            }
        };

        const tasks = Array.from({ length: num }, (_, i) => startTask(i));
        await Promise.all(tasks);

    } catch (e) {
        showToast('创作引擎中断: ' + e.message, 'error');
        document.getElementById('placeholder').classList.remove('hidden');
    } finally {
        btn.disabled = false;
        btnText.innerText = "立即生成作品";
        document.getElementById('statusInfo').classList.add('hidden');
    }
};

// 提取结果展示逻辑
const displayResult = (slot, data) => {
    if (data.type === 'text') {
        slot.className = 'image-frame relative bg-white border border-slate-100 p-8 rounded-[2.5rem] shadow-xl overflow-y-auto max-h-[600px]';
        slot.innerHTML = `<div class="prose prose-slate prose-sm max-w-none text-slate-600 font-medium leading-relaxed">${data.content.replace(/\n/g, '<br>')}</div>`;
    } else {
        const imgUrl = data.url;
        slot.className = 'image-frame group relative animate-in zoom-in-95 duration-700 flex flex-col items-center justify-center overflow-hidden bg-white shadow-2xl transition-all hover:shadow-indigo-100/50';
        slot.innerHTML = `
            <div class="w-full h-full flex items-center justify-center bg-slate-50/20 p-2">
                <img src="${imgUrl}" class="max-w-full max-h-[60vh] md:max-h-[70vh] object-contain rounded-2xl shadow-sm transition-transform duration-500 group-hover:scale-[1.01]">
            </div>
            <div class="absolute inset-0 flex items-center justify-center opacity-0 group-hover:opacity-100 transition-opacity bg-slate-900/40 backdrop-blur-[2px] rounded-[2.5rem]">
                <button onclick="downloadImage('${imgUrl}')" class="bg-white/90 p-4 rounded-full text-indigo-600 shadow-xl hover:scale-110 transition-transform">
                    <i data-lucide="download-cloud" class="w-6 h-6"></i>
                </button>
            </div>
        `;
    }
    lucide.createIcons();
};

init();

// 修改密码弹窗控制
function openPwdModal() {
    const modal = document.getElementById('pwdModal');
    if (!modal) return;
    modal.classList.remove('hidden');
    setTimeout(() => {
        modal.classList.add('opacity-100');
        modal.querySelector('div').classList.remove('scale-95');
    }, 10);
}

function closePwdModal() {
    const modal = document.getElementById('pwdModal');
    if (!modal) return;
    modal.classList.remove('opacity-100');
    modal.querySelector('div').classList.add('scale-95');
    setTimeout(() => {
        modal.classList.add('hidden');
        document.getElementById('pwdForm').reset();
    }, 300);
}

document.addEventListener('click', (e) => {
    if (e.target.closest('#openPwdModalBtn')) {
        openPwdModal();
    }
});

document.getElementById('pwdForm')?.addEventListener('submit', async (e) => {
    e.preventDefault();
    const old_password = document.getElementById('oldPwd').value;
    const new_password = document.getElementById('newPwd').value;

    try {
        const r = await fetch('/api/auth/change_password', {
            method: 'POST',
            headers: { 'Content-Type': 'application/json' },
            body: JSON.stringify({ old_password, new_password })
        });
        const d = await r.json();
        if (r.ok) {
            showToast('密码修改成功，请记牢新密码', 'success');
            closePwdModal();
        } else {
            showToast(d.error || '修改失败', 'error');
        }
    } catch (err) {
        showToast('网络连接失败', 'error');
    }
});

// 拍摄角度设置器弹窗控制
function openVisualizerModal() {
    const modal = document.getElementById('visualizerModal');
    if (!modal) return;

    // 检查是否已上传参考图
    if (uploadedFiles.length === 0) {
        return showToast('请先上传一张参考图作为基准', 'warning');
    }

    modal.classList.remove('hidden');
    setTimeout(() => {
        modal.classList.add('opacity-100');
        modal.querySelector('div').classList.remove('scale-95');

        // 将主页面的图片同步到设置器 iframe
        const reader = new FileReader();
        reader.onload = (e) => {
            const iframe = document.getElementById('visualizerFrame');
            iframe.contentWindow.postMessage({ type: 'sync_image', dataUrl: e.target.result }, '*');
        };
        reader.readAsDataURL(uploadedFiles[0]);
    }, 10);
}

function closeVisualizerModal() {
    const modal = document.getElementById('visualizerModal');
    if (!modal) return;
    modal.classList.remove('opacity-100');
    modal.querySelector('div').classList.add('scale-95');
    setTimeout(() => {
        modal.classList.add('hidden');
    }, 300);
}

// 监听来自设置器的消息
window.addEventListener('message', (e) => {
    if (e.data.type === 'apply_prompt') {
        isSetterActive = true;

        const area = document.getElementById('manualPrompt');
        const promptTpl = document.getElementById('promptTpl');
        const modelSelect = document.getElementById('modelSelect');
        const sizeGroup = document.getElementById('sizeGroup');

        // 1. 强制切换并锁定版本 2.0 (nano-banana-2)
        if (modelSelect) {
            modelSelect.value = 'nano-banana-2';
            modelSelect.disabled = true; // 锁定选择
            sizeGroup.classList.remove('hidden');
            updateCostPreview();
        }

        // 2. 隐藏模板选择器
        if (promptTpl) {
            promptTpl.classList.add('hidden');
            promptTpl.value = 'manual';
        }

        // 3. 强制图片数量为 1
        if (uploadedFiles.length > 1) {
            uploadedFiles = uploadedFiles.slice(0, 1);
            renderImagePreviews();
        }

        // 4. 替换提示词
        if (area) {
            area.value = e.data.prompt;
            area.classList.remove('hidden');
            showToast('已同步拍摄角度并切换至 2.0 引擎', 'success');
        }

        closeVisualizerModal();
    }
});

// --- 积分与钱包中心逻辑 (Modern Modal Version) ---
let pointsChart = null;

async function openPointsModal() {
    const modal = document.getElementById('pointsModal');
    if (!modal) return;
    modal.classList.remove('hidden');
    setTimeout(() => {
        modal.classList.add('opacity-100');
        modal.querySelector('div').classList.remove('scale-95');
    }, 10);

    // 加载核心数据
    loadPointStats();
    loadPointDetails(1);
    loadInviteStatsModal();

    // 实时更新余额显示
    const r = await fetch('/api/auth/me');
    const d = await r.json();
    if (d.logged_in) {
        document.getElementById('modalPointsDisplay').innerText = d.points;
    }
}

function closePointsModal() {
    const modal = document.getElementById('pointsModal');
    if (!modal) return;
    modal.classList.remove('opacity-100');
    modal.querySelector('div').classList.add('scale-95');
    setTimeout(() => modal.classList.add('hidden'), 300);
}

async function loadPointStats() {
    try {
        const r = await fetch('/api/auth/point_stats');
        const d = await r.json();

        const canvas = document.getElementById('modalTrendChart');
        if (!canvas) return;
        const ctx = canvas.getContext('2d');

        if (pointsChart) pointsChart.destroy();

        pointsChart = new Chart(ctx, {
            type: 'line',
            data: {
                labels: d.labels,
                datasets: [
                    {
                        label: '消耗积分',
                        data: d.consumption,
                        borderColor: '#6366f1',
                        backgroundColor: 'rgba(99, 102, 241, 0.05)',
                        borderWidth: 3,
                        fill: true,
                        tension: 0.4,
                        pointRadius: 4,
                        pointBackgroundColor: '#6366f1'
                    },
                    {
                        label: '充值积分',
                        data: d.recharge,
                        borderColor: '#10b981',
                        backgroundColor: 'rgba(16, 185, 129, 0.05)',
                        borderWidth: 3,
                        fill: true,
                        tension: 0.4,
                        pointRadius: 4,
                        pointBackgroundColor: '#10b981'
                    }
                ]
            },
            options: {
                responsive: true,
                maintainAspectRatio: false,
                plugins: {
                    legend: { display: false },
                    tooltip: {
                        backgroundColor: '#1e293b',
                        padding: 12,
                        cornerRadius: 12,
                        titleFont: { weight: 'bold' },
                        bodyFont: { weight: '900' }
                    }
                },
                scales: {
                    y: {
                        beginAtZero: true,
                        grid: { color: '#f1f5f9' },
                        ticks: { font: { weight: 'bold', size: 10 }, color: '#94a3b8' }
                    },
                    x: {
                        grid: { display: false },
                        ticks: { font: { weight: 'bold', size: 10 }, color: '#94a3b8' }
                    }
                }
            }
        });
    } catch (e) {
        console.error('加载统计图失败:', e);
    }
}

async function loadPointDetails(page = 1) {
    const body = document.getElementById('pointDetailsBody');
    const pagContainer = document.getElementById('pointDetailsPagination');
    if (!body) return;

    body.innerHTML = '<tr><td colspan="4" class="px-8 py-10 text-center text-slate-400">正在同步账户流水...</td></tr>';

    try {
        const r = await fetch(`/api/auth/point_history?page=${page}&per_page=5`);
        const d = await r.json();

        if (d.details.length === 0) {
            body.innerHTML = '<tr><td colspan="4" class="px-8 py-10 text-center text-slate-300 italic">暂无资金变动记录</td></tr>';
            if (pagContainer) pagContainer.innerHTML = '';
            return;
        }

        body.innerHTML = d.details.map(item => {
            const isAdd = item.amount.startsWith('+');
            const colorClass = isAdd ? 'text-emerald-500' : 'text-rose-500';
            const iconColor = isAdd ? 'bg-emerald-500' : 'bg-rose-500';

            return `
            <tr class="hover:bg-slate-50 transition-colors">
                <td class="px-8 py-5">
                    <div class="flex items-center gap-3 text-slate-700">
                        <div class="w-2 h-2 rounded-full ${iconColor}"></div>
                        <div class="flex flex-col">
                            <span class="font-bold text-xs">${item.type_label}</span>
                            <span class="text-[10px] text-slate-400 truncate max-w-[200px]">${item.desc}</span>
                        </div>
                    </div>
                </td>
                <td class="px-8 py-5">
                    <span class="px-2 py-0.5 bg-slate-100 text-slate-500 rounded text-[9px] font-black uppercase tracking-tighter border border-slate-200">
                        ${item.type === 'consumption' ? 'MODEL' : 'SYSTEM'}
                    </span>
                </td>
                <td class="px-8 py-5">
                    <span class="${colorClass} font-black">${item.amount}</span>
                </td>
                <td class="px-8 py-5 text-right font-bold text-slate-300 text-[10px] tracking-tighter">${item.time}</td>
            </tr>
            `;
        }).join('');

        // 渲染分页
        if (pagContainer) {
            if (d.pages > 1) {
                let html = '';
                if (d.current_page > 1) {
                    html += `<button onclick="loadPointDetails(${d.current_page - 1})" class="px-3 py-1.5 bg-white border border-slate-100 rounded-lg text-[10px] font-black hover:text-indigo-600 transition-all">Prev</button>`;
                }
                html += `<span class="px-4 py-1.5 text-[9px] font-black text-slate-400 uppercase">Page ${d.current_page} / ${d.pages}</span>`;
                if (d.current_page < d.pages) {
                    html += `<button onclick="loadPointDetails(${d.current_page + 1})" class="px-3 py-1.5 bg-white border border-slate-100 rounded-lg text-[10px] font-black hover:text-indigo-600 transition-all">Next</button>`;
                }
                pagContainer.innerHTML = html;
            } else {
                pagContainer.innerHTML = '';
            }
        }
    } catch (e) {
        body.innerHTML = '<tr><td colspan="4" class="px-8 py-10 text-center text-rose-400 font-bold">流水加载异常</td></tr>';
    }
}

async function loadInviteStatsModal() {
    try {
        const r = await fetch('/api/auth/invite_stats');
        const d = await r.json();
        if (d.invite_code) {
            document.getElementById('modalInviteCode').innerText = d.invite_code;
            document.getElementById('modalInvitedCount').innerText = d.invited_count;
            document.getElementById('modalTotalRewards').innerText = d.total_rewards;
        }
    } catch (e) { console.error('加载邀请统计失败', e); }
}

function copyInviteCodeModal() {
    const code = document.getElementById('modalInviteCode').innerText;
    if (code === '---') return;
    const link = `${window.location.origin}/login?invite_code=${code}`;
    navigator.clipboard.writeText(link).then(() => showToast('专属邀请链接已复制', 'success'));
}

function scrollToModalRecharge() {
    const area = document.getElementById('modalRechargeArea');
    if (area) area.scrollIntoView({ behavior: 'smooth' });
}

function submitModalRecharge(pkgId) {
    const form = document.getElementById('modalRechargeForm');
    const input = document.getElementById('modalPackageId');
    if (form && input) {
        input.value = pkgId;
        form.submit();
    }
}

// --- 自定义提示词逻辑 ---
async function loadUserPrompts() {
    try {
        const r = await fetch('/api/prompts');
        const d = await r.json();
        return d.error ? [] : d;
    } catch (e) {
        return [];
    }
}

async function refreshPromptsList() {
    try {
        const r = await fetch('/api/config');
        const d = await r.json();

        fillSelect('modelSelect', d.models);
        fillSelect('ratioSelect', d.ratios);
        fillSelect('sizeSelect', d.sizes);

        // 初始化后检查默认模型是否需要显示尺寸选择器
        const modelSelect = document.getElementById('modelSelect');
        if (modelSelect && typeof SIZE_MODELS !== 'undefined') {
            document.getElementById('sizeGroup').classList.toggle('hidden', !SIZE_MODELS.includes(modelSelect.value));
        }

        const userPrompts = await loadUserPrompts();

        const mergedPrompts = [
            { label: '✨ 自定义创作', value: 'manual' },
            ...(userPrompts.length > 0 ? [{ label: '--- 我的收藏 ---', value: 'manual', disabled: true }] : []),
            ...userPrompts.map(p => ({ id: p.id, label: '⭐ ' + p.label, value: p.value })),
            { label: '--- 系统预设 ---', value: 'manual', disabled: true },
            ...d.prompts
        ];

        fillSelect('promptTpl', mergedPrompts);
    } catch (e) { console.error(e); }
}

function openSavePromptModal() {
    // Check login by checking visibility of login button
    const loginBtn = document.getElementById('loginEntryBtn');
    if (loginBtn && !loginBtn.classList.contains('hidden')) {
        showToast('请先登录', 'warning');
        return;
    }

    const modal = document.getElementById('savePromptModal');
    const input = document.getElementById('promptTitleInput');
    const promptArea = document.getElementById('manualPrompt');

    if (!promptArea) return;
    const prompt = promptArea.value;

    if (!prompt || !prompt.trim()) {
        showToast('请先输入一些提示词内容', 'warning');
        return;
    }

    if (modal) {
        // 使用内联样式强制显示，避免 Tailwind 类可能未编译的问题
        modal.classList.remove('hidden');
        modal.style.display = 'flex';
        modal.style.zIndex = '9999';

        // 重置状态
        if (input) input.value = ''; // Clear input immediately

        // 强制重绘
        void modal.offsetWidth;

        // 简单直接的显示
        modal.style.transition = 'opacity 0.3s ease';
        modal.style.opacity = '1';

        const innerDiv = modal.querySelector('div');
        if (innerDiv) {
            innerDiv.style.transition = 'transform 0.3s ease';
            innerDiv.classList.remove('scale-95');
            innerDiv.classList.add('scale-100');
            // 同样用内联样式兜底
            innerDiv.style.transform = 'scale(1)';
        }

        // 延时聚焦
        setTimeout(() => {
            if (input) {
                try { input.focus(); } catch (e) { }
            }
        }, 50);
    }
}

function closeSavePromptModal() {
    const modal = document.getElementById('savePromptModal');
    if (modal) {
        // 先触发过渡动画
        modal.style.opacity = '0';

        const innerDiv = modal.querySelector('div');
        if (innerDiv) {
            innerDiv.style.transform = 'scale(0.95)';
        }

        // 等待动画结束后隐藏
        setTimeout(() => {
            modal.style.display = 'none';
            modal.classList.add('hidden');
            // 清理样式
            modal.style.zIndex = '';
        }, 300);
    }
}

async function confirmSavePrompt() {
    const title = document.getElementById('promptTitleInput').value.trim();
    const prompt = document.getElementById('manualPrompt').value.trim();

    if (!title) {
        showToast('请输入标题', 'warning');
        return;
    }

    try {
        const r = await fetch('/api/prompts', {
            method: 'POST',
            headers: { 'Content-Type': 'application/json' },
            body: JSON.stringify({ title, prompt })
        });
        const d = await r.json();

        if (r.ok) {
            showToast('收藏成功', 'success');
            closeSavePromptModal();
            refreshPromptsList();
        } else {
            showToast(d.error || '保存失败', 'error');
        }
    } catch (e) {
        showToast('保存异常', 'error');
    }
}

async function deleteUserPrompt(id) {
    // Legacy function replaced by openDeleteConfirmModal
    // Kept for reference or simple fallback if needed, but logic moved to executeDeletePrompt
}

function openDeleteConfirmModal(id) {
    const modal = document.getElementById('deleteConfirmModal');
    const confirmBtn = document.getElementById('confirmDeleteBtn');

    if (modal && confirmBtn) {
        // Store ID on the button
        confirmBtn.setAttribute('data-delete-id', id);
        confirmBtn.onclick = () => executeDeletePrompt(id); // Bind click

        modal.classList.remove('hidden');
        setTimeout(() => {
            modal.classList.add('opacity-100');
            modal.querySelector('div').classList.remove('scale-95');
        }, 10);
    }
}

function closeDeleteConfirmModal() {
    const modal = document.getElementById('deleteConfirmModal');
    if (modal) {
        modal.classList.remove('opacity-100');
        modal.querySelector('div').classList.add('scale-95');
        setTimeout(() => modal.classList.add('hidden'), 300);
    }
}

async function executeDeletePrompt(id) {
    const confirmBtn = document.getElementById('confirmDeleteBtn');
    const originalText = confirmBtn.innerHTML;
    confirmBtn.disabled = true;
    confirmBtn.innerHTML = '<i data-lucide="loader-2" class="w-4 h-4 animate-spin"></i>';
    lucide.createIcons();

    try {
        const r = await fetch(`/api/prompts/${id}`, {
            method: 'DELETE'
        });

        if (r.ok) {
            showToast('已删除', 'success');
            // Reset selection
            document.getElementById('promptTpl').value = 'manual';
            document.getElementById('manualPrompt').value = '';
            document.getElementById('manualPrompt').classList.remove('hidden');
            document.getElementById('deletePromptContainer').classList.add('hidden');

            closeDeleteConfirmModal();
            refreshPromptsList();
        } else {
            showToast('删除失败', 'error');
        }
    } catch (e) {
        showToast('删除异常', 'error');
    } finally {
        confirmBtn.disabled = false;
        confirmBtn.innerHTML = originalText;
    }
}