fix(chat): optimize chat with pic response (ValueCell-ai#103)

Author: hazeone

electron/main/ipc-handlers.ts

@@ -1506,6 +1506,40 @@ function registerFileHandlers(): void {
   });
 
   // Load thumbnails for file paths on disk (used to restore previews in history)
+  // Save an image to a user-chosen location (base64 data URI or existing file path)
+  ipcMain.handle('media:saveImage', async (_, params: {
+    base64?: string;
+    mimeType?: string;
+    filePath?: string;
+    defaultFileName: string;
+  }) => {
+    try {
+      const ext = params.defaultFileName.includes('.')
+        ? params.defaultFileName.split('.').pop()!
+        : (params.mimeType?.split('/')[1] || 'png');
+      const result = await dialog.showSaveDialog({
+        defaultPath: join(homedir(), 'Downloads', params.defaultFileName),
+        filters: [
+          { name: 'Images', extensions: [ext, 'png', 'jpg', 'jpeg', 'webp', 'gif'] },
+          { name: 'All Files', extensions: ['*'] },
+        ],
+      });
+      if (result.canceled || !result.filePath) return { success: false };
+
+      if (params.filePath && existsSync(params.filePath)) {
+        copyFileSync(params.filePath, result.filePath);
+      } else if (params.base64) {
+        const buffer = Buffer.from(params.base64, 'base64');
+        writeFileSync(result.filePath, buffer);
+      } else {
+        return { success: false, error: 'No image data provided' };
+      }
+      return { success: true, savedPath: result.filePath };
+    } catch (err) {
+      return { success: false, error: String(err) };
+    }
+  });
+
   ipcMain.handle('media:getThumbnails', async (_, paths: Array<{ filePath: string; mimeType: string }>) => {
     const results: Record<string, { preview: string | null; fileSize: number }> = {};
     for (const { filePath, mimeType } of paths) {

electron/preload/index.ts

@@ -116,6 +116,7 @@ const electronAPI = {
         'file:stage',
         'file:stageBuffer',
         'media:getThumbnails',
+        'media:saveImage',
         // Chat send with media (reads staged files in main process)
         'chat:sendWithMedia',
         // OpenClaw extras