+
+
+
+
+
+
+
+
+
+
+ nku.png:
+ theme: "南开大学"
+ style: "学术正式"
+ template: |
+
+
+ 
+
+
+
+
+
+
+ {{ title }}
+{{ subtitle }}
+
+
+
+
+
+
+
+
+
+
+
+ stamp.jpg:
+ theme: "印章装饰"
+ style: "传统文化"
+ template: |
+
+
+ 
+
+
+
+
+
+ 
+
+
+
+
+ {{ title }}
+{{ subtitle }}
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ background.png:
+ theme: "通用背景"
+ style: "简约现代"
+ template: |
+
+
+
+
+
+
+
+
+
+ {{ title }}
+{{ subtitle }}
+
+
+
+
+ 
+
+
+
+
+
+
+
+
+
diff --git a/outputs/80155705_20250703114134_0.png b/outputs/80155705_20250703114134_0.png
new file mode 100644
index 0000000..b154ebc
Binary files /dev/null and b/outputs/80155705_20250703114134_0.png differ
diff --git a/outputs/80155705_20250703114134_1.png b/outputs/80155705_20250703114134_1.png
new file mode 100644
index 0000000..c9ea013
Binary files /dev/null and b/outputs/80155705_20250703114134_1.png differ
diff --git a/outputs/98856890_20250702220533_0.png b/outputs/98856890_20250702220533_0.png
new file mode 100644
index 0000000..08c2392
Binary files /dev/null and b/outputs/98856890_20250702220533_0.png differ
diff --git a/outputs/98856890_20250702220533_1.png b/outputs/98856890_20250702220533_1.png
new file mode 100644
index 0000000..683939b
Binary files /dev/null and b/outputs/98856890_20250702220533_1.png differ
diff --git a/outputs/generated_code.vue b/outputs/generated_code.vue
index 04c9dc1..2a25d60 100644
--- a/outputs/generated_code.vue
+++ b/outputs/generated_code.vue
@@ -1,11 +1,23 @@
-
+
+
+
+
+
+
+
+ {{ title }}
+{{ subtitle }}
+
+
+
+
+
+
+
- ![端午节背景]()
- ![端午节内容]()
-
{{ title }}
- {{ subtitle }}
-
- ![Logo]()
+
+
@@ -13,88 +25,96 @@
\ No newline at end of file
+
+.lotus-subtitle {
+ font-size: 32px;
+ color: #4a6741;
+ margin-bottom: 40px;
+ font-family: 'KaiTi', cursive;
+}
+
+.lotus-content {
+ position: absolute;
+ top: 50%;
+ left: 50%;
+ transform: translate(-50%, -50%);
+ width: 70%;
+}
+
+.lotus-decoration {
+ width: 200px;
+ height: 200px;
+ background: radial-gradient(circle, rgba(255,192,203,0.6), rgba(255,182,193,0.3));
+ border-radius: 50%;
+ margin: 0 auto;
+}
+
+.footer-section {
+ position: absolute;
+ bottom: 10%;
+ left: 50%;
+ transform: translateX(-50%);
+}
+
+.logo-text {
+ font-size: 24px;
+ color: #2d5016;
+ font-weight: 500;
+}
+
diff --git a/outputs/poster_content.json b/outputs/poster_content.json
index b83da0f..f5cd123 100644
--- a/outputs/poster_content.json
+++ b/outputs/poster_content.json
@@ -4,13 +4,13 @@
"color": "#000000"
},
"layer6_title_content": {
- "content": "端午节安康",
+ "content": "端午节",
"font_name": "SimHei",
"color": "#000000"
},
"layer7_subtitle_content": {
- "content": "粽叶飘香，龙舟竞渡，共庆端午佳节。",
- "font_name": "Microsoft YaHei",
+ "content": "粽叶飘香，龙舟竞渡",
+ "font_name": "SimHei",
"color": "#7E0C6E"
}
}
\ No newline at end of file
diff --git a/outputs/temp_config.yaml b/outputs/temp_config.yaml
new file mode 100644
index 0000000..6834e55
--- /dev/null
+++ b/outputs/temp_config.yaml
@@ -0,0 +1,37 @@
+available_fonts:
+- displayName: 微软雅黑
+ name: Microsoft YaHei
+ roles:
+ - title
+ - subtitle
+ - content
+ tags:
+ - 现代
+ - 清晰
+- displayName: 黑体
+ name: SimHei
+ roles:
+ - title
+ - subtitle
+ - content
+ tags:
+ - 通用
+ - 标准
+default_logo_text: ''
+vue_templates:
+ background.png:
+ style: 简约现代
+ template_name: background_template
+ theme: 通用背景
+ lotus.jpg:
+ style: 传统优雅
+ template_name: lotus_template
+ theme: 荷花主题
+ nku.png:
+ style: 学术正式
+ template_name: nku_template
+ theme: 南开大学
+ stamp.jpg:
+ style: 传统文化
+ template_name: stamp_template
+ theme: 印章装饰
diff --git a/requirments.txt b/requirments.txt
deleted file mode 100644
index 948a569..0000000
Binary files a/requirments.txt and /dev/null differ
diff --git a/scripts/__pycache__/generate_layout.cpython-311.pyc b/scripts/__pycache__/generate_layout.cpython-311.pyc
index 48cf8dc..0588cad 100644
Binary files a/scripts/__pycache__/generate_layout.cpython-311.pyc and b/scripts/__pycache__/generate_layout.cpython-311.pyc differ
diff --git a/scripts/__pycache__/prompt_analysis.cpython-311.pyc b/scripts/__pycache__/prompt_analysis.cpython-311.pyc
deleted file mode 100644
index fb250f5..0000000
Binary files a/scripts/__pycache__/prompt_analysis.cpython-311.pyc and /dev/null differ
diff --git a/scripts/__pycache__/utils.cpython-311.pyc b/scripts/__pycache__/utils.cpython-311.pyc
new file mode 100644
index 0000000..c0307ec
Binary files /dev/null and b/scripts/__pycache__/utils.cpython-311.pyc differ
diff --git a/scripts/generate_layout.py b/scripts/generate_layout.py
index 5313d4a..6094cc5 100644
--- a/scripts/generate_layout.py
+++ b/scripts/generate_layout.py
@@ -1,43 +1,44 @@
"""
@file generate_layout.py
@brief Vue组件代码生成模块
- 基于DeepSeek API生成响应式Vue 3组件布局代码
+ 基于DeepSeek API生成响应式Vue 3组件布局代码，支持预定义模板
@author 王秀强 (2310460@mail.nankai.edu.cn)
@date 2025.5.19
-@version v1.0.0
+@version v2.0.0
@details
本文件主要实现：
- DeepSeek API调用封装和错误处理
- Vue 3 Composition API组件代码生成
- 海报布局的动态排版和样式生成
+ - 预定义Vue模板系统
+ - 增强Vue代码生成逻辑
- 代码清理和格式化处理
- - 备用Vue模板生成机制
@note
- 需要配置DEEPSEEK_API_KEY环境变量
- 支持流式和非流式响应模式
- 生成的Vue代码包含完整的template、script和style部分
- 具备指数退避重试机制处理API限流
+- 支持基于图片类型的预定义模板
@usage
# 生成Vue组件代码
-vue_code = generate_vue_code("生成端午节海报Vue组件")
+vue_code = generate_vue_code_enhanced("生成端午节海报Vue组件", parse_imglist, suggestions)
save_code(vue_code, "../outputs/poster.vue")
-# 调用DeepSeek API
-result, usage = call_deepseek(prompt="请生成代码", temperature=0.6)
-
@copyright
(c) 2025 砚生项目组
"""
import os
+import yaml
from openai import OpenAI
from dotenv import load_dotenv
import time
from colorama import init, Fore, Back, Style
+from typing import List, Dict, Optional
# 初始化colorama
init(autoreset=True)
@@ -125,9 +126,193 @@ def call_deepseek(
raise Exception("达到最大重试次数，API 调用失败")
+def load_vue_templates() -> Dict:
+ """加载预定义的Vue模板配置"""
+ template_path = "../configs/vue_templates.yaml"
+
+ try:
+ with open(template_path, 'r', encoding='utf-8') as f:
+ templates = yaml.safe_load(f)
+ print(f"{Fore.GREEN}✅ Vue模板配置加载成功{Style.RESET_ALL}")
+ return templates.get('vue_templates', {})
+ except Exception as e:
+ print(f"{Fore.YELLOW}⚠️ Vue模板配置加载失败: {e}，使用默认模板{Style.RESET_ALL}")
+ return {}
+
+
+def get_template_by_images(parse_imglist: List[Dict]) -> Optional[str]:
+ """
+ 根据图片列表选择合适的预定义模板
+
+ 参数:
+ parse_imglist: 图片信息列表
+
+ 返回:
+ 选中的模板代码，如果没有匹配的模板则返回None
+ """
+ templates = load_vue_templates()
+
+ if not templates or not parse_imglist:
+ return None
+
+ # 检查固定的图片文件
+ fixed_images = ["lotus.jpg", "nku.png", "stamp.jpg", "background.png"]
+
+ for img_name in fixed_images:
+ img_path = f"../outputs/{img_name}"
+ if os.path.exists(img_path) and img_name in templates:
+ print(f"{Fore.CYAN}📋 找到匹配的预定义模板: {img_name}{Style.RESET_ALL}")
+ return templates[img_name]['template']
+
+ print(f"{Fore.YELLOW}⚠️ 未找到匹配的预定义模板，将使用AI生成{Style.RESET_ALL}")
+ return None
+
+
+def generate_layout_prompt(user_input_analysis_result: Dict, parse_imglist: List[Dict], suggestions: Dict = None) -> str:
+ """
+ 生成增强的Vue布局提示，包含文案内容
+ 这个函数从run_pipeline.py移动到这里
+ """
+ width = user_input_analysis_result.get("width", 1080)
+ height = user_input_analysis_result.get("height", 1920)
+ theme = user_input_analysis_result.get("main_theme", "活动海报")
+
+ # 构造图片信息字符串
+ images_info = "\n".join(
+ [f"- {img['picture_name']} ({img['picture_description']})" for img in parse_imglist]
+ )
+
+ # 构造文案信息
+ content_info = ""
+ if suggestions:
+ try:
+ if 'layer6_title_content' in suggestions:
+ title = suggestions['layer6_title_content'].get('content', theme)
+ content_info += f"- 主标题: {title}\n"
+
+ if 'layer7_subtitle_content' in suggestions:
+ subtitle = suggestions['layer7_subtitle_content'].get('content', '精彩活动，敬请参与')
+ content_info += f"- 副标题: {subtitle}\n"
+
+ if 'layer5_logo_content' in suggestions:
+ logo = suggestions['layer5_logo_content'].get('text', '主办方')
+ content_info += f"- Logo文字: {logo}\n"
+ except Exception as e:
+ print(f"{Fore.YELLOW}⚠️ 文案信息解析错误: {e}{Style.RESET_ALL}")
+ content_info = f"- 主标题: {theme}\n- 副标题: 精彩活动，敬请参与\n"
+
+ # 调用DeepSeek生成动态排版Prompt
+ system_prompt = "你是一个擅长前端开发的AI，专注于生成Vue.js代码。请根据提供的信息生成完整的Vue组件，包含所有必要的HTML结构和基础定位样式。"
+
+ prompt = f"""
+ 请生成一个Vue.js组件代码，用于{theme}海报，要求如下：
+
+ 组件尺寸: {width}x{height}px
+
+ 图片资源:
+ {images_info}
+
+ 文案内容:
+ {content_info}
+
+ 布局要求:
+ 1. 背景图层: 使用第一张图片作为背景，占据整个组件区域
+ 2. 主标题: 位于画布上方1/3处，居中显示
+ 3. 副标题: 位于主标题下方，居中显示
+ 4. 内容区域: 使用剩余图片，合理布局在中间区域
+ 5. Logo区域: 位于底部，居中显示
+
+ 技术要求:
+ - 使用Vue 3 Composition API
+ - 使用absolute定位进行精确布局
+ - 包含完整的template、script和style部分
+ - 确保所有文本内容都正确显示
+ - 图片使用相对路径引用
+
+ 请生成完整可用的Vue组件代码，不要包含任何说明文字。
+ """
+
+ try:
+ result, _ = call_deepseek(prompt=prompt, system_prompt=system_prompt, temperature=0.4)
+ return result
+ except Exception as e:
+ print(f"{Fore.RED}❌ 布局提示生成失败: {e}{Style.RESET_ALL}")
+ return generate_fallback_vue_code(theme, width, height)
+
+
+def fill_template_content(template: str, suggestions: Dict) -> str:
+ """
+ 填充模板中的动态内容
+
+ 参数:
+ template: Vue模板代码
+ suggestions: 文案建议
+
+ 返回:
+ 填充后的Vue代码
+ """
+ filled_template = template
+
+ try:
+ # 提取文案内容
+ title_content = suggestions.get('layer6_title_content', {}).get('content', '默认标题')
+ subtitle_content = suggestions.get('layer7_subtitle_content', {}).get('content', '默认副标题')
+ logo_content = suggestions.get('layer5_logo_content', {}).get('text', '主办方')
+
+ # 替换模板占位符
+ filled_template = filled_template.replace('{{ title_content }}', title_content)
+ filled_template = filled_template.replace('{{ subtitle_content }}', subtitle_content)
+ filled_template = filled_template.replace('{{ logo_content }}', logo_content)
+
+ print(f"{Fore.GREEN}✅ 模板内容填充完成{Style.RESET_ALL}")
+
+ except Exception as e:
+ print(f"{Fore.YELLOW}⚠️ 模板填充出错: {e}，使用默认内容{Style.RESET_ALL}")
+
+ return filled_template
+
+
+def generate_vue_code_enhanced(
+ user_input_analysis_result: Dict,
+ parse_imglist: List[Dict],
+ suggestions: Dict = None,
+ prompt: str = None
+) -> str:
+ """
+ 增强的Vue代码生成函数
+ 优先使用预定义模板，如果没有匹配的模板则使用AI生成
+
+ 参数:
+ user_input_analysis_result: 用户输入分析结果
+ parse_imglist: 图片信息列表
+ suggestions: 文案建议
+ prompt: 自定义提示词（可选）
+
+ 返回:
+ Vue组件代码
+ """
+ print(f"{Fore.CYAN}🎨 开始增强Vue代码生成...{Style.RESET_ALL}")
+
+ # 1. 尝试使用预定义模板
+ template_code = get_template_by_images(parse_imglist)
+
+ if template_code and suggestions:
+ print(f"{Fore.GREEN}✅ 使用预定义模板生成Vue代码{Style.RESET_ALL}")
+ vue_code = fill_template_content(template_code, suggestions)
+ return vue_code
+
+ # 2. 如果没有合适的模板，使用AI生成
+ print(f"{Fore.BLUE}🤖 使用AI生成Vue代码{Style.RESET_ALL}")
+
+ if not prompt:
+ prompt = generate_layout_prompt(user_input_analysis_result, parse_imglist, suggestions)
+
+ return generate_vue_code(prompt)
+
+
def generate_vue_code(prompt=None):
"""
- 生成Vue组件代码
+ 原有的Vue代码生成函数（保持兼容性）
"""
if not prompt:
prompt = (
@@ -170,46 +355,32 @@ def generate_vue_code(prompt=None):
print(f"{Fore.RED}❌ Vue代码生成异常: {str(e)}{Style.RESET_ALL}")
return generate_fallback_vue_code()
-def generate_fallback_vue_code():
+def generate_fallback_vue_code(theme="默认主题", width=1080, height=1920):
"""
生成备用的Vue代码
"""
print(f"{Fore.YELLOW}⚠️ 使用备用Vue模板{Style.RESET_ALL}")
- return """
-
+
+
+
+
+
+
+ {{ title }}
+{{ subtitle }}
+
+
+
+
+
+
+ return f"""
+
-
端午节活动
-传统文化，现代体验
+{theme}
+精彩活动，敬请参与
-
@@ -217,124 +388,91 @@ def generate_fallback_vue_code():
"""
def save_code(code, file_path="../outputs/generated_code.vue"):
@@ -363,6 +501,25 @@ def save_code(code, file_path="../outputs/generated_code.vue"):
if __name__ == "__main__":
print(f"{Fore.MAGENTA}🚀 开始生成Vue组件...{Style.RESET_ALL}")
- vue_code = generate_vue_code()
+
+ # 测试增强的Vue代码生成
+ test_analysis = {
+ "width": 1080,
+ "height": 1920,
+ "main_theme": "端午节海报"
+ }
+
+ test_imglist = [
+ {"picture_name": "background", "picture_description": "背景图片"},
+ {"picture_name": "lotus", "picture_description": "荷花装饰"}
+ ]
+
+ test_suggestions = {
+ "layer6_title_content": {"content": "端午安康"},
+ "layer7_subtitle_content": {"content": "粽叶飘香，龙舟竞渡"},
+ "layer5_logo_content": {"text": "主办方"}
+ }
+
+ vue_code = generate_vue_code_enhanced(test_analysis, test_imglist, test_suggestions)
save_code(vue_code)
print(f"{Fore.GREEN}✅ Vue组件代码生成完成{Style.RESET_ALL}")
diff --git a/scripts/prompt_analysis.py b/scripts/prompt_analysis.py
deleted file mode 100644
index 18734e7..0000000
--- a/scripts/prompt_analysis.py
+++ /dev/null
@@ -1,90 +0,0 @@
-import os
-from dotenv import load_dotenv
-from generate_layout import call_deepseek
-import json
-from colorama import init, Fore, Back, Style
-
-# 初始化colorama
-init(autoreset=True)
-
-# 加载环境变量
-load_dotenv()
-
-def llm_user_analysis(user_input):
- """
- 使用DeepSeek动态分析用户输入，提取海报设计参数
- """
- if not user_input:
- user_input = "端午节海报，包含背景、活动亮点和图标"
-
- print(f"{Fore.CYAN}🔍 正在分析用户输入: {Style.BRIGHT}{user_input}{Style.RESET_ALL}")
-
- # 构建分析提示词
- analysis_prompt = f"""
- 请分析以下用户输入的海报需求，提取关键信息并以JSON格式返回：
-
- 用户输入：{user_input}
-
- 请严格按照以下JSON格式返回：
- {{
- "analyzed_prompt": "原始用户输入",
- "keywords": ["提取的关键词1", "关键词2", "关键词3"],
- "width": 1080,
- "height": 1920,
- "batch_size": 2,
- "poster_type": "海报类型（如：节日海报、活动海报、产品海报等）",
- "main_theme": "主要主题",
- "style_preference": "风格偏好（如：现代、传统、简约等）",
- "color_preference": "颜色偏好（如：暖色调、冷色调、单色等）"
- }}
-
- 注意：
- 1. keywords应该包含3-5个最重要的关键词
- 2. 根据用户输入推断合适的海报类型
- 3. 尺寸默认为1080x1920，除非用户明确指定
- 4. batch_size根据需求调整，通常为1-4
- 5. 分析用户的风格和颜色偏好
- """
-
- system_prompt = "你是一个专业的设计需求分析师，擅长从用户描述中提取海报设计的关键参数。请严格按照JSON格式返回结果，确保输出的JSON格式正确且完整。"
-
- try:
- result, _ = call_deepseek(prompt=analysis_prompt, system_prompt=system_prompt, temperature=0.3)
-
- # 解析JSON
- json_str = result.strip()
- if "```json" in json_str:
- json_str = json_str.split("```json")[1].split("```")[0].strip()
- elif json_str.startswith("```") and json_str.endswith("```"):
- json_str = json_str[3:-3].strip()
-
- analysis_result = json.loads(json_str)
-
- print(f"{Fore.GREEN}✅ 分析完成! {Style.RESET_ALL}")
- print(f"{Fore.YELLOW}📊 主题: {Style.BRIGHT}{analysis_result.get('main_theme', '未知')}{Style.RESET_ALL}")
- print(f"{Fore.YELLOW}🎨 风格: {analysis_result.get('style_preference', '未设置')}")
- print(f"{Fore.YELLOW}🔖 关键词: {', '.join(analysis_result.get('keywords', []))}{Style.RESET_ALL}")
-
- return analysis_result
-
- except Exception as e:
- print(f"{Fore.RED}❌ 分析失败: {str(e)}{Style.RESET_ALL}")
- # 返回默认值
- return {
- "analyzed_prompt": user_input,
- "keywords": ["海报", "设计", "活动"],
- "width": 1080,
- "height": 1920,
- "batch_size": 2,
- "poster_type": "通用海报",
- "main_theme": "默认主题",
- "style_preference": "现代",
- "color_preference": "暖色调"
- }
-
-if __name__ == "__main__":
- # 测试函数
- test_input = "春节海报，红色背景，包含灯笼和烟花，现代风格"
- result = llm_user_analysis(test_input)
- print(f"\n{Fore.GREEN}测试结果:")
- print(json.dumps(result, indent=2, ensure_ascii=False))
diff --git a/scripts/run_pipeline.py b/scripts/run_pipeline.py
index 276dd5a..20972a9 100644
--- a/scripts/run_pipeline.py
+++ b/scripts/run_pipeline.py
@@ -5,8 +5,8 @@
*
* @author 王秀强 (2310460@mail.nankai.edu.cn)
* @date 2025.6.9
- * @version v1.0.0
- *
+ * @version v2.0.0
+ *
* @details
* 本文件主要实现：
* - FastAPI服务器和RESTful API接口
@@ -34,38 +34,49 @@
* (c) 2025 砚生项目组
*/
"""
+
import os
+import sys
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
from dotenv import load_dotenv
-import yaml
-from prompt_analysis import llm_user_analysis
-from generate_layout import call_deepseek, generate_vue_code, save_code
-from generate_text import load_config_from_file, get_poster_content_suggestions
-from fastapi import FastAPI
-from fastapi.responses import FileResponse, JSONResponse
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
-from flux_con import comfyui_img_info
-from export_psd_from_json import create_psd_from_images as create_psd_impl
-from colorama import init, Fore, Style
import json
import shutil
import uuid
from datetime import datetime
+from typing import Optional
+
+# 导入工具函数
+from utils import (
+ print_step, print_result, get_session_folder,
+ llm_user_analysis, save_json_file, save_vue_file,
+ create_temp_config, CONFIG_PATHS
+)
+
+# 导入核心模块
+from generate_layout import generate_vue_code_enhanced, save_code
+from generate_text import load_config_from_file, get_poster_content_suggestions
+from flux_con import comfyui_img_info
+from export_psd_from_json import create_psd_from_images as create_psd_impl
+
+# FastAPI相关
+from fastapi import FastAPI
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+
+from colorama import init, Fore, Style
# 初始化colorama
init(autoreset=True)
-# 配置路径
-config_paths = {
- "font": "../configs/font.yaml",
- "output_folder": "../outputs/",
-}
-
-# 加载环境变量和配置
+# 加载环境变量
load_dotenv()
-app = FastAPI(title="AI海报生成系统API", version="1.0.0")
-# 添加CORS中间件，允许前端跨域访问
+# FastAPI应用
+app = FastAPI(title="AI海报生成系统API", version="2.0.0")
+
+# 添加CORS中间件
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # 在生产环境中应该设置具体的域名
@@ -74,143 +85,40 @@ app.add_middleware(
allow_headers=["*"],
)
-
# 请求模型
class PosterRequest(BaseModel):
- user_input: str
- session_id: str = None # 可选的会话ID，用于跟踪同一个用户的请求
-
-
-class GenerateVueRequest(BaseModel):
user_input: str
session_id: str = None
-
-class GeneratePSDRequest(BaseModel):
- user_input: str
- session_id: str = None
- use_manual_psd: bool = False # 是否使用手动创建的PSD文件
-
-
-# 响应模型
class ApiResponse(BaseModel):
status: str
message: str
data: dict = None
session_id: str = None
-
# 全局变量存储会话数据
sessions = {}
-# 加载字体配置
-try:
- with open(config_paths["font"], "r", encoding="utf-8") as f:
- fonts_config = yaml.safe_load(f)
- print(f"{Fore.GREEN}✅ 字体配置加载成功{Style.RESET_ALL}")
-except Exception as e:
- print(f"{Fore.YELLOW}⚠️ 字体配置加载失败: {e}，使用默认配置{Style.RESET_ALL}")
- fonts_config = {}
-
-# 辅助函数
-def print_step(step_num, description, status="进行中"):
- """打印带颜色的步骤信息"""
- if status == "进行中":
- print(f"{Fore.BLUE}📋 步骤{step_num}: {description}...{Style.RESET_ALL}")
- elif status == "完成":
- print(f"{Fore.GREEN}✅ 步骤{step_num}: {description} - 完成{Style.RESET_ALL}")
- elif status == "错误":
- print(f"{Fore.RED}❌ 步骤{step_num}: {description} - 出错{Style.RESET_ALL}")
-
-
-def print_result(key, value):
- """打印结果信息"""
- print(f"{Fore.CYAN}📊 {key}: {value}{Style.RESET_ALL}")
-
-
-def get_session_folder(session_id):
- """获取会话专用的输出文件夹"""
- if not session_id:
- session_id = str(uuid.uuid4())
-
- session_folder = os.path.join(config_paths["output_folder"], f"session_{session_id}")
- os.makedirs(session_folder, exist_ok=True)
- return session_folder, session_id
-
-
-# 生成prompts.yaml的函数
-def generate_prompts_yaml(user_input=None):
+def create_psd_from_fixed_images(output_path: str) -> Optional[str]:
"""
- 动态生成prompts.yaml配置文件
+ 使用固定的图片文件创建PSD文件
+ 简化的PSD创建函数，直接使用预定义的四个图片
"""
- if not user_input:
- user_input = "端午节海报，包含背景、活动亮点和图标"
-
- prompts_config = {
- "default_logo_text": "",
- "available_fonts": [
- {
- "name": "Microsoft YaHei",
- "displayName": "微软雅黑",
- "tags": ["现代", "清晰"],
- "roles": ["title", "subtitle", "content"]
- },
- {
- "name": "SimHei",
- "displayName": "黑体",
- "tags": ["通用", "标准"],
- "roles": ["title", "subtitle", "content"]
- }
- ],
- "NAMING_COLORS": {
- "primary": "#1976D2",
- "secondary": "#424242",
- "accent": "#FF5722"
- },
- "STYLE_RULES": {
- "modern": {
- "primary_font": "Microsoft YaHei",
- "secondary_font": "SimHei"
- }
- },
- "LOGO_RULES": {
- "default_position": "bottom",
- "fallback_text": "活动主办方"
- }
- }
-
- # 保存到临时文件
- temp_prompts_path = os.path.join(config_paths["output_folder"], "temp_prompts.yaml")
- os.makedirs(os.path.dirname(temp_prompts_path), exist_ok=True)
-
- with open(temp_prompts_path, 'w', encoding='utf-8') as f:
- yaml.dump(prompts_config, f, allow_unicode=True, default_flow_style=False)
-
- print(f"{Fore.GREEN}✅ 临时prompts.yaml已生成: {temp_prompts_path}{Style.RESET_ALL}")
- return temp_prompts_path
-
-
-# 修复PSD合成接口 - 使用临时固定图片列表
-def create_psd_from_images_wrapper(img_list, vue_layout_path, output_path):
- """
- 临时接口：使用固定的图片文件创建PSD文件
- 使用outputs目录下的: lotus.jpg, nankai.png, stamp.jpg, background.png
- """
- print(f"{Fore.CYAN}🎨 开始合成PSD文件(临时接口)...{Style.RESET_ALL}")
+ print(f"{Fore.CYAN}🎨 开始合成PSD文件...{Style.RESET_ALL}")
try:
# 使用固定的图片文件列表
- fixed_image_files = ["background.png", "lotus.jpg", "nankai.png", "stamp.jpg"]
+ fixed_image_files = ["background.png", "lotus.jpg", "nku.png", "stamp.jpg"]
image_paths = []
for img_file in fixed_image_files:
- img_path = os.path.join(config_paths["output_folder"], img_file)
+ img_path = os.path.join(CONFIG_PATHS["output_folder"], img_file)
if os.path.exists(img_path):
image_paths.append(img_path)
print(f"{Fore.GREEN}✓ 找到图片: {img_file}{Style.RESET_ALL}")
else:
- print(f"{Fore.YELLOW}⚠️ 图片不存在: {img_file} (路径: {img_path}){Style.RESET_ALL}")
+ print(f"{Fore.YELLOW}⚠️ 图片不存在: {img_file}{Style.RESET_ALL}")
if not image_paths:
print(f"{Fore.RED}❌ 没有找到任何指定的图片文件{Style.RESET_ALL}")
@@ -223,19 +131,19 @@ def create_psd_from_images_wrapper(img_list, vue_layout_path, output_path):
# 确保输出目录存在
os.makedirs(os.path.dirname(output_path), exist_ok=True)
- # 调用export_psd中的PSD创建函数
+ # 调用PSD创建函数
create_psd_impl(
image_paths=image_paths,
output_path=output_path,
- canvas_size=(1080, 1920), # 使用标准海报尺寸
+ canvas_size=(1080, 1920),
mode='RGB'
)
print(f"{Fore.GREEN}✅ PSD文件创建成功: {output_path}{Style.RESET_ALL}")
- # 验证PSD文件是否成功创建
+ # 验证PSD文件
if os.path.exists(output_path):
- file_size = os.path.getsize(output_path) / (1024 * 1024) # 转换为MB
+ file_size = os.path.getsize(output_path) / (1024 * 1024)
print(f"{Fore.CYAN}📁 PSD文件大小: {file_size:.2f} MB{Style.RESET_ALL}")
return output_path
@@ -247,212 +155,19 @@ def create_psd_from_images_wrapper(img_list, vue_layout_path, output_path):
return None
-# 增强Vue代码生成，确保包含文案内容
-def generate_layout_prompt(user_input_analysis_result, parse_imglist, suggestions=None):
+def run_pipeline(user_input: str = None) -> Optional[str]:
"""
- 生成更完整的Vue布局提示，包含文案内容
- """
- width = user_input_analysis_result["width"]
- height = user_input_analysis_result["height"]
- theme = user_input_analysis_result.get("main_theme", "活动海报")
-
- # 构造图片信息字符串
- images_info = "\n".join(
- [f"- {img['picture_name']} ({img['picture_description']})" for img in parse_imglist]
- )
-
- # 构造文案信息
- content_info = ""
- if suggestions:
- try:
- if 'layer6_title_content' in suggestions:
- title = suggestions['layer6_title_content'].get('content', theme)
- content_info += f"- 主标题: {title}\n"
-
- if 'layer7_subtitle_content' in suggestions:
- subtitle = suggestions['layer7_subtitle_content'].get('content', '精彩活动，敬请参与')
- content_info += f"- 副标题: {subtitle}\n"
-
- if 'layer5_logo_content' in suggestions:
- logo = suggestions['layer5_logo_content'].get('text', '主办方')
- content_info += f"- Logo文字: {logo}\n"
- except Exception as e:
- print(f"{Fore.YELLOW}⚠️ 文案信息解析错误: {e}{Style.RESET_ALL}")
- content_info = f"- 主标题: {theme}\n- 副标题: 精彩活动，敬请参与\n"
-
- # 调用DeepSeek生成动态排版Prompt
- system_prompt = "你是一个擅长前端开发的AI，专注于生成Vue.js代码。请根据提供的信息生成完整的Vue组件，包含所有必要的HTML结构和基础定位样式。"
-
- prompt = f"""
- 请生成一个Vue.js组件代码，用于{theme}海报，要求如下：
-
- 组件尺寸: {width}x{height}px
-
- 图片资源:
- {images_info}
-
- 文案内容:
- {content_info}
-
- 布局要求:
- 1. 背景图层: 使用第一张图片作为背景，占据整个组件区域
- 2. 主标题: 位于画布上方1/3处，居中显示
- 3. 副标题: 位于主标题下方，居中显示
- 4. 内容区域: 使用剩余图片，合理布局在中间区域
- 5. Logo区域: 位于底部，居中显示
-
- 技术要求:
- - 使用Vue 3 Composition API
- - 使用absolute定位进行精确布局
- - 包含完整的template、script和style部分
- - 确保所有文本内容都正确显示
- - 图片使用相对路径引用
-
- 请生成完整可用的Vue组件代码，不要包含任何说明文字。
- """
-
- try:
- result, _ = call_deepseek(prompt=prompt, system_prompt=system_prompt, temperature=0.4)
- return result
- except Exception as e:
- print(f"{Fore.RED}❌ 布局提示生成失败: {e}{Style.RESET_ALL}")
- return generate_fallback_vue_code(theme, width, height)
-
-
-def generate_fallback_vue_code(theme, width=1080, height=1920):
- """
- 生成备用的Vue代码
- """
- return f"""
-
-
- 
-
-
- 包粽子体验
-亲手制作传统粽子
-
- 
-
-
- 龙舟竞赛
-激情龙舟比赛
-
- 
-
+
- 文化表演
-精彩传统表演
-
+ 
-
-
-
-
-
-"""
-
-
-# 一键执行流程
-def run_pipeline(user_input=None):
- """
- 自动执行海报生成流程
+ 简化的海报生成流程
"""
try:
print(f"{Fore.MAGENTA}{'=' * 50}")
print(f"{Fore.MAGENTA}🎨 海报生成流程启动 🎨")
print(f"{'=' * 50}{Style.RESET_ALL}")
- print_step(1, "加载配置文件")
- prompts_yaml_path = generate_prompts_yaml(user_input)
- load_config_from_file(prompts_yaml_path)
- print_step(1, "加载配置文件", "完成")
+ print_step(1, "生成临时配置")
+ temp_config_path = create_temp_config(user_input)
+ load_config_from_file(temp_config_path)
+ print_step(1, "生成临时配置", "完成")
print_step(2, "分析用户输入")
user_input_analysis_result = llm_user_analysis(user_input)
@@ -467,40 +182,27 @@ def run_pipeline(user_input=None):
print_step(4, "生成文案建议")
suggestions = get_poster_content_suggestions(user_input_analysis_result["analyzed_prompt"])
- print(f"{Fore.CYAN}文案生成结果:")
- print(json.dumps(suggestions, indent=2, ensure_ascii=False))
# 保存文案到文件
- suggestions_path = os.path.join(config_paths["output_folder"], "poster_content.json")
- with open(suggestions_path, "w", encoding="utf-8") as f:
- json.dump(suggestions, f, indent=2, ensure_ascii=False)
+ suggestions_path = os.path.join(CONFIG_PATHS["output_folder"], "poster_content.json")
+ save_json_file(suggestions, suggestions_path)
print_step(4, "生成文案建议", "完成")
- print_step(5, "生成Vue排版")
- dynamic_prompt = generate_layout_prompt(user_input_analysis_result, parse_imglist, suggestions)
- vue_code = generate_vue_code(dynamic_prompt)
- vue_path = os.path.join(config_paths["output_folder"], "generated_code.vue")
- save_code(vue_code, file_path=vue_path)
+ print_step(5, "生成Vue组件")
+ # 使用增强的Vue代码生成（已移至generate_layout模块）
+ vue_code = generate_vue_code_enhanced(
+ user_input_analysis_result,
+ parse_imglist,
+ suggestions
+ )
- # 验证Vue文件是否成功生成
- if os.path.exists(vue_path):
- print(f"{Fore.GREEN}✅ Vue文件已生成: {vue_path}{Style.RESET_ALL}")
- # 显示Vue代码的前几行用于验证
- with open(vue_path, 'r', encoding='utf-8') as f:
- preview = f.read()[:500]
- print(f"{Fore.CYAN}Vue代码预览:\n{preview}...{Style.RESET_ALL}")
- else:
- print(f"{Fore.RED}❌ Vue文件生成失败{Style.RESET_ALL}")
- print_step(5, "生成Vue排版", "完成")
+ vue_path = os.path.join(CONFIG_PATHS["output_folder"], "generated_code.vue")
+ save_vue_file(vue_code, vue_path)
+ print_step(5, "生成Vue组件", "完成")
print_step(6, "合成PSD文件")
- img_list = [(pic["picture_name"], pic["picture_type"]) for pic in parse_imglist]
- psd_path = os.path.join(config_paths["output_folder"], "final_poster.psd")
- result_path = create_psd_from_images_wrapper(
- img_list=img_list,
- vue_layout_path=vue_path,
- output_path=psd_path
- )
+ psd_path = os.path.join(CONFIG_PATHS["output_folder"], "final_poster.psd")
+ result_path = create_psd_from_fixed_images(psd_path)
if result_path:
print_step(6, "合成PSD文件", "完成")
@@ -511,7 +213,7 @@ def run_pipeline(user_input=None):
print(f"✅ 流程执行完成！")
print(f"{'=' * 50}{Style.RESET_ALL}")
- return os.path.join(config_paths["output_folder"], "final_poster.png")
+ return result_path
except Exception as e:
print_step("X", f"Pipeline执行", "错误")
@@ -521,33 +223,38 @@ def run_pipeline(user_input=None):
return None
-# 本地运行函数
-def run_local_pipeline(user_input=None):
+def run_local_pipeline(user_input: str = None):
"""
- 本地运行整个管道流程，输出结果到控制台和文件系统。
+ 本地运行整个管道流程
"""
- print(f"{Fore.CYAN}🎬 Starting local pipeline with input: {Style.BRIGHT}{user_input}{Style.RESET_ALL}")
+ print(f"{Fore.CYAN}🎬 启动本地流程，输入: {Style.BRIGHT}{user_input}{Style.RESET_ALL}")
output_path = run_pipeline(user_input)
if output_path:
- print(f"{Fore.GREEN}🎊 Pipeline completed successfully!")
- print(f"{Fore.YELLOW}📁 Results saved to:")
- print(f" - Vue layout: {os.path.join(config_paths['output_folder'], 'generated_code.vue')}")
- print(f" - PSD file: {os.path.join(config_paths['output_folder'], 'final_poster.psd')}")
- print(f" - Content JSON: {os.path.join(config_paths['output_folder'], 'poster_content.json')}")
- print(f"{Fore.CYAN}💡 Check the outputs/ directory for generated files.{Style.RESET_ALL}")
+ print(f"{Fore.GREEN}🎊 流程执行成功!")
+ print(f"{Fore.YELLOW}📁 生成文件:")
+ print(f" - Vue组件: {os.path.join(CONFIG_PATHS['output_folder'], 'generated_code.vue')}")
+ print(f" - PSD文件: {os.path.join(CONFIG_PATHS['output_folder'], 'final_poster.psd')}")
+ print(f" - 文案JSON: {os.path.join(CONFIG_PATHS['output_folder'], 'poster_content.json')}")
+ print(f"{Fore.CYAN}💡 查看 outputs/ 目录获取生成的文件{Style.RESET_ALL}")
else:
- print(f"{Fore.RED}❌ Pipeline执行失败{Style.RESET_ALL}")
+ print(f"{Fore.RED}❌ 流程执行失败{Style.RESET_ALL}")
-# API路由
+# === API路由 ===
+
@app.get("/")
def read_root():
return {
- "message": "AI海报生成系统API",
- "version": "1.0.0",
+ "message": "AI海报生成系统API v2.0",
+ "version": "2.0.0",
+ "features": [
+ "预定义Vue模板支持",
+ "简化的代码结构",
+ "优化的流程管理"
+ ],
"endpoints": {
- "generate_poster": "/api/generate-poster", # 主要接口
+ "generate_poster": "/api/generate-poster",
"download": "/api/download/{file_type}",
"health": "/health",
"status": "/api/status/{session_id}"
@@ -557,14 +264,13 @@ def read_root():
@app.get("/health")
def health_check():
- """健康检查接口"""
return {"status": "healthy", "timestamp": datetime.now().isoformat()}
@app.post("/api/generate-poster", response_model=ApiResponse)
async def generate_poster_api(request: PosterRequest):
"""
- 一键生成完整海报（包含Vue代码和PSD文件）的主要API接口
+ 简化的海报生成API接口
"""
try:
session_folder, session_id = get_session_folder(request.session_id)
@@ -573,93 +279,35 @@ async def generate_poster_api(request: PosterRequest):
print(f"{Fore.CYAN}用户输入: {request.user_input}{Style.RESET_ALL}")
# === 步骤1: 生成配置文件 ===
- print(f"{Fore.BLUE}📋 步骤1: 生成配置文件{Style.RESET_ALL}")
- temp_prompts_path = os.path.join(session_folder, "temp_prompts.yaml")
- prompts_config = {
- "default_logo_text": "",
- "available_fonts": [
- {
- "name": "Microsoft YaHei",
- "displayName": "微软雅黑",
- "tags": ["现代", "清晰"],
- "roles": ["title", "subtitle", "content"]
- },
- {
- "name": "SimHei",
- "displayName": "黑体",
- "tags": ["通用", "标准"],
- "roles": ["title", "subtitle", "content"]
- }
- ]
- }
-
- with open(temp_prompts_path, 'w', encoding='utf-8') as f:
- yaml.dump(prompts_config, f, allow_unicode=True, default_flow_style=False)
-
- load_config_from_file(temp_prompts_path)
+ temp_config_path = create_temp_config(request.user_input)
+ load_config_from_file(temp_config_path)
# === 步骤2: 分析用户输入 ===
- print(f"{Fore.BLUE}📋 步骤2: 分析用户输入{Style.RESET_ALL}")
user_input_analysis_result = llm_user_analysis(request.user_input)
# === 步骤3: 生成图片信息 ===
- print(f"{Fore.BLUE}📋 步骤3: 生成图片信息{Style.RESET_ALL}")
system_prompt = user_input_analysis_result["analyzed_prompt"]
parse_imglist = comfyui_img_info(user_input_analysis_result, system_prompt)
# === 步骤4: 生成文案建议 ===
- print(f"{Fore.BLUE}📋 步骤4: 生成文案建议{Style.RESET_ALL}")
suggestions = get_poster_content_suggestions(user_input_analysis_result["analyzed_prompt"])
# 保存文案到会话文件夹
suggestions_path = os.path.join(session_folder, "poster_content.json")
- with open(suggestions_path, "w", encoding="utf-8") as f:
- json.dump(suggestions, f, indent=2, ensure_ascii=False)
+ save_json_file(suggestions, suggestions_path)
- # === 步骤5: 生成Vue排版 ===
- print(f"{Fore.BLUE}📋 步骤5: 生成Vue排版{Style.RESET_ALL}")
- dynamic_prompt = generate_layout_prompt(user_input_analysis_result, parse_imglist, suggestions)
- vue_code = generate_vue_code(dynamic_prompt)
+ # === 步骤5: 生成Vue组件 ===
+ vue_code = generate_vue_code_enhanced(
+ user_input_analysis_result,
+ parse_imglist,
+ suggestions
+ )
vue_path = os.path.join(session_folder, "generated_code.vue")
- save_code(vue_code, file_path=vue_path)
+ save_vue_file(vue_code, vue_path)
# === 步骤6: 合成PSD文件 ===
- print(f"{Fore.BLUE}📋 步骤6: 合成PSD文件{Style.RESET_ALL}")
- img_list = [(pic["picture_name"], pic["picture_type"]) for pic in parse_imglist]
psd_path = os.path.join(session_folder, "final_poster.psd")
-
- # 修复PSD创建调用
- try:
- # 使用固定的图片文件列表
- fixed_image_files = ["background.png", "lotus.jpg", "nankai.png", "stamp.jpg"]
- image_paths = []
-
- for img_file in fixed_image_files:
- img_path = os.path.join(config_paths["output_folder"], img_file)
- if os.path.exists(img_path):
- image_paths.append(img_path)
-
- if image_paths:
- # 确保输出目录存在
- os.makedirs(os.path.dirname(psd_path), exist_ok=True)
-
- # 调用PSD创建函数
- create_psd_impl(
- image_paths=image_paths,
- output_path=psd_path,
- canvas_size=(1080, 1920),
- mode='RGB'
- )
-
- print(f"{Fore.GREEN}✅ PSD文件创建成功: {psd_path}{Style.RESET_ALL}")
- psd_created = True
- else:
- print(f"{Fore.YELLOW}⚠️ 没有找到图片文件，跳过PSD创建{Style.RESET_ALL}")
- psd_created = False
-
- except Exception as psd_error:
- print(f"{Fore.RED}❌ PSD文件创建失败: {str(psd_error)}{Style.RESET_ALL}")
- psd_created = False
+ psd_created = create_psd_from_fixed_images(psd_path)
# 返回API响应
return ApiResponse(
@@ -671,7 +319,10 @@ async def generate_poster_api(request: PosterRequest):
"content_file": suggestions_path,
"analysis_result": user_input_analysis_result,
"images_info": parse_imglist,
- "suggestions": suggestions
+ "suggestions": suggestions,
+ "vue_code": vue_code,
+ "file_size_mb": round(os.path.getsize(psd_path) / (1024 * 1024), 2) if psd_created else 0,
+ "generated_images": len(parse_imglist)
},
session_id=session_id
)
@@ -691,62 +342,46 @@ async def generate_poster_api(request: PosterRequest):
@app.get("/api/download/{file_type}")
async def download_file(file_type: str, session_id: str = None):
- """
- 下载生成的文件
- """
+ """下载生成的文件"""
try:
if session_id:
- session_folder = os.path.join(config_paths["output_folder"], f"session_{session_id}")
+ session_folder = os.path.join(CONFIG_PATHS["output_folder"], f"session_{session_id}")
else:
- session_folder = config_paths["output_folder"]
+ session_folder = CONFIG_PATHS["output_folder"]
- if file_type == "vue":
- file_path = os.path.join(session_folder, "generated_code.vue")
- media_type = "text/plain"
- elif file_type == "psd":
- file_path = os.path.join(session_folder, "final_poster.psd")
- media_type = "application/octet-stream"
- elif file_type == "json":
- file_path = os.path.join(session_folder, "poster_content.json")
- media_type = "application/json"
- else:
- return JSONResponse(
- status_code=400,
- content={"error": "不支持的文件类型"}
- )
+ file_mapping = {
+ "vue": ("generated_code.vue", "text/plain"),
+ "psd": ("final_poster.psd", "application/octet-stream"),
+ "json": ("poster_content.json", "application/json")
+ }
+
+ if file_type not in file_mapping:
+ return JSONResponse(status_code=400, content={"error": "不支持的文件类型"})
+
+ filename, media_type = file_mapping[file_type]
+ file_path = os.path.join(session_folder, filename)
if os.path.exists(file_path):
return FileResponse(
path=file_path,
media_type=media_type,
- filename=os.path.basename(file_path)
+ filename=filename
)
else:
- return JSONResponse(
- status_code=404,
- content={"error": "文件不存在"}
- )
+ return JSONResponse(status_code=404, content={"error": "文件不存在"})
except Exception as e:
- return JSONResponse(
- status_code=500,
- content={"error": f"下载失败: {str(e)}"}
- )
+ return JSONResponse(status_code=500, content={"error": f"下载失败: {str(e)}"})
@app.get("/api/status/{session_id}")
async def get_session_status(session_id: str):
- """
- 获取会话状态
- """
+ """获取会话状态"""
try:
- session_folder = os.path.join(config_paths["output_folder"], f"session_{session_id}")
+ session_folder = os.path.join(CONFIG_PATHS["output_folder"], f"session_{session_id}")
if not os.path.exists(session_folder):
- return JSONResponse(
- status_code=404,
- content={"error": "会话不存在"}
- )
+ return JSONResponse(status_code=404, content={"error": "会话不存在"})
files_status = {
"vue_file": os.path.exists(os.path.join(session_folder, "generated_code.vue")),
@@ -761,16 +396,12 @@ async def get_session_status(session_id: str):
}
except Exception as e:
- return JSONResponse(
- status_code=500,
- content={"error": f"状态查询失败: {str(e)}"}
- )
+ return JSONResponse(status_code=500, content={"error": f"状态查询失败: {str(e)}"})
if __name__ == "__main__":
import uvicorn
- # 启动本地运行（可选）
print(f"{Fore.BLUE}🔧 运行模式选择:{Style.RESET_ALL}")
print(f"{Fore.YELLOW}1. 本地测试模式")
print(f"2. API服务器模式{Style.RESET_ALL}")
@@ -781,9 +412,9 @@ if __name__ == "__main__":
# 本地测试
test_input = input("请输入海报需求 (留空使用默认): ").strip()
if not test_input:
- test_input = "端午节安康，"
+ test_input = "端午节海报，传统风格"
run_local_pipeline(test_input)
else:
# 启动API服务器
- print(f"{Fore.GREEN}🚀 启动API服务器...{Style.RESET_ALL}")
+ print(f"{Fore.GREEN}🚀 启动API服务器 v2.0...{Style.RESET_ALL}")
uvicorn.run(app, host="0.0.0.0", port=8000)
diff --git a/scripts/utils.py b/scripts/utils.py
new file mode 100644
index 0000000..9e55062
--- /dev/null
+++ b/scripts/utils.py
@@ -0,0 +1,266 @@
+# -*- coding: utf-8 -*-
+"""
+工具函数模块
+整合所有辅助函数和公共功能
+"""
+
+import os
+import uuid
+import json
+import yaml
+from datetime import datetime
+from dotenv import load_dotenv
+from colorama import init, Fore, Style
+from pathlib import Path
+from typing import Dict, List, Optional, Tuple
+
+# 初始化colorama
+init(autoreset=True)
+
+# 加载环境变量
+load_dotenv()
+
+# 配置路径
+CONFIG_PATHS = {
+ "font": "../configs/font.yaml",
+ "output_folder": "../outputs/",
+ "workflows": "../workflows/",
+ "templates": "../configs/templates.yaml"
+}
+
+
+def print_step(step_num: int, description: str, status: str = "进行中"):
+ """打印带颜色的步骤信息"""
+ if status == "进行中":
+ print(f"{Fore.BLUE}📋 步骤{step_num}: {description}...{Style.RESET_ALL}")
+ elif status == "完成":
+ print(f"{Fore.GREEN}✅ 步骤{step_num}: {description} - 完成{Style.RESET_ALL}")
+ elif status == "错误":
+ print(f"{Fore.RED}❌ 步骤{step_num}: {description} - 出错{Style.RESET_ALL}")
+
+
+def print_result(key: str, value: str):
+ """打印结果信息"""
+ print(f"{Fore.CYAN}📊 {key}: {value}{Style.RESET_ALL}")
+
+
+def get_session_folder(session_id: Optional[str] = None) -> Tuple[str, str]:
+ """获取会话专用的输出文件夹"""
+ if not session_id:
+ session_id = str(uuid.uuid4())
+
+ session_folder = os.path.join(CONFIG_PATHS["output_folder"], f"session_{session_id}")
+ os.makedirs(session_folder, exist_ok=True)
+ return session_folder, session_id
+
+
+def load_config_file(config_type: str) -> Dict:
+ """加载配置文件"""
+ config_path = CONFIG_PATHS.get(config_type)
+ if not config_path or not os.path.exists(config_path):
+ print(f"{Fore.YELLOW}⚠️ 配置文件不存在: {config_path}，使用默认配置{Style.RESET_ALL}")
+ return {}
+
+ try:
+ with open(config_path, "r", encoding="utf-8") as f:
+ if config_path.endswith('.yaml') or config_path.endswith('.yml'):
+ config = yaml.safe_load(f)
+ else:
+ config = json.load(f)
+ print(f"{Fore.GREEN}✅ 配置文件加载成功: {config_path}{Style.RESET_ALL}")
+ return config
+ except Exception as e:
+ print(f"{Fore.RED}❌ 配置文件加载失败: {e}{Style.RESET_ALL}")
+ return {}
+
+
+def save_json_file(data: Dict, file_path: str):
+ """保存JSON文件"""
+ try:
+ os.makedirs(os.path.dirname(file_path), exist_ok=True)
+ with open(file_path, "w", encoding="utf-8") as f:
+ json.dump(data, f, indent=2, ensure_ascii=False)
+ print(f"{Fore.GREEN}✅ JSON文件已保存: {file_path}{Style.RESET_ALL}")
+ except Exception as e:
+ print(f"{Fore.RED}❌ JSON文件保存失败: {e}{Style.RESET_ALL}")
+
+
+def save_vue_file(vue_code: str, file_path: str):
+ """保存Vue文件"""
+ try:
+ os.makedirs(os.path.dirname(file_path), exist_ok=True)
+ with open(file_path, "w", encoding="utf-8") as f:
+ f.write(vue_code)
+
+ if os.path.exists(file_path):
+ file_size = os.path.getsize(file_path)
+ print(f"{Fore.GREEN}✅ Vue文件已保存: {file_path} ({file_size} 字节){Style.RESET_ALL}")
+ else:
+ print(f"{Fore.RED}❌ Vue文件保存失败{Style.RESET_ALL}")
+ except Exception as e:
+ print(f"{Fore.RED}❌ Vue文件保存失败: {e}{Style.RESET_ALL}")
+
+
+def create_temp_config(user_input: str = None) -> str:
+ """动态生成临时配置文件"""
+ if not user_input:
+ user_input = "默认海报配置"
+
+ temp_config = {
+ "default_logo_text": "",
+ "available_fonts": [
+ {
+ "name": "Microsoft YaHei",
+ "displayName": "微软雅黑",
+ "tags": ["现代", "清晰"],
+ "roles": ["title", "subtitle", "content"]
+ },
+ {
+ "name": "SimHei",
+ "displayName": "黑体",
+ "tags": ["通用", "标准"],
+ "roles": ["title", "subtitle", "content"]
+ }
+ ],
+ "vue_templates": {
+ "lotus.jpg": {
+ "theme": "荷花主题",
+ "style": "传统优雅",
+ "template_name": "lotus_template"
+ },
+ "nku.png": {
+ "theme": "南开大学",
+ "style": "学术正式",
+ "template_name": "nku_template"
+ },
+ "stamp.jpg": {
+ "theme": "印章装饰",
+ "style": "传统文化",
+ "template_name": "stamp_template"
+ },
+ "background.png": {
+ "theme": "通用背景",
+ "style": "简约现代",
+ "template_name": "background_template"
+ }
+ }
+ }
+
+ temp_config_path = os.path.join(CONFIG_PATHS["output_folder"], "temp_config.yaml")
+ with open(temp_config_path, 'w', encoding='utf-8') as f:
+ yaml.dump(temp_config, f, allow_unicode=True, default_flow_style=False)
+
+ print(f"{Fore.GREEN}✅ 临时配置文件已生成: {temp_config_path}{Style.RESET_ALL}")
+ return temp_config_path
+
+
+# === 用户输入分析模块 (从 prompt_analysis.py 整合) ===
+
+from generate_layout import call_deepseek
+
+def llm_user_analysis(user_input: str) -> Dict:
+ """
+ 使用DeepSeek动态分析用户输入，提取海报设计参数
+ """
+ if not user_input:
+ user_input = "端午节海报，包含背景、活动亮点和图标"
+
+ print(f"{Fore.CYAN}🔍 正在分析用户输入: {Style.BRIGHT}{user_input}{Style.RESET_ALL}")
+
+ # 构建分析提示词
+ analysis_prompt = f"""
+ 请分析以下用户输入的海报需求，提取关键信息并以JSON格式返回：
+
+ 用户输入：{user_input}
+
+ 请严格按照以下JSON格式返回：
+ {{
+ "analyzed_prompt": "原始用户输入",
+ "keywords": ["提取的关键词1", "关键词2", "关键词3"],
+ "width": 1080,
+ "height": 1920,
+ "batch_size": 2,
+ "poster_type": "海报类型（如：节日海报、活动海报、产品海报等）",
+ "main_theme": "主要主题",
+ "style_preference": "风格偏好（如：现代、传统、简约等）",
+ "color_preference": "颜色偏好（如：暖色调、冷色调、单色等）"
+ }}
+
+ 注意：
+ 1. keywords应该包含3-5个最重要的关键词
+ 2. 根据用户输入推断合适的海报类型
+ 3. 尺寸默认为1080x1920，除非用户明确指定
+ 4. batch_size根据需求调整，通常为1-4
+ 5. 分析用户的风格和颜色偏好
+ """
+
+ system_prompt = "你是一个专业的设计需求分析师，擅长从用户描述中提取海报设计的关键参数。请严格按照JSON格式返回结果，确保输出的JSON格式正确且完整。"
+
+ try:
+ result, _ = call_deepseek(prompt=analysis_prompt, system_prompt=system_prompt, temperature=0.3)
+
+ # 解析JSON
+ json_str = result.strip()
+ if "```json" in json_str:
+ json_str = json_str.split("```json")[1].split("```")[0].strip()
+ elif json_str.startswith("```") and json_str.endswith("```"):
+ json_str = json_str[3:-3].strip()
+
+ analysis_result = json.loads(json_str)
+
+ print(f"{Fore.GREEN}✅ 分析完成! {Style.RESET_ALL}")
+ print(f"{Fore.YELLOW}📊 主题: {Style.BRIGHT}{analysis_result.get('main_theme', '未知')}{Style.RESET_ALL}")
+ print(f"{Fore.YELLOW}🎨 风格: {analysis_result.get('style_preference', '未设置')}")
+ print(f"{Fore.YELLOW}🔖 关键词: {', '.join(analysis_result.get('keywords', []))}{Style.RESET_ALL}")
+
+ return analysis_result
+
+ except Exception as e:
+ print(f"{Fore.RED}❌ 分析失败: {str(e)}{Style.RESET_ALL}")
+ # 返回默认值
+ return {
+ "analyzed_prompt": user_input,
+ "keywords": ["海报", "设计", "活动"],
+ "width": 1080,
+ "height": 1920,
+ "batch_size": 2,
+ "poster_type": "通用海报",
+ "main_theme": "默认主题",
+ "style_preference": "现代",
+ "color_preference": "暖色调"
+ }
+
+
+def validate_file_exists(file_path: str) -> bool:
+ """验证文件是否存在"""
+ exists = os.path.exists(file_path)
+ if exists:
+ print(f"{Fore.GREEN}✓ 文件存在: {os.path.basename(file_path)}{Style.RESET_ALL}")
+ else:
+ print(f"{Fore.YELLOW}⚠️ 文件不存在: {os.path.basename(file_path)}{Style.RESET_ALL}")
+ return exists
+
+
+def get_file_info(file_path: str) -> Dict:
+ """获取文件信息"""
+ if not os.path.exists(file_path):
+ return {"exists": False}
+
+ stat = os.stat(file_path)
+ return {
+ "exists": True,
+ "size": stat.st_size,
+ "size_mb": round(stat.st_size / (1024 * 1024), 2),
+ "modified": datetime.fromtimestamp(stat.st_mtime).isoformat()
+ }
+
+
+if __name__ == "__main__":
+ # 测试工具函数
+ print(f"{Fore.MAGENTA}🧪 测试工具函数{Style.RESET_ALL}")
+
+ # 测试用户输入分析
+ test_input = "春节海报，红色背景，现代风格"
+ result = llm_user_analysis(test_input)
+ print(f"\n{Fore.GREEN}分析结果:")
+ print(json.dumps(result, indent=2, ensure_ascii=False))
diff --git a/workflows/temp/processed_workflow_affb62095da0450595bb4cf2334b173b.json b/workflows/temp/processed_workflow_affb62095da0450595bb4cf2334b173b.json
new file mode 100644
index 0000000..20c684b
--- /dev/null
+++ b/workflows/temp/processed_workflow_affb62095da0450595bb4cf2334b173b.json
@@ -0,0 +1,226 @@
+{
+ "6": {
+ "inputs": {
+ "text": [
+ "31",
+ 0
+ ],
+ "clip": [
+ "11",
+ 0
+ ]
+ },
+ "class_type": "CLIPTextEncode",
+ "_meta": {
+ "title": "CLIP Text Encode (Positive Prompt)"
+ }
+ },
+ "8": {
+ "inputs": {
+ "samples": [
+ "13",
+ 0
+ ],
+ "vae": [
+ "10",
+ 0
+ ]
+ },
+ "class_type": "VAEDecode",
+ "_meta": {
+ "title": "VAE解码"
+ }
+ },
+ "9": {
+ "inputs": {
+ "filename_prefix": "ComfyUI",
+ "images": [
+ "8",
+ 0
+ ]
+ },
+ "class_type": "SaveImage",
+ "_meta": {
+ "title": "保存图像"
+ }
+ },
+ "10": {
+ "inputs": {
+ "vae_name": "ae.safetensors"
+ },
+ "class_type": "VAELoader",
+ "_meta": {
+ "title": "加载VAE"
+ }
+ },
+ "11": {
+ "inputs": {
+ "clip_name1": "t5xxl_fp16.safetensors",
+ "clip_name2": "clip_l.safetensors",
+ "type": "flux",
+ "device": "default"
+ },
+ "class_type": "DualCLIPLoader",
+ "_meta": {
+ "title": "双CLIP加载器"
+ }
+ },
+ "12": {
+ "inputs": {
+ "unet_name": "flux1-dev.safetensors",
+ "weight_dtype": "default"
+ },
+ "class_type": "UNETLoader",
+ "_meta": {
+ "title": "UNet加载器"
+ }
+ },
+ "13": {
+ "inputs": {
+ "noise": [
+ "25",
+ 0
+ ],
+ "guider": [
+ "22",
+ 0
+ ],
+ "sampler": [
+ "16",
+ 0
+ ],
+ "sigmas": [
+ "17",
+ 0
+ ],
+ "latent_image": [
+ "27",
+ 0
+ ]
+ },
+ "class_type": "SamplerCustomAdvanced",
+ "_meta": {
+ "title": "自定义采样器（高级）"
+ }
+ },
+ "16": {
+ "inputs": {
+ "sampler_name": "euler"
+ },
+ "class_type": "KSamplerSelect",
+ "_meta": {
+ "title": "K采样器选择"
+ }
+ },
+ "17": {
+ "inputs": {
+ "scheduler": "simple",
+ "steps": 25,
+ "denoise": 1,
+ "model": [
+ "30",
+ 0
+ ]
+ },
+ "class_type": "BasicScheduler",
+ "_meta": {
+ "title": "基本调度器"
+ }
+ },
+ "22": {
+ "inputs": {
+ "model": [
+ "30",
+ 0
+ ],
+ "conditioning": [
+ "26",
+ 0
+ ]
+ },
+ "class_type": "BasicGuider",
+ "_meta": {
+ "title": "基本引导器"
+ }
+ },
+ "25": {
+ "inputs": {
+ "noise_seed": 142213168350829
+ },
+ "class_type": "RandomNoise",
+ "_meta": {
+ "title": "随机噪波"
+ }
+ },
+ "26": {
+ "inputs": {
+ "guidance": 3.5,
+ "conditioning": [
+ "6",
+ 0
+ ]
+ },
+ "class_type": "FluxGuidance",
+ "_meta": {
+ "title": "Flux引导"
+ }
+ },
+ "27": {
+ "inputs": {
+ "width": "1080",
+ "height": "1920",
+ "batch_size": "2"
+ },
+ "class_type": "EmptySD3LatentImage",
+ "_meta": {
+ "title": "空Latent图像（SD3）"
+ }
+ },
+ "30": {
+ "inputs": {
+ "max_shift": 1.1500000000000001,
+ "base_shift": 0.5000000000000001,
+ "width": 1024,
+ "height": 1024,
+ "model": [
+ "12",
+ 0
+ ]
+ },
+ "class_type": "ModelSamplingFlux",
+ "_meta": {
+ "title": "采样算法（Flux）"
+ }
+ },
+ "31": {
+ "inputs": {
+ "user_prompt": "作为一个AI提示词专家，请你仿照范例，根据我给出的主题，生成一条符合下列要求的提示词，来让CLIP模型可以更好地理解画面主体。注意：你需要仿照下面的示例详细分析（仅仿照写法，而不仿照任何内容），将用户的需求转化为详细的提示词。\\n 要求共六条，请严格遵守：\\n 1： 用自然语言简单句来描述画面，请避免出现过于长的，或者格式过于复杂的句子，句子中不要出现*等特殊符号。\\n 2.用英语表达。 \\n 3.直接给出prompt内容即可，不需要任何解释和说明。\\n 4. 每条prompt至少50词，不超过200词。\\n 5.避免模棱两可的说法。\\n 6.描述的最开始加入：“no text, no AI style”\\n 例如：\nCartoon-Style Nankai University Main Building,\nvividly depicted with rounded edges and pastel gradients, the iconic Gothic-Revival structure stands majestically. Crimson brick façade contrasts with golden-glowing arched windows, while whimsical cloud-shaped eaves drip melted clock details. A giant smiling sun hangs low, casting honey-golden rays through simplified pine trees, creating striped shadows dancing on marble stairs.\n\nTranslucent ghostly scholars from 1919 float near pillars holding glowing books, their outlines shimmering like liquid mercury. Oversized autumn leaves (stylized as maple-red origami) spiral around twin bell towers chiming visible musical notes. Puddles on the ground mirror upside-down building reflections rippling with calculus formulas.\n\nEnvironment:\nSurreal candy-pink sunset gradients blend into starry indigo sky above. Playful squirrels wearing tiny graduation caps scamper across emerald lawns textured like green velvet.",
+ "system_prompt": "端午节海报，传统风格",
+ "enable_history": false,
+ "max_history": 10,
+ "history_json": "",
+ "save_path": "./chat_history.json",
+ "api_config": [
+ "32",
+ 0
+ ]
+ },
+ "class_type": "DeepSeekChat",
+ "_meta": {
+ "title": "LLM Model Input Box"
+ }
+ },
+ "32": {
+ "inputs": {
+ "config_name": "DeepSeek官方Chat API",
+ "temperature": 1,
+ "max_tokens": 512,
+ "stream": false,
+ "api_key": ""
+ },
+ "class_type": "DeepSeekAPIConfig",
+ "_meta": {
+ "title": "LLM API Model Selector"
+ }
+ }
+}
\ No newline at end of file
diff --git a/接口文档.md b/接口文档.md
index 847480f..a71986d 100644
--- a/接口文档.md
+++ b/接口文档.md
@@ -1,26 +1,18 @@
-# AI海报生成系统 接口文档
-
-## 概述
-
-AI海报生成系统是一个基于FastAPI的Web服务，集成了多个AI模型和图像处理功能，能够一键生成Vue组件代码和PSD文件。
+# AI海报生成系统 API文档
## 系统架构
```
-前端 → FastAPI服务器(8000端口) → ComfyUI服务器(101.201.50.90:8188)
+前端 → FastAPI(8000) → ComfyUI(101.201.50.90:8188)
↓
- Vue代码生成 + PSD文件合成
+ Vue代码 + PSD文件
```
## API端点
### 1. 健康检查
-
**GET** `/health`
-**功能**: 检查服务器状态
-
-**响应**:
```json
{
"status": "healthy",
@@ -28,17 +20,14 @@ AI海报生成系统是一个基于FastAPI的Web服务，集成了多个AI模型
}
```
-### 2. 主要接口 - 生成海报
-
+### 2. 生成海报
**POST** `/api/generate-poster`
-**功能**: 一键生成完整海报（Vue代码 + PSD文件）
-
-**请求体**:
+**请求**:
```json
{
- "user_input": "端午节海报，传统风格，包含荷花和龙舟",
- "session_id": "可选-用于跟踪会话"
+ "user_input": "端午节海报，传统风格",
+ "session_id": "可选"
}
```
@@ -48,14 +37,14 @@ AI海报生成系统是一个基于FastAPI的Web服务，集成了多个AI模型
"status": "success",
"message": "海报生成完成",
"data": {
- "vue_code": "完整的Vue组件代码",
+ "vue_code": "完整Vue组件代码",
"suggestions": {
"layer5_logo_content": {
"text": "主办方",
"color": "#000000"
},
"layer6_title_content": {
- "content": "端午节安康",
+ "content": "端午安康",
"font_name": "SimHei",
"color": "#7E0C6E"
},
@@ -66,259 +55,97 @@ AI海报生成系统是一个基于FastAPI的Web服务，集成了多个AI模型
}
},
"analysis_result": {
- "analyzed_prompt": "端午节海报，传统风格",
"main_theme": "端午节祝福",
"width": 1080,
"height": 1920,
- "keywords": ["端午节", "传统", "荷花", "龙舟"]
+ "keywords": ["端午节", "传统", "荷花"]
},
- "psd_file_path": "生成的PSD文件路径",
"file_size_mb": 5.93,
- "generated_images": 2,
- "files": {
- "vue_file": "Vue文件路径",
- "psd_file": "PSD文件路径"
- }
+ "generated_images": 2
},
"session_id": "会话ID"
}
```
### 3. 文件下载
-
**GET** `/api/download/{file_type}?session_id={session_id}`
-**功能**: 下载生成的文件
-
-**参数**:
-- `file_type`: 文件类型 (`vue` | `psd` | `json`)
-- `session_id`: 会话ID（查询参数）
-
-**响应**: 直接返回文件下载
-
-### 4. 获取生成状态
+参数: `file_type` = `vue` | `psd` | `json`
+### 4. 会话状态
**GET** `/api/status/{session_id}`
-**功能**: 获取指定会话的生成状态和信息
-
-**响应**:
```json
{
- "status": "success",
- "message": "状态获取成功",
- "data": {
- "user_input": "用户输入",
- "analysis_result": "分析结果",
- "suggestions": "文案建议",
- "vue_path": "Vue文件路径",
- "psd_path": "PSD文件路径",
- "psd_size_mb": 5.93,
- "created_at": "2025-01-02T20:30:00"
+ "session_id": "会话ID",
+ "files": {
+ "vue_file": true,
+ "psd_file": true,
+ "content_file": true
},
- "session_id": "会话ID"
+ "folder": "会话文件夹路径"
}
```
-## 内部模块接口
-
-### 1. prompt_analysis
-
-**函数**: `llm_user_analysis(user_input)`
-
-**功能**: 使用DeepSeek模型分析用户输入
-
-**参数**:
-- `user_input` (str): 用户输入的海报需求
-
-**返回值**:
-```json
-{
- "analyzed_prompt": "原始用户输入",
- "keywords": ["关键词1", "关键词2"],
- "width": 1080,
- "height": 1920,
- "batch_size": 2,
- "poster_type": "海报类型",
- "main_theme": "主要主题",
- "style_preference": "风格偏好",
- "color_preference": "颜色偏好"
-}
-```
-
-### 2. flux_con
-
-**函数**: `comfyui_img_info(user_input_analysis_result, system_prompt)`
-
-**功能**: 调用ComfyUI生成图片
-
-**参数**:
-- `user_input_analysis_result`: 用户输入分析结果
-- `system_prompt`: 系统提示词
-
-**返回值**:
-```json
-[
- {
- "picture_name": "图片文件名",
- "picture_type": "png",
- "picture_description": "图片描述",
- "picture_size": "1080x1920"
- }
-]
-```
-
-### 3. generate_text
-
-**函数**: `get_poster_content_suggestions(analyzed_prompt)`
-
-**功能**: 使用Kimi生成文案建议
-
-**参数**:
-- `analyzed_prompt` (str): 分析后的提示词
-
-**返回值**:
-```json
-{
- "layer5_logo_content": {
- "text": "Logo文字",
- "color": "#000000"
- },
- "layer6_title_content": {
- "content": "主标题",
- "font_name": "字体名称",
- "color": "#颜色代码"
- },
- "layer7_subtitle_content": {
- "content": "副标题",
- "font_name": "字体名称",
- "color": "#颜色代码"
- }
-}
-```
-
-### 4. generate_layout
-
-**函数**: `generate_vue_code(prompt)`
-
-**功能**: 使用DeepSeek生成Vue组件代码
-
-**参数**:
-- `prompt` (str): 包含布局要求的提示词
-
-**返回值**: Vue组件代码字符串
-
-### 5. export_psd_from_json
-
-**函数**: `create_psd_from_images(image_paths, output_path, canvas_size, mode)`
-
-**功能**: 创建PSD文件
-
-**参数**:
-- `image_paths` (List[str]): 图片路径列表
-- `output_path` (str): 输出PSD文件路径
-- `canvas_size` (Tuple[int, int]): 画布大小
-- `mode` (str): 颜色模式
-
-## 错误处理
-
-**错误响应格式**:
-```json
-{
- "detail": "错误描述信息"
-}
-```
-
-**常见错误码**:
-- `400`: 请求参数错误
-- `404`: 资源不存在（会话不存在、文件不存在）
-- `500`: 服务器内部错误
-
## 使用示例
-### JavaScript调用示例
-
+### JavaScript
```javascript
-// 生成海报
-async function generatePoster(userInput) {
- const response = await fetch('http://localhost:8000/api/generate-poster', {
- method: 'POST',
- headers: { 'Content-Type': 'application/json' },
- body: JSON.stringify({ user_input: userInput })
- });
-
- const result = await response.json();
-
- if (result.status === 'success') {
- console.log('Vue代码:', result.data.vue_code);
- console.log('文案建议:', result.data.suggestions);
-
- // 下载文件
- downloadFile(result.session_id, 'vue');
- downloadFile(result.session_id, 'psd');
- }
-}
+const response = await fetch('http://localhost:8000/api/generate-poster', {
+ method: 'POST',
+ headers: { 'Content-Type': 'application/json' },
+ body: JSON.stringify({ user_input: "春节海报，红色背景" })
+});
-// 下载文件
-function downloadFile(sessionId, fileType) {
- const url = `http://localhost:8000/api/download/${fileType}?session_id=${sessionId}`;
- window.open(url, '_blank');
+const result = await response.json();
+if (result.status === 'success') {
+ // 下载文件
+ const url = `http://localhost:8000/api/download/psd?session_id=${result.session_id}`;
+ window.open(url);
}
-
-// 使用示例
-generatePoster("春节海报，红色背景，现代风格");
```
-### Python调用示例
-
+### Python
```python
import requests
-# 生成海报
-def generate_poster(user_input):
- url = "http://localhost:8000/api/generate-poster"
- data = {"user_input": user_input}
-
- response = requests.post(url, json=data)
- result = response.json()
-
- if result["status"] == "success":
- print("Vue代码:", result["data"]["vue_code"])
- session_id = result["session_id"]
-
- # 下载PSD文件
- download_url = f"http://localhost:8000/api/download/psd?session_id={session_id}"
- psd_response = requests.get(download_url)
-
- with open("poster.psd", "wb") as f:
- f.write(psd_response.content)
+response = requests.post('http://localhost:8000/api/generate-poster',
+ json={"user_input": "端午节海报，传统风格"})
+result = response.json()
-# 使用示例
-generate_poster("端午节海报，传统风格")
+if result["status"] == "success":
+ session_id = result["session_id"]
+ # 下载PSD文件
+ psd_url = f"http://localhost:8000/api/download/psd?session_id={session_id}"
+ psd_response = requests.get(psd_url)
+ with open("poster.psd", "wb") as f:
+ f.write(psd_response.content)
```
-## 工作流程
+## 错误处理
-1. **接收请求**: 前端发送用户输入到 `/api/generate-poster`
-2. **分析输入**: 使用DeepSeek分析用户需求
-3. **生成图片**: 调用ComfyUI(101.201.50.90:8188)生成图片
-4. **生成文案**: 使用Kimi生成文案建议
-5. **生成Vue代码**: 使用DeepSeek生成Vue组件
-6. **创建PSD**: 合成PSD文件（优先使用手动模板）
-7. **返回结果**: 一次性返回所有生成内容
-8. **文件下载**: 前端可按需下载生成的文件
+```json
+{
+ "status": "error",
+ "message": "错误描述",
+ "data": null
+}
+```
-## 部署说明
+常见错误码:
+- `400`: 请求参数错误
+- `404`: 资源不存在
+- `500`: 服务器内部错误
+
+## 部署
### 开发环境
```bash
-cd E:\砚生\ai_service\scripts
-python run_pipeline.py
-# 选择: 2 (API服务器模式)
+cd scripts
+python run_pipeline.py # 选择2
```
### 生产环境
-- 使用PM2管理进程
-- 配置Nginx反向代理
-- 设置CORS允许的域名
-- 配置HTTPS
+```bash
+uvicorn scripts.run_pipeline:app --host 0.0.0.0 --port 8000
+```
diff --git a/部署方案.md b/部署方案.md
deleted file mode 100644
index 7709caa..0000000
--- a/部署方案.md
+++ /dev/null
@@ -1,243 +0,0 @@
-# AI海报生成系统 - 简化部署方案
-
-## 🎯 核心架构
-
-```
-前端 → API服务器(8000端口) → ComfyUI服务器(101.201.50.90:8188)
-```
-
-## 🚀 简化API设计
-
-### 主要接口（只需要这一个）
-
-**POST** `/api/generate-poster`
-
-**请求：**
-```json
-{
- "user_input": "端午节海报，传统风格，包含荷花和龙舟"
-}
-```
-
-**响应：**
-```json
-{
- "status": "success",
- "message": "海报生成完成",
- "data": {
- "vue_code": "完整的Vue组件代码",
- "suggestions": {
- "layer6_title_content": {"content": "端午节安康", "font_name": "SimHei"},
- "layer7_subtitle_content": {"content": "粽叶飘香，龙舟竞渡"}
- },
- "psd_file_path": "生成的PSD文件路径",
- "file_size_mb": 5.93,
- "generated_images": 2
- },
- "session_id": "会话ID用于下载文件"
-}
-```
-
-### 文件下载接口
-
-**GET** `/api/download/{file_type}?session_id={session_id}`
-
-- `file_type`: `vue` | `psd` | `json`
-
-## 🔧 本地开发部署
-
-### 1. 启动API服务器
-```bash
-cd E:\砚生\ai_service\scripts
-python run_pipeline.py
-# 选择: 2 (API服务器模式)
-```
-
-服务器启动在: `http://localhost:8000`
-
-### 2. 前端调用示例
-
-```javascript
-// 一键生成海报
-async function generatePoster(userInput) {
- try {
- const response = await fetch('http://localhost:8000/api/generate-poster', {
- method: 'POST',
- headers: {
- 'Content-Type': 'application/json',
- },
- body: JSON.stringify({
- user_input: userInput
- })
- });
-
- const result = await response.json();
-
- if (result.status === 'success') {
- console.log('Vue代码:', result.data.vue_code);
- console.log('PSD文件大小:', result.data.file_size_mb, 'MB');
-
- // 下载Vue文件
- downloadFile(result.session_id, 'vue');
- // 下载PSD文件
- downloadFile(result.session_id, 'psd');
-
- return result;
- }
- } catch (error) {
- console.error('生成失败:', error);
- }
-}
-
-// 下载文件
-function downloadFile(sessionId, fileType) {
- const url = `http://localhost:8000/api/download/${fileType}?session_id=${sessionId}`;
- const a = document.createElement('a');
- a.href = url;
- a.download = '';
- document.body.appendChild(a);
- a.click();
- document.body.removeChild(a);
-}
-
-// 使用示例
-generatePoster("春节海报，红色背景，包含灯笼");
-```
-
-## 🌐 生产环境部署
-
-### 方案1: 单服务器部署（推荐）
-
-```bash
-# 1. 安装依赖
-pip install -r requirements.txt
-
-# 2. 配置环境变量
-cp .env.example .env
-# 编辑.env文件，设置API密钥
-
-# 3. 启动服务
-python run_pipeline.py
-# 选择: 2 (API服务器模式)
-
-# 4. 使用PM2管理进程（可选）
-npm install -g pm2
-pm2 start "python run_pipeline.py --mode=api" --name="poster-api"
-```
-
-### 方案2: Docker部署
-
-```dockerfile
-# Dockerfile
-FROM python:3.11
-
-WORKDIR /app
-COPY requirements.txt .
-RUN pip install -r requirements.txt
-
-COPY . .
-EXPOSE 8000
-
-CMD ["python", "scripts/run_pipeline.py", "--mode=api"]
-```
-
-```bash
-# 构建和运行
-docker build -t poster-api .
-docker run -p 8000:8000 poster-api
-```
-
-### 方案3: 云服务器部署
-
-```bash
-# 在云服务器上
-git clone <你的仓库>
-cd ai_service
-pip install -r requirements.txt
-
-# 配置防火墙开放8000端口
-sudo ufw allow 8000
-
-# 启动服务
-python scripts/run_pipeline.py
-```
-
-## 🔒 生产环境配置
-
-### 1. 修改CORS设置
-```python
-# 在run_pipeline.py中修改
-app.add_middleware(
- CORSMiddleware,
- allow_origins=["https://your-frontend-domain.com"], # 改为你的前端域名
- allow_credentials=True,
- allow_methods=["GET", "POST"],
- allow_headers=["*"],
-)
-```
-
-### 2. 添加API认证（可选）
-```python
-from fastapi.security import HTTPBearer
-
-security = HTTPBearer()
-
-@app.post("/api/generate-poster")
-async def generate_poster_api(request: PosterRequest, token: str = Depends(security)):
- # 验证token
- if not verify_token(token.credentials):
- raise HTTPException(status_code=401, detail="Invalid token")
- # ...existing code...
-```
-
-### 3. 配置HTTPS（Nginx反向代理）
-```nginx
-server {
- listen 80;
- server_name your-domain.com;
-
- location / {
- proxy_pass http://localhost:8000;
- proxy_set_header Host $host;
- proxy_set_header X-Real-IP $remote_addr;
- }
-}
-```
-
-## 📊 性能监控
-
-### 健康检查
-```bash
-curl http://localhost:8000/health
-```
-
-### 状态监控
-```javascript
-// 检查生成状态
-async function checkStatus(sessionId) {
- const response = await fetch(`http://localhost:8000/api/status/${sessionId}`);
- return await response.json();
-}
-```
-
-## 🎯 为什么使用单一API？
-
-1. **简化前端调用** - 一次请求获得所有结果
-2. **减少网络延迟** - 避免多次HTTP请求
-3. **统一错误处理** - 所有错误在一个地方处理
-4. **会话管理** - 自动管理文件和状态
-5. **部署简单** - 只需要一个服务端点
-
-## 🔄 工作流程
-
-1. **前端发送请求** → `POST /api/generate-poster`
-2. **API自动执行**：
- - 调用DeepSeek分析用户输入
- - 调用ComfyUI(101.201.50.90:8188)生成图片
- - 调用Kimi生成文案
- - 生成Vue代码
- - 创建PSD文件（使用手动模板）
-3. **返回完整结果** → 前端获得所有内容
-4. **前端按需下载** → 使用session_id下载文件
-
-这样设计的好处是前端只需要调用一个接口，等待完成后就能获得Vue代码和下载PSD文件，非常简单高效！
-
-
-
-
-
-
-
-
-
- {theme}
-精彩活动，敬请参与
-
-
-
-
-
-
-
-
-