diff --git a/dsLightRag/Routes/CopyFaceRoute.py b/dsLightRag/Routes/CopyFaceRoute.py index a61f3e56..2b08dba0 100644 --- a/dsLightRag/Routes/CopyFaceRoute.py +++ b/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"] ) diff --git a/dsLightRag/Routes/__pycache__/CopyFaceRoute.cpython-310.pyc b/dsLightRag/Routes/__pycache__/CopyFaceRoute.cpython-310.pyc index d018946f..5ae9b89e 100644 Binary files a/dsLightRag/Routes/__pycache__/CopyFaceRoute.cpython-310.pyc and b/dsLightRag/Routes/__pycache__/CopyFaceRoute.cpython-310.pyc differ diff --git a/dsLightRag/static/LibLib/copyface.html b/dsLightRag/static/LibLib/copyface.html index b5df03fa..e18080f6 100644 --- a/dsLightRag/static/LibLib/copyface.html +++ b/dsLightRag/static/LibLib/copyface.html @@ -188,6 +188,18 @@ margin: 0; } + /* 新增:提示词样式 */ + .prompt-textarea { + width: 100%; + height: 120px; + padding: 10px; + border: 1px solid #ddd; + border-radius: 5px; + font-size: 14px; + resize: vertical; + font-family: monospace; + } + /* 新增:图片模态框样式 */ .modal { display: none; @@ -313,10 +325,21 @@ + + +