'commit'

Controller/HaiBaoController.py

@@ -75,8 +75,9 @@ async def refine_prompt(req: RefineRequest):
         1. 描述画面主体、背景、光影、色彩、构图。
         2. 风格要求：现代感、科技感、精美、3D渲染风格或高品质插画风格。
         3. 融入新能源、绿色环保、充电桩等元素。
-        4. 直接输出提示词内容，不要包含“好的”、“以下是”等无关废话。
-        5. 字数在100-300字之间。
+        4. 关于品牌元素：画面中可自然融入品牌Logo的视觉元素（如配色、形状），能用多少就用多少，有元素体现即可，不必生搬硬套，保持画面自然和谐。
+        5. 直接输出提示词内容，不要包含“好的”、“以下是”等无关废话。
+        6. 字数在100-300字之间。
         """
         
         refined_prompt = ""
@@ -116,8 +117,9 @@ async def generate_poster(req: GenerateRequest):
                 1. 描述画面主体、背景、光影、色彩、构图。
                 2. 风格要求：现代感、科技感、精美、3D渲染风格或高品质插画风格。
                 3. 融入新能源、绿色环保、充电桩等元素。
-                4. 直接输出提示词内容，不要包含“好的”、“以下是”等无关废话。
-                5. 字数在100-300字之间。
+                4. 关于品牌元素：画面中可自然融入品牌Logo的视觉元素（如配色、形状），能用多少就用多少，有元素体现即可，不必生搬硬套，保持画面自然和谐。
+                5. 直接输出提示词内容，不要包含“好的”、“以下是”等无关废话。
+                6. 字数在100-300字之间。
                 """
                 
                 refined_prompt = ""
@@ -133,7 +135,9 @@ async def generate_poster(req: GenerateRequest):
 
             # 1.2 调用生图
             resp = await client.generate_image(prompt=final_prompt, size=f"{req.width}x{req.height}")
-            obs_urls = await client.download_and_upload_to_obs(resp)
+            # 定义Logo路径
+            LOGO_PATH = r"d:\dsWork\aiData\static\Images\login_logo.png"
+            obs_urls = await client.download_and_upload_to_obs(resp, overlay_logo_path=LOGO_PATH)
             if not obs_urls:
                 raise Exception("未获取到有效的图片URL")
             return obs_urls[0]