增加款式设计的部分标签

@@ -3,8 +3,9 @@
   <component name="NewModuleRootManager">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$">
     <content url="file://$MODULE_DIR$">
       <excludeFolder url="file://$MODULE_DIR$/.venv" />
       <excludeFolder url="file://$MODULE_DIR$/.venv" />
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
     </content>
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.11 (GoogleNanoBananaApp)" jdkType="Python SDK" />
+    <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
   </component>
   <component name="PyDocumentationSettings">
   <component name="PyDocumentationSettings">

@@ -10,7 +10,7 @@ from PySide6.QtWidgets import (
     QLabel, QLineEdit, QPushButton, QCheckBox, QTextEdit,
     QLabel, QLineEdit, QPushButton, QCheckBox, QTextEdit,
     QComboBox, QScrollArea, QGroupBox, QFileDialog, QMessageBox,
     QComboBox, QScrollArea, QGroupBox, QFileDialog, QMessageBox,
     QListWidget, QListWidgetItem, QTabWidget, QSplitter,
     QListWidget, QListWidgetItem, QTabWidget, QSplitter,
-    QMenu, QProgressBar
+    QMenu, QProgressBar, QInputDialog
 )
 )
 from PySide6.QtCore import Qt, QThread, Signal, QSize, QTimer, QMimeData
 from PySide6.QtCore import Qt, QThread, Signal, QSize, QTimer, QMimeData
 from PySide6.QtGui import QPixmap, QFont, QIcon, QDesktopServices, QAction, QImage, QDragEnterEvent, QDropEvent
 from PySide6.QtGui import QPixmap, QFont, QIcon, QDesktopServices, QAction, QImage, QDragEnterEvent, QDropEvent
@@ -25,6 +25,7 @@ import shutil
 import tempfile
 import tempfile
 import platform
 import platform
 import logging
 import logging
+import random
 from pathlib import Path
 from pathlib import Path
 from google import genai
 from google import genai
 from google.genai import types
 from google.genai import types
@@ -1264,6 +1265,16 @@ class ImageGeneratorWindow(QMainWindow):
         generation_tab = self.setup_generation_tab()
         generation_tab = self.setup_generation_tab()
         self.tab_widget.addTab(generation_tab, "图片生成")
         self.tab_widget.addTab(generation_tab, "图片生成")
 
 
+        # Create style designer tab
+        try:
+            jewelry_lib_manager = JewelryLibraryManager(self.get_config_dir())
+            style_designer_tab = StyleDesignerTab(jewelry_lib_manager, parent=self)
+            self.tab_widget.addTab(style_designer_tab, "款式设计")
+            self.logger.info("款式设计 Tab 创建成功")
+        except Exception as e:
+            self.logger.error(f"款式设计 Tab 创建失败: {e}")
+            # 即使失败也继续运行，不阻止应用启动
+
         # Create history tab
         # Create history tab
         history_tab = self.setup_history_tab()
         history_tab = self.setup_history_tab()
         self.tab_widget.addTab(history_tab, "历史记录")
         self.tab_widget.addTab(history_tab, "历史记录")
@@ -1286,7 +1297,7 @@ class ImageGeneratorWindow(QMainWindow):
 
 
         # Upload button and count
         # Upload button and count
         upload_header = QHBoxLayout()
         upload_header = QHBoxLayout()
-        upload_btn = QPushButton("+ 添加图片")
+        upload_btn = QPushButton("添加图片")
         upload_btn.clicked.connect(self.upload_images)
         upload_btn.clicked.connect(self.upload_images)
         upload_header.addWidget(upload_btn)
         upload_header.addWidget(upload_btn)
 
 
@@ -1357,7 +1368,7 @@ class ImageGeneratorWindow(QMainWindow):
         self.saved_prompts_combo = QComboBox()
         self.saved_prompts_combo = QComboBox()
         prompt_toolbar.addWidget(self.saved_prompts_combo)
         prompt_toolbar.addWidget(self.saved_prompts_combo)
 
 
-        delete_prompt_btn = QPushButton("🗑️ 删除")
+        delete_prompt_btn = QPushButton("删除")
         delete_prompt_btn.clicked.connect(self.delete_saved_prompt)
         delete_prompt_btn.clicked.connect(self.delete_saved_prompt)
         prompt_toolbar.addWidget(delete_prompt_btn)
         prompt_toolbar.addWidget(delete_prompt_btn)
         prompt_toolbar.addStretch()
         prompt_toolbar.addStretch()
@@ -2090,7 +2101,7 @@ class ImageGeneratorWindow(QMainWindow):
                 processing_method = "格式转换" if processed_bytes != self.generated_image_bytes else "原始数据"
                 processing_method = "格式转换" if processed_bytes != self.generated_image_bytes else "原始数据"
                 self.logger.info(f"图片下载完成 - 方法: {processing_method}, 文件: {file_path}, 大小: {file_size} 字节")
                 self.logger.info(f"图片下载完成 - 方法: {processing_method}, 文件: {file_path}, 大小: {file_size} 字节")
 
 
