From febccfc86399d5e41d446036e9a265328bdf3a2e Mon Sep 17 00:00:00 2001
From: shiyue <935732994@qq.com>
Date: Fri, 29 Aug 2025 18:07:03 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=9C=A8utils=E4=B8=AD=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0base64ToBlobUrl=E5=87=BD=E6=95=B0=E4=BB=A5=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E5=9B=BE=E7=89=87=E6=98=BE=E7=A4=BA=E5=92=8C=E4=B8=8B?=
 =?UTF-8?q?=E8=BD=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script.js | 96 +++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 87 insertions(+), 9 deletions(-)

diff --git a/script.js b/script.js
index e1f8adc..2b44a35 100644
--- a/script.js
+++ b/script.js
@@ -50,6 +50,45 @@ const ERROR_MESSAGES = {
 
 // 工具函数
 const utils = {
+    // 将base64转换为Blob URL
+    base64ToBlobUrl: function(base64Data) {
+        try {
+            // 检查是否已经是Blob URL
+            if (base64Data.startsWith('blob:')) {
+                return base64Data;
+            }
+            
+            // 提取base64数据的MIME类型和纯数据部分
+            const parts = base64Data.split(';base64,');
+            if (parts.length !== 2) {
+                console.warn('Invalid base64 format, returning original data');
+                return base64Data;
+            }
+            
+            const contentType = parts[0].split(':')[1];
+            const byteCharacters = atob(parts[1]);
+            
+            // 转换为ArrayBuffer
+            const byteNumbers = new Array(byteCharacters.length);
+            for (let i = 0; i < byteCharacters.length; i++) {
+                byteNumbers[i] = byteCharacters.charCodeAt(i);
+            }
+            const byteArray = new Uint8Array(byteNumbers);
+            
+            // 创建Blob和URL
+            const blob = new Blob([byteArray], { type: contentType });
+            const blobUrl = URL.createObjectURL(blob);
+            
+            // 存储原始base64数据以便后续使用（如下载）
+            blobUrl._originalBase64 = base64Data;
+            
+            return blobUrl;
+        } catch (error) {
+            console.error('Error converting base64 to Blob URL:', error);
+            return base64Data; // 出错时返回原始数据
+        }
+    },
+
     // 格式化文件大小
     formatFileSize: function(bytes) {
         if (bytes === 0) return '0 Bytes';
@@ -669,10 +708,11 @@ const apiService = {
             const messageContent = choice.message.content;
             const images = [];
             
-            // 提取图像数据
+            // 提取图像数据并转换为Blob URL
             if (choice.message.images) {
                 choice.message.images.forEach(img => {
-                    images.push(img.image_url.url);
+                    const blobUrl = utils.base64ToBlobUrl(img.image_url.url);
+                    images.push(blobUrl);
                 });
             }
 
@@ -773,8 +813,12 @@ const uiController = {
         imageDiv.className = 'image-item fade-in';
         
         const imageId = Date.now() + Math.random();
+        
+        // 确保imageUrl是Blob URL
+        const displayUrl = imageUrl.startsWith('blob:') ? imageUrl : utils.base64ToBlobUrl(imageUrl);
+        
         imageDiv.innerHTML = `
-            <img src="${imageUrl}" alt="Generated Image" loading="lazy" class="generated-image">
+            <img src="${displayUrl}" alt="Generated Image" loading="lazy" class="generated-image">
             <div class="image-overlay">
                 <div class="btn-grid">
                     <div class="btn-grid-row">
@@ -884,8 +928,11 @@ const uiController = {
         placeholder.className = 'image-item fade-in';
         placeholder.id = '';
         
+        // 确保imageUrl是Blob URL
+        const displayUrl = imageUrl.startsWith('blob:') ? imageUrl : utils.base64ToBlobUrl(imageUrl);
+        
         placeholder.innerHTML = `
-            <img src="${imageUrl}" alt="Generated Image" loading="lazy" class="generated-image">
+            <img src="${displayUrl}" alt="Generated Image" loading="lazy" class="generated-image">
             <div class="image-overlay">
                 <div class="btn-grid">
                     <div class="btn-grid-row">
@@ -1319,9 +1366,13 @@ const uiController = {
     // 显示图像预览
     displayImagePreview: function(imageData) {
         const preview = document.getElementById('imagePreview');
+        
+        // 确保图像数据是Blob URL
+        const displayUrl = imageData.data.startsWith('blob:') ? imageData.data : utils.base64ToBlobUrl(imageData.data);
+        
         preview.innerHTML = `
             <div class="card h-100">
-                <img src="${imageData.data}" class="card-img-top image-preview h-75" alt="${imageData.name}" loading="lazy" style="object-fit: contain;">
+                <img src="${displayUrl}" class="card-img-top image-preview h-75" alt="${imageData.name}" loading="lazy" style="object-fit: contain;">
                 <div class="card-body p-2">
                     <div class="d-flex justify-content-between align-items-center">
                         <small class="text-muted">${utils.formatFileSize(imageData.size)}</small>
@@ -1942,10 +1993,37 @@ const app = {
 
     // 下载图像
     downloadImage: function(imageUrl) {
-        const link = document.createElement('a');
-        link.href = imageUrl;
-        link.download = `generated-image-${Date.now()}.png`;
-        link.click();
+        // 检查是否是Blob URL并且有原始base64数据
+        if (imageUrl.startsWith('blob:') && imageUrl._originalBase64) {
+            // 使用原始base64数据下载
+            const link = document.createElement('a');
+            link.href = imageUrl._originalBase64;
+            link.download = `generated-image-${Date.now()}.png`;
+            link.click();
+        } else if (imageUrl.startsWith('blob:')) {
+            // 如果没有原始base64数据，通过fetch获取Blob数据
+            fetch(imageUrl)
+                .then(response => response.blob())
+                .then(blob => {
+                    const url = URL.createObjectURL(blob);
+                    const link = document.createElement('a');
+                    link.href = url;
+                    link.download = `generated-image-${Date.now()}.png`;
+                    link.click();
+                    // 清理临时URL
+                    setTimeout(() => URL.revokeObjectURL(url), 100);
+                })
+                .catch(error => {
+                    console.error('Error downloading image:', error);
+                    utils.showNotification('下载图像失败', 'danger');
+                });
+        } else {
+            // 如果是普通URL，直接下载
+            const link = document.createElement('a');
+            link.href = imageUrl;
+            link.download = `generated-image-${Date.now()}.png`;
+            link.click();
+        }
     },
 
     // 全部下载图像