diff --git a/api/models/poster.py b/api/models/poster.py index f8c48c9..be9451c 100644 --- a/api/models/poster.py +++ b/api/models/poster.py @@ -20,6 +20,8 @@ class PosterGenerateRequest(BaseModel): scenicSpotId: Optional[int] = Field(None, description="景区ID，用于AI生成内容") numVariations: int = Field(3, description="要生成的海报变体数量, 默认为3", ge=1, le=5) forceLlmGeneration: bool = Field(False, description="是否强制使用LLM重新生成内容") + generatePsd: bool = Field(False, description="是否生成PSD分层文件") + psdOutputPath: Optional[str] = Field(None, description="PSD文件输出路径（可选，默认自动生成）") class Config: json_schema_extra = { @@ -28,6 +30,8 @@ class PosterGenerateRequest(BaseModel): "imagesBase64": "", "numVariations": 1, "forceLlmGeneration":False, + "generatePsd": True, + "psdOutputPath": "custom_poster.psd", "contentId":1, "productId":1, "scenicSpotId":1, @@ -65,6 +69,7 @@ class PosterGenerateResponse(BaseModel): requestId: str templateId: str resultImagesBase64: List[Dict[str, Any]] = Field(description="生成的海报图像(base64编码)列表") + psdFiles: Optional[List[Dict[str, Any]]] = Field(None, description="生成的PSD文件信息列表") metadata: Dict[str, Any] = Field(default_factory=dict) class ImageUsageRequest(BaseModel): diff --git a/api/routers/poster.py b/api/routers/poster.py index 6e650c3..4e2c83b 100644 --- a/api/routers/poster.py +++ b/api/routers/poster.py @@ -87,11 +87,12 @@ async def generate_poster( - **content_id**: 内容ID（可选） - **product_id**: 产品ID（可选） - **scenic_spot_id**: 景区ID（可选） - - **image_ids**: 图像ID列表（可选） + - **images_base64**: 图像base64编码（可选） - **template_id**: 模板ID（默认为vibrant） - - **generate_collage**: 是否生成拼图 - **poster_content**: 用户提供的海报内容（可选） - **force_llm_generation**: 是否强制使用LLM生成内容（可选） + - **generate_psd**: 是否生成PSD分层文件（可选） + - **psd_output_path**: PSD文件输出路径（可选） """ try: result = await poster_service.generate_poster( @@ -102,7 +103,9 @@ async def generate_poster( scenic_spot_id=request.scenicSpotId, images_base64=request.imagesBase64, num_variations=request.numVariations, - force_llm_generation=request.forceLlmGeneration + force_llm_generation=request.forceLlmGeneration, + generate_psd=request.generatePsd, + psd_output_path=request.psdOutputPath ) return PosterGenerateResponse(**result) diff --git a/api/services/poster.py b/api/services/poster.py index 473a366..afb709f 100644 --- a/api/services/poster.py +++ b/api/services/poster.py @@ -161,7 +161,9 @@ class PosterService: scenic_spot_id: Optional[int], images_base64: Optional[List[str]] , num_variations: int = 1, - force_llm_generation: bool = False) -> Dict[str, Any]: + force_llm_generation: bool = False, + generate_psd: bool = False, + psd_output_path: Optional[str] = None) -> Dict[str, Any]: """ 统一的海报生成入口 @@ -171,12 +173,14 @@ class PosterService: content_id: 内容ID，用于从数据库获取内容（可选） product_id: 产品ID，用于从数据库获取产品信息（可选） scenic_spot_id: 景点ID，用于从数据库获取景点信息（可选） - image_ids: 图片ID列表，用于从数据库获取图片（可选） + images_base64: 图片base64编码，用于生成海报（可选） num_variations: 需要生成的变体数量 force_llm_generation: 是否强制使用LLM生成内容 + generate_psd: 是否生成PSD分层文件 + psd_output_path: PSD文件输出路径（可选，默认自动生成） Returns: - 生成结果字典 + 生成结果字典，包含PNG图像和可选的PSD文件 """ start_time = time.time() @@ -230,6 +234,7 @@ class PosterService: # 5. 保存海报并返回结果 variations = [] + psd_files = [] i=0 ## 用于多个海报时，指定海报的编号，此时只有一个没有用上，但是接口开放着。 output_path = self._save_poster(posters, template_id, i) if output_path: @@ -238,6 +243,15 @@ class PosterService: "poster_path": str(output_path), "base64": self._image_to_base64(posters) }) + + # 6. 如果需要，生成PSD分层文件 + if generate_psd: + psd_result = self._generate_psd_file( + template_handler, images, final_content, + template_id, i, psd_output_path + ) + if psd_result: + psd_files.append(psd_result) # 记录模板使用情况 self._update_template_stats(template_id, bool(variations), time.time() - start_time) @@ -246,10 +260,12 @@ class PosterService: "requestId": f"poster-{datetime.now().strftime('%Y%m%d-%H%M%S')}-{str(uuid.uuid4())[:8]}", "templateId": template_id, "resultImagesBase64": variations, + "psdFiles": psd_files if psd_files else None, "metadata": { "generation_time": f"{time.time() - start_time:.2f}s", "model_used": self.ai_agent.config.model if force_llm_generation or not poster_content else None, - "num_variations": len(variations) + "num_variations": len(variations), + "psd_generated": bool(psd_files) } } except Exception as e: @@ -336,4 +352,104 @@ class PosterService: return None except Exception as e: logger.error(f"生成内容时发生错误: {e}", exc_info=True) + return None + + def _generate_psd_file(self, template_handler: BaseTemplate, images: Image.Image, + content: Dict[str, Any], template_id: str, + variation_id: int, custom_output_path: Optional[str] = None) -> Optional[Dict[str, Any]]: + """ + 生成PSD分层文件 + + Args: + template_handler: 模板处理器实例 + images: 图像数据 + content: 海报内容 + template_id: 模板ID + variation_id: 变体ID + custom_output_path: 自定义输出路径 + + Returns: + PSD文件信息字典，包含文件路径、base64编码等 + """ + try: + # 检查模板是否支持PSD生成 + if not hasattr(template_handler, 'generate_layered_psd'): + logger.warning(f"模板 {template_id} 不支持PSD分层输出") + return None + + # 生成PSD文件路径 + if custom_output_path: + psd_filename = custom_output_path + if not psd_filename.endswith('.psd'): + psd_filename += '.psd' + else: + timestamp = datetime.now().strftime('%Y%m%d_%H%M%S') + psd_filename = f"{template_id}_layered_v{variation_id}_{timestamp}.psd" + + # 获取输出目录 + topic_id = f"poster_{template_id}_{datetime.now().strftime('%Y%m%d_%H%M%S')}" + output_dir = self.output_manager.get_topic_dir(topic_id) + psd_path = output_dir / psd_filename + + # 调用模板的PSD生成方法 + logger.info(f"开始生成PSD分层文件: {psd_path}") + generated_psd_path = template_handler.generate_layered_psd( + images=images, + content=content, + output_path=str(psd_path) + ) + + if not generated_psd_path or not Path(generated_psd_path).exists(): + logger.error("PSD文件生成失败或文件不存在") + return None + + # 获取文件信息 + file_size = Path(generated_psd_path).stat().st_size + + # 可选：生成PSD的base64编码（注意：PSD文件通常较大） + psd_base64 = None + if file_size < 10 * 1024 * 1024: # 如果文件小于10MB，才生成base64 + try: + with open(generated_psd_path, 'rb') as f: + psd_base64 = base64.b64encode(f.read()).decode('utf-8') + except Exception as e: + logger.warning(f"生成PSD文件base64编码失败: {e}") + + # 生成预览图（从PSD合成PNG预览） + preview_base64 = None + try: + from psd_tools import PSDImage + psd = PSDImage.open(generated_psd_path) + preview_image = psd.composite() + if preview_image: + preview_base64 = self._image_to_base64(preview_image) + logger.info("PSD预览图生成成功") + except Exception as e: + logger.warning(f"生成PSD预览图失败: {e}") + + logger.info(f"PSD文件生成成功: {generated_psd_path} ({file_size/1024:.1f}KB)") + + return { + "variation_id": variation_id, + "psd_path": str(generated_psd_path), + "file_name": psd_filename, + "file_size": file_size, + "base64": psd_base64, + "preview_base64": preview_base64, + "layer_count": self._get_psd_layer_count(generated_psd_path), + "generation_time": datetime.now().isoformat() + } + + except Exception as e: + logger.error(f"生成PSD文件时发生错误: {e}", exc_info=True) + return None + + def _get_psd_layer_count(self, psd_path: str) -> Optional[int]: + """获取PSD文件的图层数量""" + try: + from psd_tools import PSDImage + psd = PSDImage.open(psd_path) + return len(list(psd)) + except Exception as e: + logger.warning(f"获取PSD图层数量失败: {e}") return None \ No newline at end of file