-                QMessageBox.information(self, "成功", f"图片已保存到:\n{file_path}\n\n文件大小: {file_size:,} 字节")
+                # 只使用状态栏提示，不显示弹窗
                 self.status_label.setText("● 图片已保存")
                 self.status_label.setText("● 图片已保存")
                 self.status_label.setStyleSheet("QLabel { color: #34C759; }")
                 self.status_label.setStyleSheet("QLabel { color: #34C759; }")
             except Exception as e:
             except Exception as e:
@@ -2527,6 +2538,874 @@ class ImageGenerationWorker(QThread):
             self.error.emit(error_msg)
             self.error.emit(error_msg)
 
 
 
 
+# ============================================================================
+# 珠宝设计功能模块
+# ============================================================================
+
+# 默认珠宝词库（纯中文）
+DEFAULT_JEWELRY_LIBRARY = {
+    "主石形状": [
+        "椭圆形",
+        "圆形",
+        "祖母绿形",
+        "梨形",
+        "垫形",
+        "公主方形",
+        "心形"
+    ],
+    "主石材质": [
+        "黑发晶",
+        "莫桑石",
+        "红宝石",
+        "蓝宝石",
+        "绿碧玺",
+        "黄水晶带天然包裹体",
+        "钻石",
+        "粉红蓝宝石"
+    ],
+    "金属": [
+        "14K黄金",
+        "18K玫瑰金",
+        "14K白金",
+        "铂金(PT950)",
+        "双色14K黄金+白金",
+        "950银镀铑",
+        "钛金属"
+    ],
+    "花头形式": [
+        "花卉风格光环",
+        "经典圆形光环",
+        "复古风格米粒边光环",
+        "双层光环",
+        "几何六边形光环",
+        "非对称光环",
+        "简约爪镶无光环"
+    ],
+    "戒臂结构": [
+        "扭转戒臂",
+        "分裂戒臂",
+        "刀刃戒臂",
+        "大教堂戒臂",
+        "交叉戒臂",
+        "直线平滑戒臂",
+        "三股编织戒臂"
+    ],
+    "戒臂处理": [
+        "雕刻镂空花丝工艺",
+        "密钉镶戒臂",
+        "抛光平滑戒臂",
+        "浮雕雕刻",
+        "编织纹理",
+        "极简干净戒臂",
+        "穿孔镂空细节",
+        "锤纹处理"
+    ],
+    "特殊元素": [
+        "凯尔特结图案",
+        "装饰艺术几何元素",
+        "自然风格叶子图案",
+        "复古米粒边细节",
+        "哥特式图案元素",
+        "天体图案(星星、月亮)",
+        "藤蔓装饰曲线",
+        "蝴蝶结装饰"
+    ],
+    "辅石镶嵌": [
+        "密钉镶",
+        "微密钉镶",
+        "槽镶",
+        "珠粒镶",
+        "共爪镶",
+        "包镶",
+        "轨道镶"
+    ]
+}
+
+
+class JewelryLibraryManager:
+    """珠宝词库管理器"""
+
+    def __init__(self, config_dir: Path):
+        """初始化词库管理器
+
+        Args:
+            config_dir: 配置文件目录
+        """
+        self.logger = logging.getLogger(__name__)
+        self.config_dir = config_dir
+        self.config_path = config_dir / "jewelry_library.json"
+        self.library = self.load_library()
+
+    def load_library(self) -> Dict[str, List[str]]:
+        """加载词库，不存在则创建默认词库"""
+        if self.config_path.exists():
+            try:
+                with open(self.config_path, 'r', encoding='utf-8') as f:
+                    library = json.load(f)
+
+                # 检查是否需要进行数据迁移
+                library = self._migrate_library_if_needed(library)
+
+                self.logger.info(f"珠宝词库加载成功: {self.config_path}")
+                return library
+            except Exception as e:
+                self.logger.error(f"珠宝词库加载失败: {e}，使用默认词库")
+                return DEFAULT_JEWELRY_LIBRARY.copy()
+        else:
+            # 首次使用，创建默认词库
+            self.logger.info("未找到珠宝词库文件，创建默认词库")
+            library = DEFAULT_JEWELRY_LIBRARY.copy()
+            self.save_library(library)
+            return library
+
+    def _migrate_library_if_needed(self, library: Dict[str, List[str]]) -> Dict[str, List[str]]:
+        """检查并执行数据迁移"""
+        # 如果存在旧的"主石"字段，需要拆分
+        if "主石" in library:
+            self.logger.info("检测到旧版主石字段，执行数据迁移...")
+
+            # 创建新字段
+            shapes = []
+            materials = []
+
+            # 分析现有数据
+            for item in library["主石"]:
+                # 提取形状信息
+                shape = self._extract_shape(item)
+                if shape and shape not in shapes:
+                    shapes.append(shape)
+
+                # 提取材质信息
+                material = self._extract_material(item)
+                if material and material not in materials:
+                    materials.append(material)
+
+            # 更新词库
+            library["主石形状"] = shapes if shapes else DEFAULT_JEWELRY_LIBRARY["主石形状"]
+            library["主石材质"] = materials if materials else DEFAULT_JEWELRY_LIBRARY["主石材质"]
+
+            # 保留旧字段作为备份
+            # library.pop("主石", None)  # 可以选择是否移除
+
+            # 保存迁移后的数据
+            self.save_library(library)
+            self.logger.info(f"数据迁移完成：形状 {len(shapes)} 个，材质 {len(materials)} 个")
+
+        return library
+
+    def _extract_shape(self, item: str) -> Optional[str]:
+        """从主石描述中提取形状"""
+        shapes = ["椭圆形", "圆形", "祖母绿形", "梨形", "垫形", "公主方形", "心形"]
+        for shape in shapes:
+            if item.startswith(shape):
+                return shape
+        return None
+
+    def _extract_material(self, item: str) -> Optional[str]:
+        """从主石描述中提取材质"""
+        shapes = ["椭圆形", "圆形", "祖母绿形", "梨形", "垫形", "公主方形", "心形"]
+        for shape in shapes:
+            if item.startswith(shape):
+                material = item[len(shape):]
+                # 去除可能的连接词
+                if material.startswith("的"):
+                    material = material[1:]
+                return material if material else None
+        return item  # 如果没有形状前缀，返回整个字符串
+
+    def save_library(self, library: Dict[str, List[str]] = None):
+        """保存词库到配置文件
+
+        Args:
+            library: 要保存的词库，如果为None则保存当前词库
+        """
+        if library is None:
+            library = self.library
+
+        try:
+            self.config_dir.mkdir(parents=True, exist_ok=True)
+            with open(self.config_path, 'w', encoding='utf-8') as f:
+                json.dump(library, f, ensure_ascii=False, indent=2)
+            self.logger.info(f"珠宝词库保存成功: {self.config_path}")
+        except Exception as e:
+            self.logger.error(f"珠宝词库保存失败: {e}")
+            raise
+
+    def add_item(self, category: str, value: str):
+        """添加词库项
+
+        Args:
+            category: 类别名称（如"主石"）
+            value: 词条内容（纯中文）
+        """
+        if category not in self.library:
+            self.library[category] = []
+
+        if value not in self.library[category]:
+            self.library[category].append(value)
+            self.save_library()
+            self.logger.info(f"添加词库项: {category} -> {value}")
+        else:
+            self.logger.warning(f"词库项已存在: {category} -> {value}")
+
+    def remove_item(self, category: str, value: str):
+        """删除词库项
+
+        Args:
+            category: 类别名称
+            value: 词条内容
+        """
+        if category in self.library and value in self.library[category]:
+            self.library[category].remove(value)
+            self.save_library()
+            self.logger.info(f"删除词库项: {category} -> {value}")
+        else:
+            self.logger.warning(f"词库项不存在: {category} -> {value}")
+
+    def reset_category(self, category: str):
+        """恢复单个类别的默认词库
+
+        Args:
+            category: 类别名称
+        """
+        if category in DEFAULT_JEWELRY_LIBRARY:
+            self.library[category] = DEFAULT_JEWELRY_LIBRARY[category].copy()
+            self.save_library()
+            self.logger.info(f"恢复默认词库: {category}")
+        else:
+            self.logger.warning(f"未知的类别: {category}")
+
+    def reset_all(self):
+        """恢复所有类别的默认词库"""
+        self.library = DEFAULT_JEWELRY_LIBRARY.copy()
+        self.save_library()
+        self.logger.info("恢复所有默认词库")
+
+
+class PromptAssembler:
+    """Prompt 组装器（纯中文）"""
+
+    BASE_TEMPLATE = """一款高端精品珠宝戒指设计，主石为{主石}，镶嵌于{金属}中。戒头采用{花头形式}围绕主石。戒臂采用{戒臂结构}设计，表面处理为{戒臂处理}。风格元素包括{特殊元素}。辅石采用{辅石镶嵌}增加光彩。高端珠宝渲染，干净的摄影棚光线，精准的金属抛光，强调宝石清晰度，不要出现手部。"""
+
+    @staticmethod
+    def assemble(form_data: Dict[str, str]) -> str:
+        """智能组装 prompt，处理空值
+
+        Args:
+            form_data: 表单数据，key为字段名，value为选中的值
+
+        Returns:
+            组装后的 prompt 字符串
+        """
+        # 提取字段值（支持新旧两种格式）
+        center_stone = form_data.get("主石", "").strip()  # 向后兼容
+
+        # 新格式：拆分形状和材质
+        center_stone_shape = form_data.get("主石形状", "").strip()
+        center_stone_material = form_data.get("主石材质", "").strip()
+        metal = form_data.get("金属", "").strip()
+        halo_style = form_data.get("花头形式", "").strip()
+        shank_structure = form_data.get("戒臂结构", "").strip()
+        shank_treatment = form_data.get("戒臂处理", "").strip()
+        special_motifs = form_data.get("特殊元素", "").strip()
+        accent_setting = form_data.get("辅石镶嵌", "").strip()
+
+        # 构建 prompt 片段
+        parts = []
+
+        # 构建主石描述（优先使用新格式）
+        if center_stone_shape and center_stone_material:
+            # 新格式：形状+材质
+            main_stone_desc = f"{center_stone_shape}{center_stone_material}"
+        elif center_stone:
+            # 旧格式：直接使用
+            main_stone_desc = center_stone
+        elif center_stone_shape:
+            # 只有形状
+            main_stone_desc = center_stone_shape
+        elif center_stone_material:
+            # 只有材质
+            main_stone_desc = center_stone_material
+        else:
+            main_stone_desc = ""
+
+        # 主体部分（主石和金属）
+        if main_stone_desc and metal:
+            parts.append(f"一款高端精品珠宝戒指设计，主石为{main_stone_desc}，镶嵌于{metal}中。")
+        elif metal:
+            parts.append(f"一款高端精品珠宝戒指设计，镶嵌于{metal}中。")
+        elif main_stone_desc:
+            parts.append(f"一款高端精品珠宝戒指设计，主石为{main_stone_desc}。")
+        else:
+            parts.append("一款高端精品珠宝戒指设计。")
+
+        # 戒头部分
+        if halo_style:
+            if main_stone_desc:
+                parts.append(f"戒头采用{halo_style}围绕主石。")
+            else:
+                parts.append(f"戒头采用{halo_style}设计。")
+
+        # 戒臂部分
+        if shank_structure and shank_treatment:
+            parts.append(f"戒臂采用{shank_structure}设计，表面处理为{shank_treatment}。")
+        elif shank_structure:
+            parts.append(f"戒臂采用{shank_structure}设计。")
+        elif shank_treatment:
+            parts.append(f"戒臂表面处理为{shank_treatment}。")
+
+        # 特殊元素
+        if special_motifs:
+            parts.append(f"风格元素包括{special_motifs}。")
+
+        # 辅石镶嵌
+        if accent_setting:
+            parts.append(f"辅石采用{accent_setting}增加光彩。")
+
+        # 固定的渲染要求
+        parts.append("高端珠宝渲染，干净的摄影棚光线，精准的金属抛光，强调宝石清晰度，不要出现手部。")
+
+        return "".join(parts)
+
+
+class StyleDesignerTab(QWidget):
+    """款式设计 Tab（纯中文）"""
+
+    def __init__(self, library_manager: JewelryLibraryManager, parent=None):
+        super().__init__(parent)
+        self.logger = logging.getLogger(__name__)
+        self.library_manager = library_manager
+        self.parent_window = parent
+
+        # 字段顺序
+        self.categories = ["主石形状", "主石材质", "金属", "花头形式", "戒臂结构", "戒臂处理", "特殊元素", "辅石镶嵌"]
+
+        # 存储每个类别的 ComboBox
+        self.combo_boxes = {}
+
+        # 生成相关
+        self.generated_image_bytes = None
+
+        self.setup_ui()
+
+    def setup_ui(self):
+        """创建 UI 布局"""
+        main_layout = QVBoxLayout()
+        main_layout.setContentsMargins(10, 10, 10, 10)
+        main_layout.setSpacing(15)
+
+        # 表单区域 - 每行两列布局
+        form_group = QGroupBox("珠宝元素选择")
+        form_layout = QVBoxLayout()
+
+        # 使用网格布局实现每行两列
+        grid_layout = QGridLayout()
+        grid_layout.setHorizontalSpacing(20)  # 列间距
+        grid_layout.setVerticalSpacing(10)    # 行间距
+
+        for i, category in enumerate(self.categories):
+            field_widget = self.create_field_widget(category)
+            row = i // 2
+            col = i % 2
+            grid_layout.addWidget(field_widget, row, col)
+
+        form_layout.addLayout(grid_layout)
+
+        # 全局按钮区域
+        buttons_layout = QHBoxLayout()
+
+        # 随机生成按钮
+        random_btn = QPushButton("🎲 随机生成参数")
+        random_btn.clicked.connect(self.randomize_parameters)
+        random_btn.setStyleSheet("""
+            QPushButton {
+                background-color: #4CAF50;
+                color: white;
+                padding: 8px;
+                border-radius: 4px;
+                font-weight: bold;
+            }
+            QPushButton:hover {
+                background-color: #45a049;
+            }
+        """)
+        buttons_layout.addWidget(random_btn)
+
+        # 全局恢复按钮
+        reset_all_btn = QPushButton("🔄 恢复所有默认词库")
+        reset_all_btn.clicked.connect(self.reset_all_library)
+        reset_all_btn.setStyleSheet("""
+            QPushButton {
+                background-color: #ff9800;
+                color: white;
+                padding: 8px;
+                border-radius: 4px;
+                font-weight: bold;
+            }
+            QPushButton:hover {
+                background-color: #f57c00;
+            }
+        """)
+        buttons_layout.addWidget(reset_all_btn)
+
+        buttons_layout.addStretch()
+        form_layout.addLayout(buttons_layout)
+
+        form_group.setLayout(form_layout)
+        main_layout.addWidget(form_group)
+
+        # Content row: Prompt (left) + Settings (right) - 与图片生成页面保持一致
+        content_row = QHBoxLayout()
+
+        # Prompt section
+        prompt_group = QGroupBox("Prompt 预览")
+        prompt_layout = QVBoxLayout()
+        self.prompt_preview = QTextEdit()
+        self.prompt_preview.setReadOnly(True)
+        prompt_layout.addWidget(self.prompt_preview)
+        prompt_group.setLayout(prompt_layout)
+        content_row.addWidget(prompt_group, 2)
+
+        # Settings section
+        settings_group = QGroupBox("生成设置")
+        settings_layout = QVBoxLayout()
+
+        aspect_label = QLabel("宽高比")
+        aspect_label.setStyleSheet("QLabel { font-size: 14px; line-height: 18px; }")
+        settings_layout.addWidget(aspect_label)
+        self.aspect_ratio = QComboBox()
+        self.aspect_ratio.addItems(["1:1", "2:3", "3:2", "3:4", "4:3", "16:9"])
+        settings_layout.addWidget(self.aspect_ratio)
+
+        settings_layout.addSpacing(10)
+
+        size_label = QLabel("图片尺寸")
+        size_label.setStyleSheet("QLabel { font-size: 14px; line-height: 18px; }")
+        settings_layout.addWidget(size_label)
+        self.image_size = QComboBox()
+        self.image_size.addItems(["1K", "2K", "4K"])
+        self.image_size.setCurrentIndex(1)
+        settings_layout.addWidget(self.image_size)
+
+        settings_layout.addSpacing(10)
+
+        settings_layout.addStretch()
+        settings_group.setLayout(settings_layout)
+        content_row.addWidget(settings_group, 1)
+
+        main_layout.addLayout(content_row)
+
+        # Action buttons - 与图片生成页面保持一致
+        action_layout = QHBoxLayout()
+        self.generate_btn = QPushButton("🎨 生成珠宝图片")
+        self.generate_btn.clicked.connect(self.generate_image)
+        self.generate_btn.setStyleSheet("""
+            QPushButton {
+                background-color: #007AFF;
+                color: white;
+                padding: 12px;
+                border-radius: 4px;
+                font-size: 14px;
+                font-weight: bold;
+            }
+            QPushButton:hover {
+                background-color: #0051D5;
+            }
+        """)
+        action_layout.addWidget(self.generate_btn)
+
+        self.download_btn = QPushButton("💾 下载图片")
+        self.download_btn.clicked.connect(self.download_image)
+        self.download_btn.setEnabled(False)
+        action_layout.addWidget(self.download_btn)
+
+        self.status_label = QLabel("● 就绪")
+        self.status_label.setStyleSheet("QLabel { font-size: 14px; line-height: 18px; }")
+        action_layout.addWidget(self.status_label)
+        action_layout.addStretch()
+
+        main_layout.addLayout(action_layout)
+
+        # Preview section - 与图片生成页面保持一致
+        preview_group = QGroupBox("预览")
+        preview_layout = QVBoxLayout()
+
+        self.result_label = QLabel("生成的图片将在这里显示\n双击用系统查看器打开")
+        self.result_label.setAlignment(Qt.AlignCenter)
+        self.result_label.setMinimumHeight(300)
+        self.result_label.setMinimumWidth(400)
+        self.result_label.setStyleSheet("""
+            QLabel {
+                color: #999999;
+                font-size: 14px;
+                line-height: 18px;
+                border: 1px solid #ddd;
+                border-radius: 4px;
+                background-color: #f9f9f9;
+            }
+        """)
+        self.result_label.setScaledContents(False)
+        # 保存原始的双击处理方法引用
+        self.original_mouseDoubleClickEvent = self.result_label.mouseDoubleClickEvent
+        # 重写双击事件
+        self.result_label.mouseDoubleClickEvent = self.open_fullsize_view
+        preview_layout.addWidget(self.result_label)
+
+        preview_group.setLayout(preview_layout)
+        main_layout.addWidget(preview_group, 1)
+
+        main_layout.addStretch()
+        self.setLayout(main_layout)
+
+        # 初始化 prompt 预览
+        self.update_prompt_preview()
+
+    def randomize_parameters(self):
+        """随机生成一套参数"""
+        for category, combo in self.combo_boxes.items():
+            if combo.count() > 0:
+                random_index = random.randint(0, combo.count() - 1)
+                combo.setCurrentIndex(random_index)
+        # 触发prompt预览更新
+        self.update_prompt_preview()
+
+    def create_field_widget(self, category: str) -> QWidget:
+        """创建单个字段的 UI 组件"""
+        widget = QWidget()
+        layout = QHBoxLayout()
+        layout.setContentsMargins(0, 0, 0, 0)
+
+        # 标签
+        label = QLabel(f"{category}:")
+        label.setMinimumWidth(70)  # 减小标签宽度以适应两列布局
+        label.setStyleSheet("QLabel { font-size: 14px; line-height: 18px; }")
+        layout.addWidget(label)
+
+        # 下拉框
+        combo = QComboBox()
+        combo.addItem("")  # 空选项
+        items = self.library_manager.library.get(category, [])
+        combo.addItems(items)
+        combo.currentTextChanged.connect(self.update_prompt_preview)
+        self.combo_boxes[category] = combo
+        layout.addWidget(combo, 3)
+
+        # 添加按钮
+        add_btn = QPushButton("添加")
+        add_btn.clicked.connect(lambda: self.add_library_item(category))
+        add_btn.setFixedWidth(60)
+        layout.addWidget(add_btn)
+
+        # 删除按钮
+        del_btn = QPushButton("删除")
+        del_btn.clicked.connect(lambda: self.remove_library_item(category))
+        del_btn.setFixedWidth(60)
+        layout.addWidget(del_btn)
+
+        widget.setLayout(layout)
+        return widget
+
+    def update_prompt_preview(self):
+        """实时更新 prompt 预览"""
+        form_data = {}
+        for category, combo in self.combo_boxes.items():
+            form_data[category] = combo.currentText()
+
+        prompt = PromptAssembler.assemble(form_data)
+        self.prompt_preview.setPlainText(prompt)
+
+    def add_library_item(self, category: str):
+        """添加词库项 UI"""
+        value, ok = QInputDialog.getText(
+            self,
+            f"添加{category}词库项",
+            f"请输入新的{category}词条（纯中文）:",
+            QLineEdit.Normal
+        )
+
+        if ok and value.strip():
+            value = value.strip()
+            try:
+                self.library_manager.add_item(category, value)
+                # 刷新下拉框
+                self.refresh_combo_box(category)
+                QMessageBox.information(self, "成功", f"已添加词库项: {value}")
+            except Exception as e:
+                self.logger.error(f"添加词库项失败: {e}")
+                QMessageBox.warning(self, "错误", f"添加失败: {e}")
+        else:
+            self.logger.info(f"用户取消了添加{category}词库项的操作")
+
+    def remove_library_item(self, category: str):
+        """删除词库项 UI"""
+        combo = self.combo_boxes.get(category)
+        if not combo:
+            return
+
+        current_value = combo.currentText()
+        if not current_value:
+            QMessageBox.warning(self, "提示", "请先选择要删除的词库项")
+            return
+
+        reply = QMessageBox.question(
+            self,
+            "确认删除",
+            f"确定要删除词库项 \"{current_value}\" 吗？",
+            QMessageBox.Yes | QMessageBox.No
+        )
+
+        if reply == QMessageBox.Yes:
+            try:
+                self.library_manager.remove_item(category, current_value)
+                # 刷新下拉框
+                self.refresh_combo_box(category)
+                QMessageBox.information(self, "成功", f"已删除词库项: {current_value}")
+            except Exception as e:
+                QMessageBox.warning(self, "错误", f"删除失败: {e}")
+
+    def reset_category_library(self, category: str):
+        """恢复单个类别的默认词库 UI"""
+        reply = QMessageBox.question(
+            self,
+            "确认恢复",
+            f"确定要恢复 \"{category}\" 的默认词库吗？\n这将删除所有自定义词条。",
+            QMessageBox.Yes | QMessageBox.No
+        )
+
+        if reply == QMessageBox.Yes:
+            try:
+                self.library_manager.reset_category(category)
+                # 刷新下拉框
+                self.refresh_combo_box(category)
+                QMessageBox.information(self, "成功", f"已恢复 {category} 的默认词库")
+            except Exception as e:
+                QMessageBox.warning(self, "错误", f"恢复失败: {e}")
+
+    def reset_all_library(self):
+        """恢复所有字段的默认词库 UI"""
+        reply = QMessageBox.question(
+            self,
+            "确认恢复所有默认词库",
+            "确定要恢复所有字段的默认词库吗？\n这将删除所有自定义词条。",
+            QMessageBox.Yes | QMessageBox.No
+        )
+
+        if reply == QMessageBox.Yes:
+            try:
+                self.library_manager.reset_all()
+                # 刷新所有下拉框
+                for category in self.categories:
+                    self.refresh_combo_box(category)
+                QMessageBox.information(self, "成功", "已恢复所有默认词库")
+            except Exception as e:
+                QMessageBox.warning(self, "错误", f"恢复失败: {e}")
+
+    def open_fullsize_view(self, event):
+        """双击打开完整尺寸查看器"""
+        if not self.generated_image_bytes:
+            QMessageBox.warning(self, "提示", "没有可预览的图片")
+            return
+
+        # 保存为临时文件并使用系统查看器打开
+        import tempfile
+        with tempfile.NamedTemporaryFile(suffix='.png', delete=False) as tmp_file:
+            tmp_file.write(self.generated_image_bytes)
+            temp_path = Path(tmp_file.name)
+
+        # 使用系统默认程序打开
+        QDesktopServices.openUrl(QUrl.fromLocalFile(str(temp_path)))
+
+    def refresh_combo_box(self, category: str):
+        """刷新指定类别的下拉框"""
+        combo = self.combo_boxes.get(category)
+        if not combo:
+            return
+
+        # 保存当前选中值
+        current_value = combo.currentText()
+
+        # 清空并重新加载
+        combo.clear()
+        combo.addItem("")  # 空选项
+        items = self.library_manager.library.get(category, [])
+        combo.addItems(items)
+
+        # 尝试恢复选中值
+        index = combo.findText(current_value)
+        if index >= 0:
+            combo.setCurrentIndex(index)
+        else:
+            combo.setCurrentIndex(0)
+
+    def generate_image(self):
+        """调用文生图 API"""
+        # 获取 prompt
+        prompt = self.prompt_preview.toPlainText()
+        if not prompt.strip():
+            QMessageBox.warning(self, "提示", "Prompt 不能为空")
+            return
+
+        # 获取设置
+        aspect_ratio = self.aspect_ratio.currentText()
+        image_size = self.image_size.currentText()
+        model = "imagen-3.0-generate-002"  # 硬编码使用默认模型
+
+        # 获取父窗口的 API key
+        if not hasattr(self.parent_window, 'api_key') or not self.parent_window.api_key:
+            QMessageBox.warning(self, "错误", "未找到 API 密钥，请在主窗口配置")
+            return
+
+        # 非阻塞状态提示
+        self.generate_btn.setEnabled(False)
+        self.status_label.setText("● 生成中...")
+        QApplication.processEvents()
+
+        try:
+            # 创建生成线程
+            self.gen_thread = ImageGenerationWorker(
+                api_key=self.parent_window.api_key,
+                prompt=prompt,
+                images=[],
+                aspect_ratio=aspect_ratio,
+                image_size=image_size,
+                model=model
+            )
+            self.gen_thread.finished.connect(self.on_generation_success)
+            self.gen_thread.error.connect(self.on_generation_error)
+            self.gen_thread.start()
+
+        except Exception as e:
+            self.generate_btn.setEnabled(True)
+            self.status_label.setText("● 就绪")
+            QMessageBox.critical(self, "错误", f"生成失败: {e}")
+
+    def on_generation_success(self, image_bytes: bytes):
+        """生成成功回调"""
+        # 恢复按钮状态
+        self.generate_btn.setEnabled(True)
+        self.status_label.setText("● 就绪")
+
+        self.generated_image_bytes = image_bytes
+
+        # 启用下载按钮
+        self.download_btn.setEnabled(True)
+
+        # 显示图片
+        pixmap = QPixmap()
+        pixmap.loadFromData(image_bytes)
+
+        # 清除之前的文本
+        self.result_label.setText("")
+
+        # 获取标签可用空间
+        label_size = self.result_label.size()
+        if label_size.width() < 100 or label_size.height() < 100:
+            # 如果标签还没有正确的尺寸，使用默认尺寸
+            label_size = QSize(400, 300)
+
+        # 缩放图片以适应显示区域，保持宽高比且完整显示
+        scaled_pixmap = pixmap.scaled(
+            label_size,
+            Qt.KeepAspectRatio,
+            Qt.SmoothTransformation
+        )
+
+        # 设置缩放后的图片，并确保居中对齐
+        self.result_label.setPixmap(scaled_pixmap)
+        self.result_label.setAlignment(Qt.AlignCenter)
+
+        # 保存到历史记录
+        try:
+            # 添加到历史记录管理器
+            if hasattr(self.parent_window, 'history_manager'):
+                timestamp = self.parent_window.history_manager.save_generation(
+                    image_bytes=image_bytes,
+                    prompt=self.prompt_preview.toPlainText(),
+                    reference_images=[],  # 款式设计无参考图
+                    aspect_ratio=self.aspect_ratio.currentText(),
+                    image_size=self.image_size.currentText(),
+                    model="imagen-3.0-generate-002"
+                )
+                self.logger.info(f"款式设计已添加到历史记录: {timestamp}")
+
+                # 刷新历史记录列表
+                if hasattr(self.parent_window, 'refresh_history'):
+                    self.parent_window.refresh_history()
+            else:
+                self.logger.warning("未找到历史记录管理器")
+
+        except Exception as e:
+            self.logger.error(f"保存历史记录失败: {e}")
+
+        # 更新状态提示
+        self.status_label.setText("● 图片生成成功")
+        self.status_label.setStyleSheet("QLabel { color: #34C759; }")
+
+    def on_generation_error(self, error_msg: str):
+        """生成失败回调"""
+        # 恢复按钮状态
+        self.generate_btn.setEnabled(True)
+        self.status_label.setText("● 就绪")
+        QMessageBox.critical(self, "生成失败", error_msg)
+
+    def download_image(self):
+        """下载图片"""
+        if not self.generated_image_bytes:
+            QMessageBox.warning(self, "提示", "没有可下载的图片")
+            return
+
+        file_path, _ = QFileDialog.getSaveFileName(
+            self,
+            "保存图片",
+            f"jewelry_design_{datetime.now().strftime('%Y%m%d%H%M%S')}.png",
+            "PNG Files (*.png)"
+        )
+
+        if file_path:
+            try:
+                with open(file_path, 'wb') as f:
+                    f.write(self.generated_image_bytes)
+                # 使用状态栏提示而不是弹窗
+                self.status_label.setText("● 图片已保存")
+                self.status_label.setStyleSheet("QLabel { color: #34C759; }")
+            except Exception as e:
+                QMessageBox.critical(self, "错误", f"保存失败: {e}")
+
+    def preview_image(self):
+        """预览大图"""
+        if not self.generated_image_bytes:
+            QMessageBox.warning(self, "提示", "没有可预览的图片")
+            return
+
+        # 创建预览对话框
+        dialog = QDialog(self)
+        dialog.setWindowTitle("预览大图")
+        dialog.resize(800, 600)
+
+        layout = QVBoxLayout()
+
+        # 图片标签
+        label = QLabel()
+        pixmap = QPixmap()
+        pixmap.loadFromData(self.generated_image_bytes)
+        label.setPixmap(pixmap)
+        label.setScaledContents(True)
+
+        # 滚动区域
+        scroll = QScrollArea()
+        scroll.setWidget(label)
+        scroll.setWidgetResizable(True)
+        layout.addWidget(scroll)
+
+        # 关闭按钮
+        close_btn = QPushButton("关闭")
+        close_btn.clicked.connect(dialog.close)
+        layout.addWidget(close_btn)
+
+        dialog.setLayout(layout)
+        dialog.exec()
+
+
 def main():
 def main():
     """Main application entry point"""
     """Main application entry point"""
     # 初始化日志系统
     # 初始化日志系统

