Upload 13 files

index.html

@@ -379,6 +379,7 @@
 
     };
     const IMAGE_SIZE_STORAGE_KEY = 'BananaPro_ImageSize';
+    const IMAGE_RESPONSE_STORAGE_KEY = 'BananaPro_ImageResponse';
     const API_BASE_STORAGE_KEY = 'BananaPro_ApiBase';
 
     const resolveApiBase = () => {
@@ -423,6 +424,27 @@
 
     const IMAGE_SIZE = resolveImageSize();
 
+    const normalizeResponseMode = (value, fallback) => {
+        const normalized = String(value || '').trim().toLowerCase();
+        if (normalized === 'url' || normalized === 'base64' || normalized === 'both') {
+            return normalized;
+        }
+        return fallback;
+    };
+
+    const resolveResponseMode = () => {
+        const params = new URLSearchParams(window.location.search);
+        const paramMode = params.get('response');
+        if (paramMode) {
+            localStorage.setItem(IMAGE_RESPONSE_STORAGE_KEY, paramMode);
+        }
+        const stored = localStorage.getItem(IMAGE_RESPONSE_STORAGE_KEY);
+        const fallback = Device.isMobile ? 'url' : 'base64';
+        return normalizeResponseMode(paramMode || stored, fallback);
+    };
+
+    const RESPONSE_MODE = resolveResponseMode();
+
     const resolveAssetUrl = (value) => {
         if (!value || typeof value !== 'string') return '';
         if (/^https?:\/\//i.test(value) || value.startsWith('data:') || value.startsWith('blob:')) {
@@ -3005,10 +3027,13 @@
 
 
 
+                const wantsUrl = RESPONSE_MODE !== 'base64';
+                const wantsBase64 = RESPONSE_MODE !== 'url';
                 const requestBody = {
                     prompt: prompt,
                     images: AppState.currentImages,
-                    preferUrl: false,
+                    preferUrl: wantsUrl,
+                    returnBase64: wantsBase64,
                     imageSize: IMAGE_SIZE
                 };
 

server.js

@@ -648,7 +648,7 @@ app.get('/api/check-auth', (req, res) => {
 
 // Generate image (multi-image)
 app.post('/api/generate', authMiddleware, async (req, res) => {
-    const { prompt, images, preferUrl, imageSize, aspectRatio } = req.body;
+    const { prompt, images, preferUrl, imageSize, aspectRatio, returnBase64 } = req.body;
     const requestStart = Date.now();
     StatsStore.activeRequests += 1;
     let recorded = false;
@@ -717,6 +717,7 @@ app.post('/api/generate', authMiddleware, async (req, res) => {
         const apiResponse = await APIService.generateImage(trimmedPrompt, uploadedImages, apiOptions);
         const imageData = APIService.extractImageFromResponse(apiResponse);
         const wantsUrl = Boolean(preferUrl);
+        const wantsBase64 = returnBase64 !== false;
         let imageUrl = null;
         let imageId = null;
 
@@ -739,6 +740,10 @@ app.post('/api/generate', authMiddleware, async (req, res) => {
         console.log(`[${new Date().toISOString()}] Image generated`);
 
         recordOnce(true);
+        if (wantsUrl && !imageUrl && !wantsBase64) {
+            throw new Error('生成图片保存失败，请稍后重试');
+        }
+
         const responsePayload = {
             success: true,
             imageUrl,
@@ -748,7 +753,7 @@ app.post('/api/generate', authMiddleware, async (req, res) => {
             timestamp: new Date().toISOString()
         };
 
-        if (!wantsUrl || !imageUrl) {
+        if ((!wantsUrl || !imageUrl) && wantsBase64) {
             responsePayload.image = imageData;
         }