From 78bf76e965ab187dd0686126b62bc545286fba47 Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Thu, 9 Apr 2026 10:23:04 +0800
Subject: [PATCH] fix: python code for eidting image

---
 src/pages/playground/view-code/image.ts | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/pages/playground/view-code/image.ts b/src/pages/playground/view-code/image.ts
index 4fbb87fd..f10aae5d 100644
--- a/src/pages/playground/view-code/image.ts
+++ b/src/pages/playground/view-code/image.ts
@@ -52,7 +52,7 @@ export const generateImageCode = ({
   });
 
   // ========================= Python =========================
-  const pythonCode = `
+  let pythonCode = `
 import requests\n
 url="${host}${api}"
 headers = {
@@ -63,6 +63,25 @@ data = ${JSON.stringify(parameters, null, 2).replace(/null/g, 'None')}\n
 response = requests.post(url, headers=headers, json=data)
 print(response.json()['data'][0]['b64_json'])`.trim();
 
+  if (edit) {
+    const files = {
+      image: `open(image.png, 'rb')`,
+      mask: `open(mask.png, 'rb')`
+    };
+
+    pythonCode = `
+import requests\n
+url="${host}${api}"
+headers = {
+  "Content-type": "multipart/form-data",
+  "Authorization": "Bearer $\{YOUR_GPUSTACK_API_KEY}"
+}
+data = ${JSON.stringify(_.omit(parameters, ['mask', 'image']), null, 2).replace(/null/g, 'None')}
+files = ${JSON.stringify(files, null, 2).replace(/null/g, 'None')}\n
+response = requests.post(url, headers=headers, json=data, files=files)
+print(response.json()['data'][0]['b64_json'])`.trim();
+  }
+
   // ========================= Node.js =========================
   const nodeJsCode = `
 const axios = require('axios');