Galileo AI：智能UI设计生成

import numpy as npfrom dataclasses import dataclass, fieldfrom typing import List, Dict, Optional @dataclassclass UIComponent:    """UI组件: 对应设计系统中的可复用组件"""    component_type: str    properties: Dict = field(default_factory=dict)    children: List = field(default_factory=list) @dataclassclass DesignIntent:    """设计意图: 从自然语言解析出的结构化需求"""    page_type: str = ""    elements: List[str] = field(default_factory=list)    style: str = "default"    interactions: List[str] = field(default_factory=list)  class IntentParser:    """自然语言意图解析器: 模拟NLU提取设计需求"""     PAGE_KEYWORDS = {        "登录": "login", "注册": "register", "购物车": "cart",        "列表": "list", "详情": "detail", "设置": "settings"    }     ELEMENT_KEYWORDS = {        "输入框": "input", "按钮": "button", "卡片": "card",        "图片": "image", "复选框": "checkbox", "下拉": "dropdown",        "验证码": "verification_input", "手机号": "phone_input"    }     STYLE_KEYWORDS = {"简约": "minimal", "商务": "business", "活泼": "vibrant"}     def parse(self, description: str) -> DesignIntent:        intent = DesignIntent()        for keyword, page_type in self.PAGE_KEYWORDS.items():            if keyword in description:                intent.page_type = page_type                break        for keyword, elem_type in self.ELEMENT_KEYWORDS.items():            if keyword in description:                intent.elements.append(elem_type)        for keyword, style in self.STYLE_KEYWORDS.items():            if keyword in description:                intent.style = style                break        if "微信" in description:            intent.elements.append("wechat_button")        if "刷新" in description:            intent.interactions.append("pull_to_refresh")        return intent  class ComponentMatcher:    """组件匹配器: 将意图元素映射到设计系统组件"""     COMPONENT_MAP = {        "input": "TextInput", "button": "Button", "card": "Card",        "image": "Image", "checkbox": "Checkbox", "dropdown": "Dropdown",        "verification_input": "VerificationInput", "phone_input": "PhoneInput",        "wechat_button": "SocialButton"    }     def match(self, elements: List[str]) -> List[UIComponent]:        components = []        for elem in elements:            comp_type = self.COMPONENT_MAP.get(elem, "Container")            component = UIComponent(component_type=comp_type)            if comp_type == "Button":                component.properties = {"variant": "primary", "size": "large"}            elif comp_type == "TextInput":                component.properties = {"placeholder": "", "validation": "none"}            components.append(component)        return components  class LayoutGenerator:    """布局生成器: 根据组件和页面类型生成布局结构"""     LAYOUT_TEMPLATES = {        "login": {"layout": "vertical_center", "spacing": 16, "padding": 24},        "register": {"layout": "vertical_center", "spacing": 16, "padding": 24},        "cart": {"layout": "vertical_scroll", "spacing": 12, "padding": 16},        "list": {"layout": "vertical_scroll", "spacing": 8, "padding": 16},        "detail": {"layout": "vertical_scroll", "spacing": 16, "padding": 0},    }     def generate(self, page_type: str, components: List[UIComponent]) -> Dict:        template = self.LAYOUT_TEMPLATES.get(page_type, {"layout": "vertical", "spacing": 12, "padding": 16})        return {            "page_type": page_type,            "layout": template["layout"],            "spacing": template["spacing"],            "padding": template["padding"],            "components": [{"type": c.component_type, "props": c.properties} for c in components],            "component_count": len(components)        }  class DesignSystemEnforcer:    """设计系统约束执行器: 确保生成结果符合规范"""     def __init__(self):        self.color_tokens = {"primary": "#1890FF", "success": "#52C41A", "danger": "#FF4D4F"}        self.font_sizes = {"heading": 24, "body": 14, "caption": 12}        self.radius = {"small": 4, "medium": 8, "large": 12}     def apply_constraints(self, layout: Dict) -> Dict:        for comp in layout.get("components", []):            if comp["type"] == "Button":                comp["props"]["color"] = self.color_tokens["primary"]                comp["props"]["font_size"] = self.font_sizes["body"]                comp["props"]["border_radius"] = self.radius["medium"]            elif "Input" in comp["type"]:                comp["props"]["border_radius"] = self.radius["small"]                comp["props"]["border_color"] = "#D9D9D9"        layout["design_system_applied"] = True        layout["color_tokens"] = self.color_tokens        return layout  class GalileoAIGenerator:    """Galileo AI生成器: 整合意图解析、组件匹配、布局生成和规范约束"""     def __init__(self):        self.parser = IntentParser()        self.matcher = ComponentMatcher()        self.layout_gen = LayoutGenerator()        self.enforcer = DesignSystemEnforcer()     def generate(self, description: str, num_variants: int = 3) -> List[Dict]:        intent = self.parser.parse(description)        if not intent.elements:            return [{"error": "无法识别UI元素,请更详细描述"}]        variants = []        for i in range(num_variants):            components = self.matcher.match(intent.elements)            layout = self.layout_gen.generate(intent.page_type, components)            layout = self.enforcer.apply_constraints(layout)            layout["variant_id"] = i + 1            layout["style"] = intent.style            variants.append(layout)        return variants  generator = GalileoAIGenerator()description = "设计一个用户登录页面包含手机号验证码和微信登录"variants = generator.generate(description, num_variants=3) print(f"输入: {description}")print(f"生成方案数: {len(variants)}")for v in variants:    print(f"  方案{v['variant_id']}: {v['layout']}, {v['component_count']}个组件")    for comp in v["components"]:        print(f"    - {comp['type']}: {comp['props']}")    print(f"    设计系统已应用: {v['design_system_applied']}")