Beautiful.ai：AI驱动PPT设计

# 模拟Beautiful.ai智能图表自动选择逻辑from typing import List, Dict, Any, Optionalfrom dataclasses import dataclass  @dataclassclass ChartData:    """图表数据结构"""    labels: List[str]  # 类别标签    series: Dict[str, List[float]]  # 数据系列：系列名->数值列表    data_type: str = ""  # 数据类型：categorical/time/part-to-whole/correlation  class SmartChartRecommender:    """智能图表推荐器（模拟Beautiful.ai的图表选择逻辑）"""        def __init__(self):        self.chart_types = {            "bar": {"name": "柱状图", "best_for": "类别比较"},            "line": {"name": "折线图", "best_for": "趋势变化"},            "pie": {"name": "饼图", "best_for": "占比关系"},            "donut": {"name": "环形图", "best_for": "占比关系（更现代）"},            "scatter": {"name": "散点图", "best_for": "相关性分析"},            "area": {"name": "面积图", "best_for": "累计趋势"},            "table": {"name": "表格", "best_for": "精确数值"},        }        def analyze_data_type(self, data: ChartData) -> str:        """分析数据类型                判断数据更适合哪种可视化方式        """        num_categories = len(data.labels)        num_series = len(data.series)                # 检查是否是时间序列（简化判断：标签是否包含年份/月份/日期特征）        time_keywords = ["年", "月", "日", "季", "Q", "202", "Jan", "Feb", "Mar"]        has_time = any(any(kw in str(l) for kw in time_keywords) for l in data.labels)                # 检查是否是占比数据（单系列且数值总和接近100或1）        if num_series == 1:            values = list(data.series.values())[0]            total = sum(abs(v) for v in values)            if abs(total - 100) < 5 or abs(total - 1.0) < 0.05:                return "part-to-whole"                if has_time and num_categories >= 3:            return "time-series"                if num_series == 2 and num_categories > 10:            return "correlation"                return "categorical"        def recommend_chart(self, data: ChartData) -> List[Dict]:        """推荐合适的图表类型（按推荐程度排序）                Returns:            推荐的图表类型列表，包含推荐理由        """        data_type = self.analyze_data_type(data)        num_categories = len(data.labels)        num_series = len(data.series)                recommendations = []                if data_type == "time-series":            # 时间序列数据            recommendations.append({                "type": "line",                "confidence": 0.9,                "reason": "时间序列数据，折线图最适合展示趋势变化"            })            if num_series <= 5:                recommendations.append({                    "type": "area",                    "confidence": 0.7,                    "reason": "面积图也适合趋势展示，视觉上更有分量感"                })                        elif data_type == "part-to-whole":            # 占比数据            if num_categories <= 6:                recommendations.append({                    "type": "donut",                    "confidence": 0.85,                    "reason": "类别较少的占比数据，环形图现代且清晰"                })                recommendations.append({                    "type": "pie",                    "confidence": 0.7,                    "reason": "饼图是经典的占比可视化方式"                })            else:                recommendations.append({                    "type": "bar",                    "confidence": 0.8,                    "reason": "类别较多时，柱状图比饼图更清晰易读"                })                        elif data_type == "correlation":            # 相关性数据            recommendations.append({                "type": "scatter",                "confidence": 0.85,                "reason": "两个变量的关系，散点图最适合展示相关性"            })                    else:            # 类别比较数据（默认）            recommendations.append({                "type": "bar",                "confidence": 0.9,                "reason": "类别比较是柱状图的经典应用场景，直观易读"            })            if num_categories <= 5 and num_series == 1:                recommendations.append({                    "type": "donut",                    "confidence": 0.6,                    "reason": "类别少时也可以用环形图增加视觉变化"                })                # 表格永远作为备选        recommendations.append({            "type": "table",            "confidence": 0.5,            "reason": "需要精确数值时使用表格"        })                # 按置信度排序        recommendations.sort(key=lambda x: x["confidence"], reverse=True)        return recommendations  # 演示def demo():    print("=== Beautiful.ai 智能图表推荐演示 ===\n")        recommender = SmartChartRecommender()        # 示例1：月度销售数据（时间序列）    print("示例1：月度销售数据")    sales_data = ChartData(        labels=["1月", "2月", "3月", "4月", "5月", "6月"],        series={            "产品A": [120, 150, 180, 165, 200, 230],            "产品B": [80, 95, 110, 100, 125, 140]        }    )    print(f"  数据类型: {recommender.analyze_data_type(sales_data)}")    recs = recommender.recommend_chart(sales_data)    for i, r in enumerate(recs[:3], 1):        print(f"  推荐{i}: {r['type']} (置信度: {r['confidence']}) - {r['reason']}")        print("\n=== 演示完成 ===")  if __name__ == "__main__":    demo()

