Python结合wxPython实现AI绘图提示词生成器

def get_category_name(self, words):
    """根据词组内容推断分类名称"""
    first_words = ' '.join(words[:3]).lower()
    
    if any(kw in first_words for kw in ['masterpiece', 'quality', 'detailed']):
        return "画质质量"
    elif any(kw in first_words for kw in ['face', 'skin', 'eyes']):
        return "面部特征"
    # ... 更多分类逻辑

def translate_to_chinese(self, text):
    """将英文提示词翻译成中文"""
    translations = {
        'masterpiece': '杰作',
        'best quality': '最佳质量',
        'soft lighting': '柔和光照',
        # ... 50+个翻译条目
    }
    return translations.get(text, text)

# 构建下拉框选项
for word in words:
    chinese = self.translate_to_chinese(word)
    display_text = f'{word} ({chinese})'  # 显示：英文(中文)
    combo_choices_display.append(display_text)
    combo_choices_value.append(word)  # 保存：纯英文

# 获取选择时使用英文值
english_value = combo.combo_values[selection_index]

class RandomWordPicker(wx.Frame):
    def __init__(self):
        # 初始化窗口和控件
        self.combo_boxes = []      # 存储所有下拉框
        self.line_data = []        # 存储每行的词组数据
        
    # 翻译系统
    def translate_to_chinese(self, text)
    
    # 分类识别
    def get_category_name(self, words)
    
    # 生成下拉框
    def on_create_combos(self, event)
    
    # 监听下拉框变化
    def on_combo_change(self, event)
    
    # 更新输出区域
    def update_memo2(self)
    
    # 随机生成
    def on_generate(self, event)
    
    # 复制到剪贴板
    def on_copy(self, event)

def on_create_combos(self, event):
    # 清空现有控件
    self.combo_sizer.Clear(True)
    self.combo_boxes = []
    
    # 动态创建每行的下拉框
    for i, line in enumerate(lines):
        words = [w.strip() for w in line.split(',') if w.strip()]
        
        # 创建布局和控件
        row_sizer = wx.BoxSizer(wx.HORIZONTAL)
        label = wx.StaticText(...)
        combo = wx.ComboBox(...)
        
        # 绑定事件
        combo.Bind(wx.EVT_COMBOBOX, self.on_combo_change)
        
        self.combo_boxes.append(combo)

def on_combo_change(self, event):
    # 任何下拉框改变都触发此函数
    self.update_memo2()

def update_memo2(self):
    selected_words = []
    for combo in self.combo_boxes:
        selection_index = combo.GetSelection()
        if selection_index > 0:  # 跳过空选项
            english_value = combo.combo_values[selection_index]
            selected_words.append(english_value)
    
    result = ', '.join(selected_words)
    self.memo2.SetValue(result)

def on_generate(self, event):
    num_to_pick = self.spin_ctrl.GetValue()
    all_selected = []
    
    for line in lines:
        words = [w.strip() for w in line.split(',') if w.strip()]
        
        if len(words) < num_to_pick:
            selected = words  # 不足时全选
        else:
            selected = random.sample(words, num_to_pick)  # 随机抽取
        
        all_selected.extend(selected)
    
    result_text = ', '.join(all_selected)

def on_copy(self, event):
    text = self.memo2.GetValue()
    
    if wx.TheClipboard.Open():
        wx.TheClipboard.SetData(wx.TextDataObject(text))
        wx.TheClipboard.Close()
        wx.MessageBox('已复制到剪贴板！', '成功', wx.OK | wx.ICON_INFORMATION)

masterpiece, best quality, ultra-detailed, 8k
anime style, cel shaded, flat color
soft lighting, natural light, cinematic lighting
red hair, vibrant red hair, detailed hair

masterpiece, anime style, soft lighting, red hair

# 支持(word:1.2)格式
combo_choices_display.append(f'{word} ({chinese}) [权重]')

def save_config(self):
    config = {
        'selections': [combo.GetSelection() for combo in self.combo_boxes]
    }
    with open('config.json', 'w') as f:
        json.dump(config, f)

templates = {
    '人物肖像': ['masterpiece', 'perfect face', 'soft lighting'],
    '风景场景': ['scenery', 'detailed environment', 'natural light'],
}

# 只更新choices而不是重建控件
combo.Clear()
combo.Append(new_choices)

def on_combo_change(self, event):
    if hasattr(self, 'update_timer'):
        self.update_timer.Stop()
    
    self.update_timer = wx.CallLater(300, self.update_memo2)