'commit'

dsLightRag/Routes/CopyFaceRoute.py

@@ -19,6 +19,7 @@ COPY_FACE_CONFIG_PATH = os.path.join(os.path.dirname(__file__), "..", "Liblib",
 
 class CopyFaceRequest(BaseModel):
     image_url: str
+    prompt: Optional[str] = None  # 新增：可选的自定义提示词
 
 class ModelSample(BaseModel):
     name: str
@@ -119,7 +120,7 @@ async def generate_copy_face_with_model(model_name: str, request: CopyFaceReques
     
     Args:
         model_name: 模型名称（如：炫酷机甲美女_majicflus）
-        request: 包含图片URL的请求体
+        request: 包含图片URL和可选提示词的请求体
     
     Returns:
         包含生成图片OBS地址的字典
@@ -150,9 +151,12 @@ async def generate_copy_face_with_model(model_name: str, request: CopyFaceReques
             height=model_config["height"]
         )
         
+        # 新增：使用自定义提示词或默认提示词
+        prompt = request.prompt if request.prompt else model_config["prompt"]
+        
         # 生成图像
         obs_url = generator.generate_image(
-            prompt=model_config["prompt"],
+            prompt=prompt,  # 使用可能已修改的提示词
             reference_image_url=request.image_url,
             control_weight=model_config["control_weight"]
         )