From b3643944f3caa9f700d3c50cea81d2f3b44f43c8 Mon Sep 17 00:00:00 2001
From: Caelan <79105826+jin-wang-c@users.noreply.github.com>
Date: Sat, 16 Aug 2025 23:32:26 +0800
Subject: [PATCH] feat(DMXAPI): new add painting seededit (#9226)

adjust api
---
 .../src/pages/paintings/DmxapiPage.tsx        | 21 ++++++++++++++-----
 .../pages/paintings/config/DmxapiConfig.ts    |  2 +-
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/src/renderer/src/pages/paintings/DmxapiPage.tsx b/src/renderer/src/pages/paintings/DmxapiPage.tsx
index a829320ddd..d9167ac69e 100644
--- a/src/renderer/src/pages/paintings/DmxapiPage.tsx
+++ b/src/renderer/src/pages/paintings/DmxapiPage.tsx
@@ -13,7 +13,7 @@ import FileManager from '@renderer/services/FileManager'
 import { useAppDispatch } from '@renderer/store'
 import { setGenerating } from '@renderer/store/runtime'
 import type { FileMetadata, PaintingsState } from '@renderer/types'
-import { uuid } from '@renderer/utils'
+import { convertToBase64, uuid } from '@renderer/utils'
 import { DmxapiPainting } from '@types'
 import { Avatar, Button, Input, InputNumber, Segmented, Select, Switch, Tooltip } from 'antd'
 import TextArea from 'antd/es/input/TextArea'
@@ -364,7 +364,7 @@ const DmxapiPage: FC<{ Options: string[] }> = ({ Options }) => {
   }
 
   // 准备V1生成请求函数
-  const prepareV1GenerateRequest = (prompt: string, painting: DmxapiPainting) => {
+  const prepareV1GenerateRequest = async (prompt: string, painting: DmxapiPainting) => {
     const params = {
       prompt,
       model: painting.model,
@@ -391,6 +391,13 @@ const DmxapiPage: FC<{ Options: string[] }> = ({ Options }) => {
       params.prompt = prompt + ',风格：' + painting.style_type
     }
 
+    if (Array.isArray(fileMap.imageFiles) && fileMap.imageFiles.length > 0) {
+      const imageFile = fileMap.imageFiles[0]
+      if (imageFile instanceof File) {
+        params['image'] = await convertToBase64(imageFile)
+      }
+    }
+
     return {
       body: JSON.stringify(params),
       headerExpand: headerExpand,
@@ -508,13 +515,17 @@ const DmxapiPage: FC<{ Options: string[] }> = ({ Options }) => {
   }
 
   // 准备请求配置函数
-  const prepareRequestConfig = (prompt: string, painting: DmxapiPainting) => {
+  const prepareRequestConfig = async (prompt: string, painting: DmxapiPainting) => {
     // 根据模式和模型版本返回不同的请求配置
     if (
       painting.generationMode !== undefined &&
       [generationModeType.MERGE, generationModeType.EDIT].includes(painting.generationMode)
     ) {
-      return prepareV2GenerateRequest(prompt, painting)
+      if (painting.model === 'seededit-3.0') {
+        return await prepareV1GenerateRequest(prompt, painting)
+      } else {
+        return prepareV2GenerateRequest(prompt, painting)
+      }
     } else {
       return prepareV1GenerateRequest(prompt, painting)
     }
@@ -550,7 +561,7 @@ const DmxapiPage: FC<{ Options: string[] }> = ({ Options }) => {
       dispatch(setGenerating(true))
 
       // 准备请求配置
-      const requestConfig = prepareRequestConfig(prompt, painting)
+      const requestConfig = await prepareRequestConfig(prompt, painting)
 
       // 发送API请求
       const urls = await callApi(requestConfig, controller)
diff --git a/src/renderer/src/pages/paintings/config/DmxapiConfig.ts b/src/renderer/src/pages/paintings/config/DmxapiConfig.ts
index 6dbe820a78..43bdf33c2c 100644
--- a/src/renderer/src/pages/paintings/config/DmxapiConfig.ts
+++ b/src/renderer/src/pages/paintings/config/DmxapiConfig.ts
@@ -83,7 +83,7 @@ export const MODEOPTIONS = [
 // 获取模型分组数据
 export const GetModelGroup = async (): Promise<DMXApiModelGroups> => {
   try {
-    const response = await fetch('https://dmxapi.cn/cherry_painting_models.json')
+    const response = await fetch('https://dmxapi.cn/cherry_painting_models_v2.json')
 
     if (response.ok) {
       const data = await response.json()