save code

app.py

@@ -504,6 +504,7 @@ async def hair_grow(
     gender: Optional[str] = Form(default=None, description="性别 male/female（必填）"),
     beauty_enabled: bool = Form(default=False, description="是否开启美颜（本期不生效）"),
     use_mask: bool = Form(default=True, description="是否启用 inpaint 遮罩（测试对比用）。false 时用干净原图生成（空遮罩，不烧模板线）"),
+    prompt: str = Form(default="补充遮罩区域的头发", description="ComfyUI 提示词，会替换工作流节点60的文本"),
 ):
     # 1. gender 必填校验（非法/缺失 → 1004）
     if gender not in ("male", "female"):
@@ -530,7 +531,7 @@ async def hair_grow(
         from hairline.service import generate_grow_results
 
         # 预览 + 生发(ComfyUI) 都是阻塞且较慢，放线程池避免卡住事件循环
-        items = await run_in_threadpool(generate_grow_results, image, gender, use_mask)
+        items = await run_in_threadpool(generate_grow_results, image, gender, use_mask, prompt)
         if items is None:
             return err(1001, "无法识别人像")
 
@@ -596,6 +597,7 @@ async def hair_grow_b(
     marked_image_url: Optional[str] = Form(default=None, description="划线图片 URL"),
     marked_image_base64: Optional[str] = Form(default=None, description="划线图片 base64"),
     use_mask: bool = Form(default=True, description="是否画发际线（测试对比用）。false 时跳过划线检测、直接送划线图"),
+    prompt: str = Form(default="补充遮罩区域的头发", description="ComfyUI 提示词，会替换工作流节点60的文本"),
 ):
     # 划线图三选一取图（只需这一张）
     marked_raw, e = await resolve_image_bytes(marked_image_file, marked_image_url, marked_image_base64)
@@ -617,7 +619,7 @@ async def hair_grow_b(
         from fastapi.concurrency import run_in_threadpool
         from hairline.service import generate_grow_b
 
-        res = await run_in_threadpool(generate_grow_b, marked, use_mask)
+        res = await run_in_threadpool(generate_grow_b, marked, use_mask, prompt)
         if res["status"] == "no_face":
             return err(1001, "无法识别人像")
         if res["status"] == "no_line":