我要评论项目概述
这是一个基于wxpython开发的ai绘图提示词（prompt）生成器，专门用于帮助用户快速构建和管理stable diffusion、midjourney等ai绘图工具的提示词。程序提供了两种生成模式：随机生成和手动选择，极大提升了提示词创作效率。
运行结果如下：
核心功能
1. 双模式生成系统
随机生成模式
- 用户设置每行抽取数量（1-5个词组）
- 程序从每行随机选择指定数量的词组
- 自动组合成完整的提示词
手动选择模式
- 为每一行生成独立的下拉框
- 实时预览选择效果
- 支持空选项，灵活控制输出
2. 智能分类系统
程序能够自动识别提示词类型，并为每个下拉框添加有意义的分类标签：
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 "面部特征"
# ... 更多分类逻辑
支持的分类包括：
- 画质质量（masterpiece, 8k, hdr等）
- 面部特征（perfect face, smooth skin等）
- 光照效果（soft lighting, cinematic lighting等）
- 赛博朋克（cyberpunk, neon lights等）
- 景深效果（depth of field, bokeh等）
- 背景设置（simple background, white background等）
- 风格类别（anime style, oil painting, sketch等）
- 人物特征（服装、发型等）
- 场景环境（scenery, forest等）
3. 双语显示机制
这是程序的一大亮点：下拉框显示中英文，但输出纯英文。
def translate_to_chinese(self, text):
"""将英文提示词翻译成中文"""
translations = {
'masterpiece': '杰作',
'best quality': '最佳质量',
'soft lighting': '柔和光照',
# ... 50+个翻译条目
}
return translations.get(text, text)
实现原理：
- 创建两个列表：
combo_choices_display（显示用）和combo_choices_value（值用） - 显示格式：
masterpiece (杰作) - 选择时获取索引，从值列表中提取纯英文
- 最终输出：
masterpiece, best quality, soft lighting
# 构建下拉框选项
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]
技术架构详解
界面布局结构
frame (主窗口)
├── panel (主面板)
├── statictext (输入区标签)
├── textctrl (memo1 - 多行输入框)
├── boxsizer (控制区域)
│ ├── spinctrl (数量选择器)
│ ├── button (随机生成)
│ └── button (生成下拉框)
├── scrolledwindow (可滚动下拉框区域)
│ └── boxsizer (垂直布局)
│ ├── boxsizer (每行水平布局)
│ │ ├── statictext (分类标签)
│ │ └── combobox (下拉选择框)
│ └── ... (多行重复)
├── textctrl (memo2 - 只读输出框)
└── button (复制按钮)
核心类设计
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)
关键技术点
1. 动态创建控件
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)
要点：
- 使用
clear(true)清空并销毁旧控件 - 循环创建控件并绑定事件
- 使用列表存储控件引用，便于后续访问
2. 实时更新机制
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)
实现逻辑：
- 所有下拉框共享同一个事件处理函数
- 事件触发时遍历所有下拉框
- 提取选中项的英文值
- 组合成字符串更新到memo2
3. 随机生成算法
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)
特点：
- 使用
random.sample()保证不重复 - 处理词组数量不足的边界情况
- 按行顺序组合结果
4. 剪贴板操作
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)
wxpython剪贴板api：
wx.theclipboard.open()- 打开剪贴板wx.textdataobject(text)- 创建文本数据对象setdata()- 设置数据close()- 关闭剪贴板
数据流程图
用户输入文本
↓
点击"生成下拉框"
↓
解析每行文本 → 分割词组 → 识别分类
↓
创建下拉框 + 添加翻译 → 绑定事件
↓
用户选择词组
↓
触发on_combo_change事件
↓
遍历所有下拉框 → 提取英文值
↓
更新memo2显示
↓
用户点击复制 → 写入剪贴板
使用场景示例
ai绘图工作流
步骤1：准备提示词库
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
步骤2：生成下拉框
- 画质质量：masterpiece (杰作)
- 动漫风格：anime style (动漫风格)
- 光照效果：soft lighting (柔和光照)
- 发型发色：red hair (红发)
步骤3：选择组合
点击下拉框选择想要的效果
步骤4：导出使用
masterpiece, anime style, soft lighting, red hair
复制到stable diffusion或midjourney使用
扩展建议
1. 增加提示词权重
# 支持(word:1.2)格式
combo_choices_display.append(f'{word} ({chinese}) [权重]')
2. 保存/加载配置
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)
3. 负面提示词支持
添加第二个memo区域用于生成negative prompts。
4. 预设模板系统
templates = {
'人物肖像': ['masterpiece', 'perfect face', 'soft lighting'],
'风景场景': ['scenery', 'detailed environment', 'natural light'],
}
5. 批量生成
添加循环生成功能，一次生成多组不同的提示词。
性能优化
1. 控件复用
对于频繁切换的场景，考虑复用控件而不是销毁重建：
# 只更新choices而不是重建控件 combo.clear() combo.append(new_choices)
2. 延迟更新
用户快速切换下拉框时避免频繁更新：
def on_combo_change(self, event):
if hasattr(self, 'update_timer'):
self.update_timer.stop()
self.update_timer = wx.calllater(300, self.update_memo2)
3. 虚拟滚动
当行数超过100时，考虑使用虚拟列表控件。
代码规范亮点
- 清晰的职责分离：翻译、分类、ui逻辑分别封装
- 防御性编程：处理空输入、边界情况
- 用户友好：提供提示消息、默认值、示例数据
- 数据与显示分离：
combo_values存储实际值，显示中英文
以上就是python结合wxpython实现ai绘图提示词生成器的详细内容，更多关于python ai绘图提示词生成器的资料请关注代码网其它相关文章！
发表评论