From 09910c63b87f865b242bd322a2d6a388701adc82 Mon Sep 17 00:00:00 2001
From: iHeyTang <dehui1012@qq.com>
Date: Wed, 13 Aug 2025 17:46:10 +0800
Subject: [PATCH] fix

---
 workflow_service/main.py                      |   3 +-
 workflow_service/routes/runx/__init__.py      |   4 +
 workflow_service/routes/runx/_base.py         |   6 +
 .../routes/runx/model_with_multi_dress.py     | 131 ++++++++++++++++++
 4 files changed, 143 insertions(+), 1 deletion(-)
 create mode 100644 workflow_service/routes/runx/__init__.py
 create mode 100644 workflow_service/routes/runx/_base.py
 create mode 100644 workflow_service/routes/runx/model_with_multi_dress.py

diff --git a/workflow_service/main.py b/workflow_service/main.py
index 260babe..bc9c873 100644
--- a/workflow_service/main.py
+++ b/workflow_service/main.py
@@ -6,7 +6,7 @@ from fastapi.middleware.cors import CORSMiddleware
 
 from workflow_service import database
 from workflow_service.config import Settings
-from workflow_service.routes import service, workflow, run
+from workflow_service.routes import service, workflow, run, runx
 
 settings = Settings()
 
@@ -24,6 +24,7 @@ web_app.add_middleware(
 web_app.include_router(service.service_router)
 web_app.include_router(workflow.workflow_router)
 web_app.include_router(run.run_router)
+web_app.include_router(runx.runx_router)
 
 
 @web_app.on_event("startup")
diff --git a/workflow_service/routes/runx/__init__.py b/workflow_service/routes/runx/__init__.py
new file mode 100644
index 0000000..e71f4a9
--- /dev/null
+++ b/workflow_service/routes/runx/__init__.py
@@ -0,0 +1,4 @@
+from ._base import runx_router
+from .model_with_multi_dress import runx_router as model_with_multi_dress_router
+
+__all__ = ["runx_router", "model_with_multi_dress_router"]
diff --git a/workflow_service/routes/runx/_base.py b/workflow_service/routes/runx/_base.py
new file mode 100644
index 0000000..8cda99f
--- /dev/null
+++ b/workflow_service/routes/runx/_base.py
@@ -0,0 +1,6 @@
+from fastapi import APIRouter
+
+runx_router = APIRouter(
+    prefix="/api/runx",
+    tags=["RunX"],
+)
diff --git a/workflow_service/routes/runx/model_with_multi_dress.py b/workflow_service/routes/runx/model_with_multi_dress.py
new file mode 100644
index 0000000..485108d
--- /dev/null
+++ b/workflow_service/routes/runx/model_with_multi_dress.py
@@ -0,0 +1,131 @@
+import json
+from typing import Optional
+
+from fastapi import Request, HTTPException
+from fastapi.responses import JSONResponse
+from workflow_service import comfyui_client, database
+from ._base import runx_router
+
+RUNX_NAME = "model_with_multi_dress"
+RUNX_WORKFLOW_NAME = "测试"
+
+
+@runx_router.post(f"/{RUNX_NAME}")
+async def model_with_multi_dress(
+    request: Request,
+    workflow_version: Optional[str] = None,
+):
+    """
+    一个模特，穿多件不同的衣服
+    """
+    workflow_name = RUNX_WORKFLOW_NAME
+
+    # 获取工作流定义
+    workflow_data = await database.get_workflow(workflow_name, workflow_version)
+    if not workflow_data:
+        detail = (
+            f"工作流 '{workflow_name}'"
+            + (f" 带版本 '{workflow_version}'" if workflow_version else " (最新版)")
+            + " 未找到。"
+        )
+        raise HTTPException(status_code=404, detail=detail)
+
+    workflow = json.loads(workflow_data["workflow_json"])
+    api_spec = comfyui_client.parse_api_spec(workflow)
+
+    # 将请求拆分为多个请求
+    data = await request.json()
+    batch_data = _convert(data)
+
+    # 提交到队列
+    workflow_run_ids: list[str] = []
+    for item in batch_data:
+        workflow_run_id = await comfyui_client.submit_workflow_to_queue(
+            workflow_name=workflow_name,
+            workflow_data=workflow,
+            api_spec=api_spec,
+            request_data=item,
+        )
+        workflow_run_ids.append(workflow_run_id)
+
+    return JSONResponse(
+        content={
+            "workflow_run_ids": workflow_run_ids,
+            "status": "queued",
+            "message": "工作流已提交到队列，正在等待执行",
+        },
+        status_code=202,
+    )
+
+
+@runx_router.get(f"/{RUNX_NAME}/json_schema")
+async def model_with_multi_dress_json_schema():
+    """
+    获取工作流定义
+    """
+    return {
+        "type": "object",
+        "properties": {
+            "模特图": {
+                "type": "object",
+                "properties": {
+                    "image": {"type": "string"},
+                },
+            },
+            "模特描述": {
+                "type": "object",
+                "properties": {
+                    "value": {"type": "string"},
+                },
+            },
+            "穿搭图": {
+                "type": "array",
+                "items": {
+                    "type": "object",
+                    "properties": {
+                        "image": {"type": "string"},
+                    },
+                },
+            },
+        },
+        "required": ["模特图", "模特描述", "穿搭图"],
+    }
+
+
+def _convert(data: dict) -> list[dict]:
+    """
+    数据格式
+    {
+        "模特图": {"image" : "xxx" },
+        "模特描述" : { "value" : "xxx" },
+        "穿搭图": [
+            {"image" : "xxx" },
+            {"image" : "xxx" }
+        ]
+    }
+    转换为
+    [
+        {
+            "模特图": {"image" : "xxx" },
+            "模特描述" : { "value" : "xxx" },
+            "穿搭图": { "image" : "xxx" }
+        }
+    ]
+    """
+    result: list[dict] = []
+
+    # 获取基础信息
+    model_image = data.get("模特图", {})
+    model_description = data.get("模特描述", {})
+    dress_images = data.get("穿搭图", [])
+
+    # 为每个穿搭图创建一个记录
+    for dress_image in dress_images:
+        record = {
+            "模特图": model_image,
+            "模特描述": model_description,
+            "穿搭图": dress_image,
+        }
+        result.append(record)
+
+    return result