From 6b8acda14a2f763f581ed3daebd4b68c7d6a4d0f Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Thu, 4 Sep 2025 13:15:40 +0800
Subject: [PATCH] 'commit'

---
 dsLightRag/Routes/CopyFaceRoute.py            |   8 +-
 .../__pycache__/CopyFaceRoute.cpython-310.pyc | Bin 4507 -> 4586 bytes
 dsLightRag/static/LibLib/copyface.html        |  83 ++++++++++++++++--
 3 files changed, 83 insertions(+), 8 deletions(-)

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 d018946f4bf8b277e941fb51ff03b7278e0d0723..5ae9b89e2b95467e0fb2491068ebe478abef283b 100644
GIT binary patch
delta 809
zcmZuvO=uHA7@co-lI(7>Kl$03q}qxp!5~twU@>?oh~mK?uxcSt*0nRKEol-r+mx0~
z(F%fMtvJV8Q1D<+rS>3t5<GkG(nACfLjCDouR4>8ROv4B_`Wx9-|W1Z<=l48sOh?f
z_|tvx^^NuPy0HzzdEQb~0)k4?6+~r<37kYkDurm6${ECaS@grGfZOJ&T7CA6J4P?j
zo71%6x7CR$cbq!YUgbhNT=!~Ib-xf{15z>R>D<_pIIJQsDe5F^N~g^*FEyk<3cM&U
z*(b=d&(ix)N6%i##j}PGn*uBWaRCVqZF!>Vw?pMh&Go(gf)(z6y#H1yQP~{Q>;%q%
zy)lYENDycJkxYK)H-tc_+V(j09k=ODRNU(o>NMP`dWAOf8`Vfj8r&yjX#p7lSq`h%
z+k@TNXpfkeJ)Qkh9q3|?w#nB#9xdknTQf0EtJHIS>WuLyoifjIh_UFrEzU)ZcYHLi
z0YbN7?k>y6Rz_el;$00mR^<Q_GlN=@AUvrA5zeWcu`-Cn7v+``DBhf}J&lq?fR+(S
z2Ffy8!Os8?Nd*eA14O!>;1M*8xGKgp{lLW~K;m6M82S3_8laX6XwDY%B<mZ2dIz|N
zBs_df(z`jLE=mhPa7&$6XJ7^ux~F?tQ-7P-di-E(Y3=*1dta6weSQAs+u9-<Rm(h-
z6{9b^SAa#nWuDwe;~|)A((D7B-8T<~#laUe_R{RncJ9)UkPQmhl_~qll|CmMmsyW>
vB$QIHEU|Oe3ozIZ>loPVaJ)C$xkKmw{j_6}Vs8AXjx9rjSjK|bKpK7nM?kaT

delta 717
zcmZut&ubGw82!H8X?8csCYxW)FRfy#5JVI`6jDK<Y7e587AXWFter_rO`15_7+N+#
z>%mj?Fc%>f^wjnsDD~n^@a(}$ksQQx{{&CYC}@?=FpoFi`{oVb+b;|Yk(O--L9RRD
z5BIjy?;`s+Jua^d%1EKC-3F#JLh&L{Z2^k<C_2+>uU>PP*gSjCWgTy1)thdORl2-!
zvp6QMYb9$3cwAcWxOkymwTYxb2qO<kBEDfx1oY45DA4AuX4?~A^%Aj_5+liTBfY-j
zjmD~tmh15&%0pA|+wBiaD3+9v7G+Yv<nDlc(a;fdM)uU-R@F61)tFatAG-C1d$++V
z9k<zTu+GWd=SEqZQgiaOf{cQ!gdM^Usq(ziCzP)9yf_}LU`jj+?#VwFL#4uhf5vNU
zg>lzol_eQfr7E+!B;JPxDN7ottUt-qnL~ibxHb^8;mz5=vf&dN&4Q21resbQXlz68
z8@|E0XT1v2xDP|13EvpPCV7nrH0c{O<pUjmNfu!NWLHQ`+edFbMNoSk2zcSuErh;_
zu$EFf&3U$OK0)cBDG!g(%)uJ9Hner5q;C$)9`>M^@7iL^{wM}kRZjdpGMPIposyhK
z#xfCoh7s{5Iw|&|69GpSi<bBiJ)63$^eF`gk`gn{i9kmdJM-ecb6O{6Kn$I2w8fR!
jc}$94>{w2D@EiX={AH04Z)0a{-H8M-oOLjKA%i~wdGwsX

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 @@
         <input type="url" id="imageUrl" placeholder="请输入图片URL地址"
                value="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/HuangWanQiao.jpg">
     </div>
+
+    <!-- 新增：提示词编辑区域 -->
+    <div class="form-group">
+        <label for="promptInput">提示词：</label>
+        <textarea id="promptInput" class="prompt-textarea" placeholder="请输入提示词..."></textarea>
+        <div style="display: flex; justify-content: space-between; margin-top: 8px;">
+            <small style="color: #666;">提示：修改提示词可以改变生成图片的风格和细节</small>
+            <button id="resetPromptBtn" style="padding: 4px 8px; font-size: 12px;">重置默认</button>
+        </div>
+    </div>
+
     <div class="image-preview" id="previewContainer">
-        <h3>参考图片预览：</h3>
+        <h3>人像照片</h3>
         <img id="previewImage" src="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/Backup/HuangWanQiao.jpg"
-             alt="参考图片预览" onerror="this.style.display='none'">
+             alt="参考图片" onerror="this.style.display='none'">
     </div>
 
     <button onclick="generateImage()" id="generateBtn">生成换脸图片</button>
@@ -337,7 +360,7 @@
         <h3 class="history-title">📜 生成历史</h3>
         <div class="history-images">
             <div class="history-image-item">
-                <img src="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/bdde20e3143049c3916aa0b8d523b469.jpg" 
+                <img src="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/7e2c6203189344388964c29963f666ad.jpg"
                      alt="生成历史图片1" onclick="openModal(this.src)"/>
             </div>
             <div class="history-image-item">
@@ -356,6 +379,10 @@
                 <img src="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/215df7724cba4bf9a64e7863ecb4ebc0.jpg" 
                      alt="生成历史图片5" onclick="openModal(this.src)"/>
             </div>
+            <div class="history-image-item">
+                <img src="https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/77eb450d7aa74c888a852f66b3bfb3b2.jpg"
+                     alt="生成历史图片6" onclick="openModal(this.src)"/>
+            </div>
         </div>
     </div>
 
@@ -368,6 +395,9 @@
 <script>
     const API_BASE = '/api/copyface';
     let selectedModel = '';
+    // 新增：存储模型提示词
+    let modelPrompts = {};
+    
     // 新增：模态框相关变量
     const modal = document.getElementById('imageModal');
     const modalImage = document.getElementById('modalImage');
@@ -391,7 +421,40 @@
                 previewImg.style.display = 'none';
             }
         });
+        
+        // 新增：获取所有模型提示词
+        fetchModelPrompts();
+        
+        // 新增：重置提示词按钮事件
+        document.getElementById('resetPromptBtn').addEventListener('click', resetPromptToDefault);
     });
+    
+    // 新增：获取所有模型提示词
+    async function fetchModelPrompts() {
+        try {
+            const response = await fetch(`${API_BASE}/samples`);
+            if (response.ok) {
+                const models = await response.json();
+                models.forEach(model => {
+                    modelPrompts[model.name] = model.prompt;
+                });
+                
+                // 如果已有选中的模型，加载其提示词
+                if (selectedModel && modelPrompts[selectedModel]) {
+                    document.getElementById('promptInput').value = modelPrompts[selectedModel];
+                }
+            }
+        } catch (error) {
+            console.error('获取模型提示词失败:', error);
+        }
+    }
+    
+    // 新增：重置提示词为默认值
+    function resetPromptToDefault() {
+        if (selectedModel && modelPrompts[selectedModel]) {
+            document.getElementById('promptInput').value = modelPrompts[selectedModel];
+        }
+    }
 
     // 新增：打开模态框显示大图
     function openModal(imageUrl) {
@@ -411,7 +474,7 @@
         }
     }
 
-    // 选择模型
+    // 修改：选择模型函数，加载对应提示词
     function selectModel(card, modelName) {
         // 移除所有选中状态
         document.querySelectorAll('.model-card').forEach(card => {
@@ -421,13 +484,19 @@
         // 添加选中状态
         card.classList.add('selected');
         selectedModel = modelName;
+        
+        // 新增：加载模型对应的提示词
+        if (modelPrompts[selectedModel]) {
+            document.getElementById('promptInput').value = modelPrompts[selectedModel];
+        }
 
         console.log('已选择模型:', selectedModel);
     }
 
-    // 生成换脸图片
+    // 修改：生成换脸图片函数，支持自定义提示词
     async function generateImage() {
         const imageUrl = document.getElementById('imageUrl').value.trim();
+        const customPrompt = document.getElementById('promptInput').value.trim();
         const generateBtn = document.getElementById('generateBtn');
         const loading = document.getElementById('loading');
         const resultContainer = document.getElementById('resultContainer');
@@ -459,7 +528,9 @@
                     'Content-Type': 'application/json',
                 },
                 body: JSON.stringify({
-                    image_url: imageUrl
+                    image_url: imageUrl,
+                    // 新增：传递自定义提示词
+                    prompt: customPrompt
                 })
             });