+{
+  "主石": [
+    "椭圆形黑发晶",
+    "圆形莫桑石",
+    "祖母绿形红宝石",
+    "梨形蓝宝石",
+    "垫形绿碧玺",
+    "椭圆形黄水晶带天然包裹体",
+    "公主方形钻石",
+    "心形粉红蓝宝石"
+  ],
+  "金属": [
+    "14K黄金",
+    "18K玫瑰金",
+    "14K白金",
+    "铂金(PT950)",
+    "双色14K黄金+白金",
+    "950银镀铑",
+    "钛金属"
+  ],
+  "花头形式": [
+    "花卉风格光环",
+    "经典圆形光环",
+    "复古风格米粒边光环",
+    "双层光环",
+    "几何六边形光环",
+    "非对称光环",
+    "简约爪镶无光环"
+  ],
+  "戒臂结构": [
+    "扭转戒臂",
+    "分裂戒臂",
+    "刀刃戒臂",
+    "大教堂戒臂",
+    "交叉戒臂",
+    "直线平滑戒臂",
+    "三股编织戒臂"
+  ],
+  "戒臂处理": [
+    "雕刻镂空花丝工艺",
+    "密钉镶戒臂",
+    "抛光平滑戒臂",
+    "浮雕雕刻",
+    "编织纹理",
+    "极简干净戒臂",
+    "穿孔镂空细节",
+    "锤纹处理"
+  ],
+  "特殊元素": [
+    "凯尔特结图案",
+    "装饰艺术几何元素",
+    "自然风格叶子图案",
+    "复古米粒边细节",
+    "哥特式图案元素",
+    "天体图案(星星、月亮)",
+    "藤蔓装饰曲线",
+    "蝴蝶结装饰"
+  ],
+  "辅石镶嵌": [
+    "密钉镶",
+    "微密钉镶",
+    "槽镶",
+    "珠粒镶",
+    "共爪镶",
+    "包镶",
+    "轨道镶"
+  ],
+  "主石形状": [
+    "椭圆形",
+    "圆形",
+    "祖母绿形",
+    "梨形",
+    "垫形",
+    "公主方形",
+    "心形"
+  ],
+  "主石材质": [
+    "黑发晶",
+    "莫桑石",
+    "红宝石",
+    "蓝宝石",
+    "绿碧玺",
+    "黄水晶带天然包裹体",
+    "钻石",
+    "粉红蓝宝石"
+  ]
+}
\ No newline at end of file