save code

change2/CLAUDE.md

@@ -0,0 +1,71 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Project Overview
+
+This is a FastAPI-based virtual try-on service that integrates with ComfyUI for AI-powered clothing try-on. The service accepts base64-encoded images (model photo, shirt, pants) and returns a generated try-on result.
+
+## Running the Service
+
+Start the server:
+```bash
+python service2.py
+```
+
+The server runs on `http://0.0.0.0:47698` by default.
+
+## Dependencies
+
+Required Python packages (install via pip):
+- fastapi
+- uvicorn
+- httpx
+- pydantic
+
+## Architecture
+
+### Request Flow
+1. Client sends POST to `/try-on` with base64 images (model_image, shirt_image, pants_image)
+2. Server decodes images and uploads to ComfyUI at `http://127.0.0.1:8188`
+3. Server modifies workflow JSON (change2_1_ex.json) with uploaded image filenames
+4. Workflow submitted to ComfyUI queue
+5. Server polls ComfyUI history endpoint until completion (max 300s timeout)
+6. Result image downloaded from ComfyUI and returned as base64
+
+### Key Components
+
+**service2.py**: Main FastAPI application
+- `/try-on`: Main endpoint for virtual try-on
+- `/health`: Health check endpoint
+- `/`: Serves static frontend (index.html)
+
+**change2_1_ex.json**: ComfyUI workflow configuration
+- Node "11": Model image input (LoadImage)
+- Node "10": Shirt image input (LoadImage)
+- Node "33": Output image (SaveImage)
+- Node "31": Text prompt for image editing
+
+**static/index.html**: Frontend interface for testing
+
+### ComfyUI Integration
+
+The service expects ComfyUI to be running at `http://127.0.0.1:8188` with:
+- FireRed-Image-Edit model loaded
+- Qwen 2.5 VL CLIP model
+- Required custom nodes for QwenEditConfigPreparer, QwenEditOutputExtractor
+
+### Workflow Modification
+
+When processing requests, the code dynamically updates these workflow nodes:
+- `workflow["11"]["inputs"]["image"]`: Model photo filename
+- `workflow["10"]["inputs"]["image"]`: Shirt photo filename
+- `workflow["31"]["inputs"]["value"]`: Text prompt (currently hardcoded in Chinese)
+
+## Important Notes
+
+- Images are uploaded with unique filenames (UUID prefix) to avoid caching issues
+- The pants_image parameter is currently unused (commented out in code)
+- Workflow uses 8 sampling steps with euler sampler and beta scheduler
+- Output resolution is 768x1024 pixels
+- CORS is enabled for all origins

change2/service2.py

@@ -44,7 +44,6 @@ app.add_middleware(
 class TryOnRequest(BaseModel):
     model_image: str   # base64 模特图片
     shirt_image: str   # base64 上衣图片
-    pants_image: str   # base64 裤子图片
 
 
 class TryOnResponse(BaseModel):
@@ -113,7 +112,6 @@ async def try_on(req: TryOnRequest):
     换装接口
     - model_image: 模特图片 base64
     - shirt_image: 上衣图片 base64
-    - pants_image: 裤子图片 base64
     返回 result_image: 换装结果 base64
     """
 
@@ -124,11 +122,10 @@ async def try_on(req: TryOnRequest):
 
 
     async with httpx.AsyncClient(timeout=60.0) as client:
-        # 解码三张图片
+        # 解码两张图片
         try:
             model_bytes = decode_base64_image(req.model_image)
             shirt_bytes = decode_base64_image(req.shirt_image)
-            # pants_bytes = decode_base64_image(req.pants_image)
         except Exception as e:
             raise HTTPException(status_code=400, detail=f"图片解码失败: {e}")
 
@@ -136,14 +133,12 @@ async def try_on(req: TryOnRequest):
         uid = uuid.uuid4().hex[:8]
         model_fname = f"model_{uid}.jpg"
         shirt_fname = f"shirt_{uid}.jpg"
-        # pants_fname = f"pants_{uid}.jpg"
 
-        # 并发上传三张图片
+        # 并发上传两张图片
         try:
-            model_name, shirt_name, pants_name = await asyncio.gather(
+            model_name, shirt_name = await asyncio.gather(
                 upload_image(client, model_bytes, model_fname),
                 upload_image(client, shirt_bytes, shirt_fname),
-                # upload_image(client, pants_bytes, pants_fname),
             )
         except httpx.HTTPError as e:
             raise HTTPException(status_code=502, detail=f"上传图片到 ComfyUI 失败: {e}")
@@ -151,7 +146,6 @@ async def try_on(req: TryOnRequest):
         # 修改工作流节点的图片输入
         workflow[NODE_MODEL]["inputs"]["image"] = model_name
         workflow[NODE_SHIRT]["inputs"]["image"] = shirt_name
-        # workflow[NODE_PANTS]["inputs"]["image"] = pants_name
 
         # 提交工作流
         try:

change2/static/index.html

@@ -37,7 +37,7 @@
 
     .upload-grid {
       display: grid;
-      grid-template-columns: repeat(3, 1fr);
+      grid-template-columns: repeat(2, 1fr);
       gap: 20px;
       width: 100%;
       max-width: 900px;
@@ -204,7 +204,7 @@
 </head>
 <body>
   <h1>AI 换装系统</h1>
-  <p class="subtitle">上传模特、上衣、裤子三张图片，AI 自动完成换装</p>
+  <p class="subtitle">上传模特和上衣两张图片，AI 自动完成换装</p>
 
   <div class="upload-grid">
     <!-- 模特 -->
@@ -224,15 +224,6 @@
       <div class="upload-hint">节点 10 · 上身服装</div>
       <img class="preview-img" id="preview-shirt" alt="上衣预览" />
     </div>
-
-    <!-- 裤子 -->
-    <div class="upload-card" id="card-pants">
-      <input type="file" accept="image/*" id="input-pants" />
-      <div class="upload-icon">👖</div>
-      <div class="upload-label">裤子图片</div>
-      <div class="upload-hint">节点 4 · 下身服装</div>
-      <img class="preview-img" id="preview-pants" alt="裤子预览" />
-    </div>
   </div>
 
   <button class="btn-run" id="btn-run" disabled>开始换装</button>
@@ -245,7 +236,7 @@
   </div>
 
   <script>
-    const images = { model: null, shirt: null, pants: null };
+    const images = { model: null, shirt: null };
 
     function setupUpload(inputId, previewId, cardId, key) {
       const input = document.getElementById(inputId);
@@ -269,11 +260,10 @@
 
     setupUpload('input-model', 'preview-model', 'card-model', 'model');
     setupUpload('input-shirt', 'preview-shirt', 'card-shirt', 'shirt');
-    setupUpload('input-pants', 'preview-pants', 'card-pants', 'pants');
 
     function updateRunButton() {
       const btn = document.getElementById('btn-run');
-      btn.disabled = !(images.model && images.shirt && images.pants);
+      btn.disabled = !(images.model && images.shirt);
     }
 
     function setStatus(msg, cls = '') {
@@ -308,7 +298,6 @@
           body: JSON.stringify({
             model_image: images.model,
             shirt_image: images.shirt,
-            // pants_image: images.pants,
           }),
         });