# -*- coding: utf-8 -*-"""Beautiful.ai 简化实现演示模拟核心机制：智能模板自适应、布局优化引擎、DesignBot、智能图表推荐"""import refrom typing import Dict, List, Any, Optional, Tuplefrom dataclasses import dataclass, fieldfrom abc import ABC, abstractmethod  # ===================== 设计规则系统 ===================== @dataclassclass DesignRules:    """设计规则库        模拟Beautiful.ai的设计规则系统    """    # 网格系统    grid_columns: int = 12    grid_gutter: int = 20  # 列间距（像素）    page_padding: int = 40  # 页面边距        # 字体层级    title_size: int = 36    subtitle_size: int = 24    body_size: int = 16    caption_size: int = 12        # 间距系统（基于8点网格）    spacing_xs: int = 8    spacing_sm: int = 16    spacing_md: int = 24    spacing_lg: int = 40    spacing_xl: int = 64        # 配色方案    primary_color: str = "#2563eb"    secondary_color: str = "#64748b"    accent_color: str = "#f59e0b"    text_color: str = "#1e293b"    background_color: str = "#ffffff"        # 对齐方式    default_alignment: str = "left"  # left/center/right  # ===================== 页面元素 ===================== @dataclassclass Element:    """页面元素基类"""    element_id: str    element_type: str  # title/text/image/chart/list    content: Any    x: float = 0    y: float = 0    width: float = 0    height: float = 0    priority: int = 1  # 布局优先级  @dataclassclass SlideLayout:    """幻灯片布局"""    slide_id: str    layout_type: str  # title-only/title-content/two-column/chart-focused    elements: List[Element] = field(default_factory=list)    page_width: float = 960    page_height: float = 540  # ===================== 布局优化引擎 ===================== class LayoutEngine:    """布局优化引擎（简化版）        模拟Beautiful.ai的布局自动优化能力    """        def __init__(self, rules: DesignRules = None):        self.rules = rules or DesignRules()        def compute_element_height(self, element: Element) -> float:        """根据内容计算元素高度（简化估算）"""        if element.element_type == "title":            return self.rules.title_size * 1.5        elif element.element_type == "subtitle":            return self.rules.subtitle_size * 1.5        elif element.element_type == "text":            # 估算：根据字数和宽度计算行数            text = str(element.content)            chars_per_line = max(20, element.width // (self.rules.body_size // 2)) if element.width > 0 else 50            lines = max(1, len(text) // chars_per_line + 1)            return lines * self.rules.body_size * 1.5        elif element.element_type == "chart":            return 250  # 默认图表高度        elif element.element_type == "image":            return 200  # 默认图片高度        elif element.element_type == "list":            items = element.content if isinstance(element.content, list) else [element.content]            return len(items) * self.rules.body_size * 1.8        else:            return 100        def optimize_layout(self, slide: SlideLayout) -> SlideLayout:        """优化页面布局                根据设计规则和内容，计算每个元素的最优位置和大小        """        rules = self.rules        padding = rules.page_padding        available_width = slide.page_width - 2 * padding                # 第一步：计算每个元素的宽度（基于布局类型）        if slide.layout_type == "title-only":            self._layout_title_only(slide, padding, available_width)        elif slide.layout_type == "title-content":            self._layout_title_content(slide, padding, available_width)        elif slide.layout_type == "two-column":            self._layout_two_column(slide, padding, available_width)        elif slide.layout_type == "chart-focused":            self._layout_chart_focused(slide, padding, available_width)        else:            self._layout_title_content(slide, padding, available_width)                return slide        def _layout_title_only(self, slide: SlideLayout, padding: float, avail_w: float):        """仅标题布局"""        current_y = padding + 100  # 垂直居中偏上                for elem in slide.elements:            elem.width = avail_w            elem.x = padding            elem.y = current_y            elem.height = self.compute_element_height(elem)            current_y += elem.height + self.rules.spacing_md        def _layout_title_content(self, slide: SlideLayout, padding: float, avail_w: float):        """标题+内容布局"""        current_y = padding                for elem in slide.elements:            elem.width = avail_w            elem.x = padding            elem.y = current_y                        if elem.element_type in ("title", "subtitle"):                elem.height = self.compute_element_height(elem)                current_y += elem.height + self.rules.spacing_sm            else:                # 内容元素占据剩余空间                remaining = slide.page_height - current_y - padding                elem.height = max(100, remaining * 0.85)                current_y += elem.height        def _layout_two_column(self, slide: SlideLayout, padding: float, avail_w: float):        """双栏布局"""        col_width = (avail_w - self.rules.spacing_md) / 2        current_y = padding                left_elems = [e for e in slide.elements if e.element_type in ("title", "subtitle", "text", "list")]        right_elems = [e for e in slide.elements if e.element_type in ("chart", "image")]                # 标题（通栏）        for elem in slide.elements:            if elem.element_type == "title":                elem.width = avail_w                elem.x = padding                elem.y = current_y                elem.height = self.compute_element_height(elem)                current_y += elem.height + self.rules.spacing_md                # 左右分栏内容        content_top = current_y        content_height = slide.page_height - current_y - padding                left_x = padding        right_x = padding + col_width + self.rules.spacing_md                left_y = content_top        for elem in left_elems:            if elem.element_type == "title":                continue            elem.width = col_width            elem.x = left_x            elem.y = left_y            elem.height = self.compute_element_height(elem)            left_y += elem.height + self.rules.spacing_sm                right_y = content_top        for elem in right_elems:            elem.width = col_width            elem.x = right_x            elem.y = right_y            elem.height = content_height * 0.8            right_y += elem.height + self.rules.spacing_sm        def _layout_chart_focused(self, slide: SlideLayout, padding: float, avail_w: float):        """图表为主布局"""        current_y = padding                # 标题在上        title_elem = None        chart_elem = None        other_elems = []                for elem in slide.elements:            if elem.element_type == "title":                title_elem = elem            elif elem.element_type == "chart":                chart_elem = elem            else:                other_elems.append(elem)                if title_elem:            title_elem.width = avail_w            title_elem.x = padding            title_elem.y = current_y            title_elem.height = self.compute_element_height(title_elem)            current_y += title_elem.height + self.rules.spacing_md                # 图表占主要空间        if chart_elem:            chart_elem.width = avail_w            chart_elem.x = padding            chart_elem.y = current_y            chart_elem.height = slide.page_height - current_y - padding - 60            current_y += chart_elem.height + self.rules.spacing_sm        def validate_layout(self, slide: SlideLayout) -> Tuple[bool, List[str]]:        """验证布局是否符合设计规则                Returns:            (是否通过, 问题列表)        """        issues = []        rules = self.rules                for elem in slide.elements:            # 检查是否溢出页面            if elem.x + elem.width > slide.page_width - 10:                issues.append(f"元素{elem.element_id}宽度溢出页面")            if elem.y + elem.height > slide.page_height - 10:                issues.append(f"元素{elem.element_id}高度溢出页面")                        # 检查边距是否足够            if elem.x < rules.page_padding - 5:                issues.append(f"元素{elem.element_id}左边距不足")                return (len(issues) == 0, issues)  # ===================== 智能模板系统 ===================== class SmartTemplateSystem:    """智能模板系统（简化版）"""        def __init__(self, layout_engine: LayoutEngine):        self.layout_engine = layout_engine        self.templates = self._init_templates()        def _init_templates(self) -> Dict[str, Dict]:        """初始化模板库"""        return {            "title-slide": {                "name": "标题页",                "layout_type": "title-only",                "best_for": "演示文稿开场",                "elements": ["title", "subtitle"]            },            "content-slide": {                "name": "内容页",                "layout_type": "title-content",                "best_for": "文字内容为主",                "elements": ["title", "text"]            },            "chart-slide": {                "name": "图表页",                "layout_type": "chart-focused",                "best_for": "数据展示",                "elements": ["title", "chart"]            },            "two-column-slide": {                "name": "双栏页",                "layout_type": "two-column",                "best_for": "图文混排",                "elements": ["title", "text", "chart"]            }        }        def recommend_template(self, content_description: str, elements: List[str]) -> str:        """根据内容推荐合适的模板                Args:            content_description: 内容描述            elements: 元素类型列表                    Returns:            推荐的模板ID        """        # 简单规则：根据元素类型推荐        has_chart = "chart" in elements or "image" in elements        has_lots_of_text = elements.count("text") + elements.count("list") >= 2        has_only_title = len(elements) == 1 and elements[0] == "title"                if has_only_title:            return "title-slide"        elif has_chart and not has_lots_of_text:            return "chart-slide"        elif has_chart and has_lots_of_text:            return "two-column-slide"        else:            return "content-slide"        def create_slide(self, template_id: str, elements: List[Element]) -> SlideLayout:        """基于模板创建幻灯片并自动优化布局                Args:            template_id: 模板ID            elements: 页面元素列表                    Returns:            布局优化后的幻灯片        """        template = self.templates.get(template_id)        if not template:            template = self.templates["content-slide"]                slide = SlideLayout(            slide_id=f"slide-{template_id}",            layout_type=template["layout_type"],            elements=elements        )                # 应用布局优化        self.layout_engine.optimize_layout(slide)                return slide        def adapt_layout(self, slide: SlideLayout, new_elements: List[Element]) -> SlideLayout:        """内容变化时自适应调整布局                模拟添加/修改内容时自动重新布局        """        slide.elements = new_elements        self.layout_engine.optimize_layout(slide)                # 验证布局，如果有问题自动调整        valid, issues = self.layout_engine.validate_layout(slide)        if not valid:            # 简单处理：缩小字体或间距（简化处理）            for elem in slide.elements:                elem.height *= 0.9                return slide  # ===================== DesignBot AI助手 ===================== class DesignBot:    """DesignBot AI助手（简化版）        模拟Beautiful.ai的自然语言交互功能    """        def __init__(self, template_system: SmartTemplateSystem):        self.template_system = template_system        def parse_request(self, user_input: str) -> Dict[str, Any]:        """解析用户的自然语言请求                返回结构化的需求描述        """        request = {            "action": "add_slide",  # add_slide/modify/add_element            "slide_type": "content-slide",            "elements": [],            "topic": "",            "has_chart": False,            "has_image": False,            "is_title": False        }                # 简单的关键词匹配（实际是NLP模型）        input_lower = user_input.lower()                # 判断动作类型        if "加" in user_input or "添加" in user_input or "新建" in user_input:            request["action"] = "add_slide"        elif "改" in user_input or "修改" in user_input or "调整" in user_input:            request["action"] = "modify"                # 判断是否是标题页        if "封面" in user_input or "标题页" in user_input or "开场" in user_input:            request["is_title"] = True            request["elements"].append("title")            request["elements"].append("subtitle")                # 判断是否有图表        if "图表" in user_input or "数据" in user_input or "柱状图" in user_input or "折线图" in user_input:            request["has_chart"] = True            request["elements"].append("chart")                # 判断是否有图片        if "图片" in user_input or "图像" in user_input or "照片" in user_input:            request["has_image"] = True            request["elements"].append("image")                # 默认有标题和文字        if "title" not in request["elements"] and not request["is_title"]:            request["elements"].append("title")        if "text" not in request["elements"] and not request["has_chart"] and not request["is_title"]:            request["elements"].append("text")                # 提取主题（简化）        for keyword in ["关于", "介绍", "展示", "的"]:            if keyword in user_input:                idx = user_input.find(keyword)                topic = user_input[idx + len(keyword):].strip()                if len(topic) > 2:                    request["topic"] = topic[:30]                    break                return request        def generate_slide_from_request(self, user_input: str) -> SlideLayout:        """根据用户请求生成幻灯片                Args:            user_input: 用户的自然语言输入                    Returns:            生成的幻灯片        """        request = self.parse_request(user_input)                # 推荐模板        template_id = self.template_system.recommend_template(            request["topic"],            request["elements"]        )                # 创建元素        elements = []        for i, elem_type in enumerate(request["elements"]):            elem_id = f"elem-{i}-{elem_type}"            content = self._generate_sample_content(elem_type, request["topic"])            elements.append(Element(                element_id=elem_id,                element_type=elem_type,                content=content            ))                # 创建幻灯片        slide = self.template_system.create_slide(template_id, elements)                return slide        def _generate_sample_content(self, elem_type: str, topic: str) -> Any:        """生成示例内容（实际是大模型生成）"""        if elem_type == "title":            return f"{topic or '演示文稿标题'}"        elif elem_type == "subtitle":            return f"关于{topic or '主题'}的详细介绍"        elif elem_type == "text":            return f"这是关于{topic or '主题'}的详细内容介绍。这里会展示主要观点和关键信息，帮助观众快速理解核心内容。"        elif elem_type == "chart":            return {"type": "bar", "data": "示例数据"}        elif elem_type == "list":            return ["要点一", "要点二", "要点三"]        elif elem_type == "image":            return "[示例图片]"        else:            return "示例内容"  # ===================== 演示 ===================== def demo():    print("=" * 60)    print("  Beautiful.ai 简化实现演示")    print("=" * 60)        # 初始化组件    rules = DesignRules()    layout_engine = LayoutEngine(rules)    template_system = SmartTemplateSystem(layout_engine)    design_bot = DesignBot(template_system)        # 演示1：布局优化引擎    print("\n[演示1] 布局优化引擎 - 图表页自动布局")    elements = [        Element("t1", "title", "2024年Q1销售数据报告"),        Element("c1", "chart", {"type": "bar", "values": [120, 200, 150]})    ]    slide = template_system.create_slide("chart-slide", elements)    for elem in slide.elements:        print(f"  {elem.element_type}: ({elem.x:.0f}, {elem.y:.0f}) size={elem.width:.0f}x{elem.height:.0f}")    valid, issues = layout_engine.validate_layout(slide)    print(f"  布局验证: {'通过' if valid else '有问题'} {issues if issues else ''}")        print(f"\n[Demo] 演示完成！")  if __name__ == "__main__":    demo()