From a6624ca44907d161f4e2cfdcf544f99acb7b6052 Mon Sep 17 00:00:00 2001 From: 13315423919 <13315423919@qq.com> Date: Fri, 7 Nov 2025 09:05:20 +0800 Subject: [PATCH] Add File --- .../services/models/slide_image_info.py | 263 ++++++++++++++++++ 1 file changed, 263 insertions(+) create mode 100644 src/landppt/services/models/slide_image_info.py diff --git a/src/landppt/services/models/slide_image_info.py b/src/landppt/services/models/slide_image_info.py new file mode 100644 index 0000000..b007a9c --- /dev/null +++ b/src/landppt/services/models/slide_image_info.py @@ -0,0 +1,263 @@ +""" +PPT幻灯片图片信息数据模型 +支持多图片信息的数据结构，包含图片来源、用途、内容描述、绝对地址等信息 +""" + +from typing import List, Dict, Any, Optional +from dataclasses import dataclass +from enum import Enum + + +class ImageSource(Enum): + """图片来源枚举""" + LOCAL = "local" # 本地图床 + NETWORK = "network" # 网络搜索 + AI_GENERATED = "ai_generated" # AI生成 + + +class ImagePurpose(Enum): + """图片用途枚举""" + DECORATION = "decoration" # 装饰性图片 + ILLUSTRATION = "illustration" # 说明性图片 + BACKGROUND = "background" # 背景图片 + ICON = "icon" # 图标 + CHART_SUPPORT = "chart_support" # 图表辅助 + CONTENT_VISUAL = "content_visual" # 内容可视化 + + +@dataclass +class SlideImageInfo: + """单个图片信息""" + image_id: str # 图片ID + absolute_url: str # 绝对地址URL + source: ImageSource # 图片来源 + purpose: ImagePurpose # 图片用途 + content_description: str # 内容描述 + search_keywords: Optional[str] = None # 搜索关键词（用于本地/网络搜索） + generation_prompt: Optional[str] = None # 生成提示词（用于AI生成） + width: Optional[int] = None # 图片宽度 + height: Optional[int] = None # 图片高度 + file_size: Optional[int] = None # 文件大小 + format: Optional[str] = None # 图片格式 + alt_text: str = "" # 替代文本 + title: str = "" # 图片标题 + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "image_id": self.image_id, + "absolute_url": self.absolute_url, + "source": self.source.value, + "purpose": self.purpose.value, + "content_description": self.content_description, + "search_keywords": self.search_keywords, + "generation_prompt": self.generation_prompt, + "width": self.width, + "height": self.height, + "file_size": self.file_size, + "format": self.format, + "alt_text": self.alt_text, + "title": self.title + } + + @classmethod + def from_dict(cls, data: Dict[str, Any]) -> 'SlideImageInfo': + """从字典创建实例""" + return cls( + image_id=data["image_id"], + absolute_url=data["absolute_url"], + source=ImageSource(data["source"]), + purpose=ImagePurpose(data["purpose"]), + content_description=data["content_description"], + search_keywords=data.get("search_keywords"), + generation_prompt=data.get("generation_prompt"), + width=data.get("width"), + height=data.get("height"), + file_size=data.get("file_size"), + format=data.get("format"), + alt_text=data.get("alt_text", ""), + title=data.get("title", "") + ) + + +@dataclass +class SlideImagesCollection: + """幻灯片图片集合""" + page_number: int # 页码 + images: List[SlideImageInfo] # 图片列表 + total_count: int = 0 # 总图片数量 + local_count: int = 0 # 本地图片数量 + network_count: int = 0 # 网络图片数量 + ai_generated_count: int = 0 # AI生成图片数量 + + def __post_init__(self): + """初始化后处理，计算各类型图片数量""" + self.total_count = len(self.images) + self.local_count = sum(1 for img in self.images if img.source == ImageSource.LOCAL) + self.network_count = sum(1 for img in self.images if img.source == ImageSource.NETWORK) + self.ai_generated_count = sum(1 for img in self.images if img.source == ImageSource.AI_GENERATED) + + def add_image(self, image_info: SlideImageInfo): + """添加图片""" + self.images.append(image_info) + self.__post_init__() # 重新计算数量 + + def get_images_by_source(self, source: ImageSource) -> List[SlideImageInfo]: + """根据来源获取图片""" + return [img for img in self.images if img.source == source] + + def get_images_by_purpose(self, purpose: ImagePurpose) -> List[SlideImageInfo]: + """根据用途获取图片""" + return [img for img in self.images if img.purpose == purpose] + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "page_number": self.page_number, + "total_count": self.total_count, + "local_count": self.local_count, + "network_count": self.network_count, + "ai_generated_count": self.ai_generated_count, + "images": [img.to_dict() for img in self.images] + } + + @classmethod + def from_dict(cls, data: Dict[str, Any]) -> 'SlideImagesCollection': + """从字典创建实例""" + images = [SlideImageInfo.from_dict(img_data) for img_data in data.get("images", [])] + return cls( + page_number=data["page_number"], + images=images + ) + + def get_summary_for_ai(self) -> str: + """获取用于AI参考的图片摘要信息""" + if not self.images: + return "本页面无图片" + + summary_parts = [ + f"本页面共有{self.total_count}张图片：" + ] + + def _format_image_with_metadata(img: SlideImageInfo) -> str: + """格式化图片信息，包含元数据""" + # 基本描述 + desc = f"{img.content_description}({img.purpose.value})" + + # 添加尺寸信息 + if img.width and img.height: + desc += f" [尺寸: {img.width}x{img.height}px]" + + # 添加文件大小信息 + if img.file_size: + if img.file_size < 1024: + size_str = f"{img.file_size}B" + elif img.file_size < 1024 * 1024: + size_str = f"{img.file_size / 1024:.1f}KB" + else: + size_str = f"{img.file_size / (1024 * 1024):.1f}MB" + desc += f" [大小: {size_str}]" + + # 添加格式信息 + if img.format: + desc += f" [格式: {img.format.upper()}]" + + return desc + + if self.local_count > 0: + local_images = self.get_images_by_source(ImageSource.LOCAL) + local_desc = ", ".join([_format_image_with_metadata(img) for img in local_images]) + summary_parts.append(f"- 本地图片{self.local_count}张: {local_desc}") + + if self.network_count > 0: + network_images = self.get_images_by_source(ImageSource.NETWORK) + network_desc = ", ".join([_format_image_with_metadata(img) for img in network_images]) + summary_parts.append(f"- 网络图片{self.network_count}张: {network_desc}") + + if self.ai_generated_count > 0: + ai_images = self.get_images_by_source(ImageSource.AI_GENERATED) + ai_desc = ", ".join([_format_image_with_metadata(img) for img in ai_images]) + summary_parts.append(f"- AI生成图片{self.ai_generated_count}张: {ai_desc}") + + # 添加详细的图片信息，包含元数据 + summary_parts.append("\n详细图片信息：") + for i, img in enumerate(self.images, 1): + # 构建详细信息 + details = [] + details.append(f"地址: {img.absolute_url}") + details.append(f"描述: {img.content_description}") + details.append(f"用途: {img.purpose.value}") + + if img.width and img.height: + details.append(f"尺寸: {img.width}x{img.height}px") + + if img.file_size: + if img.file_size < 1024: + size_str = f"{img.file_size}B" + elif img.file_size < 1024 * 1024: + size_str = f"{img.file_size / 1024:.1f}KB" + else: + size_str = f"{img.file_size / (1024 * 1024):.1f}MB" + details.append(f"大小: {size_str}") + + if img.format: + details.append(f"格式: {img.format.upper()}") + + if img.title: + details.append(f"标题: {img.title}") + + if img.alt_text: + details.append(f"替代文本: {img.alt_text}") + + summary_parts.append(f"{i}. {' | '.join(details)}") + + return "\n".join(summary_parts) + + +@dataclass +class ImageRequirement: + """图片需求信息""" + source: ImageSource # 图片来源 + count: int # 需要数量 + purpose: ImagePurpose # 图片用途 + description: str # 需求描述 + priority: int = 1 # 优先级 (1-5, 5最高) + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "source": self.source.value, + "count": self.count, + "purpose": self.purpose.value, + "description": self.description, + "priority": self.priority + } + + +@dataclass +class SlideImageRequirements: + """幻灯片图片需求集合""" + page_number: int # 页码 + requirements: List[ImageRequirement] # 需求列表 + total_images_needed: int = 0 # 总需求图片数量 + + def __post_init__(self): + """初始化后处理""" + self.total_images_needed = sum(req.count for req in self.requirements) + + def add_requirement(self, requirement: ImageRequirement): + """添加需求""" + self.requirements.append(requirement) + self.__post_init__() + + def get_requirements_by_source(self, source: ImageSource) -> List[ImageRequirement]: + """根据来源获取需求""" + return [req for req in self.requirements if req.source == source] + + def to_dict(self) -> Dict[str, Any]: + """转换为字典格式""" + return { + "page_number": self.page_number, + "total_images_needed": self.total_images_needed, + "requirements": [req.to_dict() for req in self.requirements] + }