Upload idol_generator.py

idol_generator.py

@@ -0,0 +1,800 @@
+import random
+from typing import Dict, List, Optional
+
+
+class IdolPhotoGenerator:
+    def __init__(self):
+        # 初始化所有类别的词库
+        self.vocab = self._initialize_vocabulary()
+
+    def _initialize_vocabulary(self) -> Dict[str, List[str]]:
+        """初始化写真描述词库"""
+        return {
+            # 地点 (50个选项)
+            "locations": [
+                "东京表参道梧桐树街道",
+                "京都伏见稻荷千本鸟居",
+                "北海道富良野薰衣草田",
+                "冲绳宫古岛翡翠海滩",
+                "轻井泽森林教堂",
+                "镰仓高校前海岸线",
+                "大阪环球影城哈利波特区",
+                "白川乡合掌造村落",
+                "奈良公园鹿群中",
+                "六本木之丘观景台",
+                "涩谷全向十字路口",
+                "隅田川花火大会现场",
+                "代代木公园樱花树下",
+                "横滨红砖仓库广场",
+                "箱根芦之湖海盗船上",
+                "浅草寺雷门灯笼前",
+                "皇居东御苑碎石路",
+                "竹下通可丽饼店前",
+                "台场彩虹摩天轮车厢",
+                "明治神宫祈福木牌处",
+                "上野动物园熊猫馆",
+                "新宿御苑日式庭院",
+                "池袋太阳城水族馆",
+                "原宿竹下通彩虹棉花糖店",
+                "秋叶原女仆咖啡厅",
+                "筑地市场金枪鱼拍卖场",
+                "银座三越百货橱窗前",
+                "谷中银座夕阳阶梯",
+                "自由之丘甜品店内",
+                "下北泽古着店试衣间",
+                "吉祥寺井之头公园",
+                "三鹰之森吉卜力美术馆",
+                "东京站丸之内出口",
+                "东京塔特别观景层",
+                "晴空塔天空回廊",
+                "月岛文字烧小巷",
+                "两国国技馆相扑赛场",
+                "柴又帝释天参道",
+                "深大寺荞麦面屋",
+                "高尾山登山缆车",
+                "江之岛展望灯塔",
+                "镰仓文学馆玫瑰园",
+                "热海亲水公园",
+                "伊豆仙人掌动物园",
+                "日光东照宫阳明门",
+                "郡山鹤城天守阁",
+                "松本城黑漆城墙",
+                "金泽兼六园雪吊前",
+                "飞驒高山古街酒藏",
+                "熊本城樱花广场",
+                "巴黎埃菲尔铁塔观景平台",
+                "伦敦大本钟前红色电话亭",
+                "纽约时代广场霓虹灯下",
+                "罗马斗兽场拱门前",
+                "巴塞罗那圣家堂彩窗光影中",
+                "威尼斯圣马可广场鸽子群",
+                "布拉格查理大桥雕像旁",
+                "阿姆斯特丹运河游船上",
+                "维也纳金色大厅台阶",
+                "瑞士少女峰滑雪场",
+                "希腊圣托里尼蓝顶教堂",
+                "冰岛蓝湖温泉雾气中",
+                "芬兰玻璃极光屋",
+                "挪威松恩峡湾观景台",
+                "莫斯科红场圣瓦西里教堂前",
+                "里斯本28号电车车厢",
+                "布达佩斯塞切尼温泉",
+                "哥本哈根新港彩色房屋",
+                "斯德哥尔摩地铁艺术站",
+                "都柏林圣三一学院图书馆",
+                "首尔明洞购物街",
+                "济州岛汉拿山茶园",
+                "釜山海云台沙滩",
+                "首尔N首尔塔爱情锁墙",
+                "台北101观景台",
+                "九份山城红灯笼街",
+                "垦丁白沙湾",
+                "香港维多利亚港夜景",
+                "澳门大三巴牌坊前",
+                "新加坡滨海湾金沙无边泳池",
+                "新加坡植物园交响乐湖",
+                "吉隆坡双子塔空中桥梁",
+                "曼谷大皇宫玉佛寺",
+                "清迈素贴山双龙寺",
+                "巴厘岛乌布丛林秋千",
+                "岘港巴拿山黄金桥",
+                "河内还剑湖龟塔",
+                "吴哥窟日出剪影",
+                "悉尼歌剧院贝壳前",
+                "墨尔本布莱顿彩虹小屋",
+                "皇后镇极限运动中心",
+                "大堡礁心形礁水上飞机",
+                "罗托鲁瓦地热公园",
+                "斐济珊瑚海岸",
+                "塔希提岛水上屋",
+                "上海外滩万国建筑群",
+                "北京故宫太和殿广场",
+                "杭州西湖断桥残雪",
+                "张家界国家森林公园",
+                "桂林漓江竹筏上",
+                "西安大唐不夜城",
+                "重庆太古里",
+                "厦门鼓浪屿钢琴码头",
+                "南京秦淮河畔",
+                "苏州拙政园水廊",
+                "大理洱海白色桌椅",
+                "丽江古城四方街",
+                "青海茶卡盐湖天空之镜",
+                "三亚亚龙湾白沙滩",
+                "哈尔滨冰雪大世界冰雕前",
+                "敦煌鸣沙山月牙泉",
+                "香格里拉普达措国家公园",
+                "台北故宫博物院翠玉白菜展柜",
+                "香港迪士尼城堡前",
+                "开罗金字塔骆驼队前",
+                "马拉喀什马约尔花园",
+                "开普敦桌山缆车",
+                "塞伦盖蒂草原热气球",
+                "毛里求斯七色土",
+                "马达加斯加猴面包树大道",
+                "里约热内卢基督像脚下",
+                "秘鲁马丘比丘太阳门",
+                "布宜诺斯艾利斯博卡区彩色房子",
+                "乌尤尼盐沼天空之镜",
+                "加拉帕戈斯群岛海狮旁",
+                "墨西哥奇琴伊察金字塔",
+                "古巴哈瓦那老爷车街头",
+                "智利百内国家公园",
+                "哥伦比亚瓜塔佩巨岩",
+                "温哥华斯坦利公园",
+                "尼亚加拉大瀑布观景台",
+                "黄石国家公园老忠实泉",
+                "大峡谷国家公园玻璃桥",
+                "纽约中央公园弓桥",
+                "旧金山金门大桥雾中",
+                "洛杉矶好莱坞标志",
+                "迈阿密南海滩装饰艺术区",
+                "芝加哥云门雕塑",
+                "新奥尔良法国区",
+                "夏威夷威基基海滩",
+                "阿拉斯加极光观测站",
+                "波士顿自由之路",
+                "西雅图太空针塔",
+                "拉斯维加斯百乐宫喷泉",
+                "华盛顿国家广场",
+                "费城自由钟",
+                "圣迭戈海洋世界",
+                "奥兰多迪士尼魔法王国",
+                "纽约布鲁克林大桥",
+                "羚羊峡谷光束下",
+                "优胜美地国家公园",
+                "拱门国家公园精致拱门",
+                "大提顿国家公园",
+                "佛罗里达清水湾",
+                "基韦斯特日落庆典",
+            ],
+            # 穿搭 (50个选项)
+            "outfits": [
+                "纯白蕾丝连衣短裙",
+                "浅蓝水手制服配红色领巾",
+                "樱花粉针织开衫+格纹短裙",
+                "薄荷绿吊带纱裙",
+                "米色牛角扣大衣+贝雷帽",
+                "深蓝学院风西服套装",
+                "乳白色羊绒高领毛衣",
+                "鹅黄色碎花连衣裙",
+                "珊瑚粉卫衣+牛仔背带裤",
+                "香芋紫泡泡袖上衣",
+                "珍珠白缎面吊带睡裙",
+                "红白条纹海魂衫",
+                "橄榄绿工装连体裤",
+                "焦糖色麂皮A字裙",
+                "星空蓝渐变星空裙",
+                "经典款深色振袖和服",
+                "浅灰运动套装+白色球鞋",
+                "黑白女仆装+蕾丝头饰",
+                "酒红色天鹅绒晚礼服",
+                "卡其色风衣+碎花连衣裙",
+                "海军蓝双排扣呢子大衣",
+                "淡黄色蓬蓬裙芭蕾舞服",
+                "墨绿色灯芯绒背带裙",
+                "象牙白针织长裙",
+                "藏青色弓道服+袴",
+                "彩虹条纹针织衫",
+                "雾霾蓝毛衣+白色短裤",
+                "樱桃图案居家T恤",
+                "宝蓝色露肩晚装",
+                "姜黄色针织帽+围巾套装",
+                "孔雀绿丝绸衬衫",
+                "黑白波点雪纺裙",
+                "珊瑚绒连体动物睡衣",
+                "深灰色百褶裙+白衬衫",
+                "玫瑰金亮片舞台装",
+                "燕麦色麻花毛衣",
+                "复古格子三件套",
+                "荧光粉运动Bra+瑜伽裤",
+                "深紫色魔法学院袍",
+                "米奇联名款卫衣",
+                "渐变星空色打歌服",
+                "薄荷巧克力色套装",
+                "珍珠镶边小黑裙",
+                "向日葵印花连衣裙",
+                "蒸汽朋克风皮质背心",
+                "淡蓝色汉元素襦裙",
+                "红黑棋盘格针织衫",
+                "银色未来感PVC外套",
+                "泰迪熊绒外套",
+                "手绘涂鸦牛仔外套",
+            ],
+            # 妆容 (50个选项)
+            "makeup": [
+                "清透伪素颜妆，仅涂润唇膏",
+                "蜜桃色系果汁妆，脸颊闪着细闪",
+                "粉棕色系通勤妆，眼线微微上挑",
+                "橘调元气妆，大面积腮红横扫",
+                "浆果色宿醉妆，眼睑下方泛红",
+                "金色闪片舞台妆，夸张假睫毛",
+                "冷冽烟熏妆，搭配深色唇釉",
+                "红茶系温柔妆，眼角点缀泪痣",
+                "蜜糖棕色系妆，卧蚕明显",
+                "草莓牛奶妆，唇峰点涂高光",
+                "人鱼姬偏光眼影妆",
+                "番茄红色系夏日妆",
+                "南瓜色秋季限定妆",
+                "紫罗兰梦幻渐变眼影",
+                "大地色系安全日常妆",
+                "钻石泪滴闪片妆",
+                "复古红唇港风妆",
+                "小鹿斑比无辜大眼妆",
+                "晒伤雀斑妆",
+                "玻璃唇釉水光妆",
+                "雾面哑光高级妆",
+                "粉色系桃花约会妆",
+                "蓝色妖姬大胆眼影",
+                "银色系未来感妆容",
+                "绿色清新春日妆",
+                "焦糖奶茶色全妆",
+                "血色感唇颊统一妆",
+                "金色华丽和风妆",
+                "黑色哥特式暗黑妆",
+                "彩虹渐变眼影妆",
+                "珍珠白新娘妆",
+                "巧克力棕色系妆",
+                "玫瑰金晚宴妆",
+                "橘棕色枫叶妆",
+                "紫灰色小烟熏妆",
+                "蜜粉烘焙定妆法",
+                "光泽肌水润妆",
+                "雾眉+野生眉组合",
+                "截断式cut crease眼妆",
+                "全身亮粉派对妆",
+                "病弱感苍白妆容",
+                "晒伤感高原红妆",
+                "金属色机械姬妆",
+                "亮片星星眼妆",
+                "渐变腮红微醺妆",
+                "纯黑眼线猫眼妆",
+                "泪眼汪汪楚楚可怜妆",
+                "金属蓝埃及艳后妆",
+                "樱花粉日系麻豆妆",
+                "赤茶色秋冬妆",
+                "透明感夏日祭浴衣妆",
+            ],
+            # 动作 (50个选项)
+            "poses": [
+                "双手捧脸歪头微笑",
+                "跳跃时长发飞扬瞬间",
+                "回头时眼神捕捉镜头",
+                "坐在台阶上抱膝",
+                "用手指轻轻点嘴唇",
+                "伸懒腰时展现身体曲线",
+                "双手比心放在胸前",
+                "假装被风吹起裙摆",
+                "躲在树后探出半个身子",
+                "躺在草坪上看云",
+                "用吸管喝饮料时抬眼",
+                "对着镜子涂口红",
+                "系鞋带时抬头微笑",
+                "手拿气球奔跑回头",
+                "趴在咖啡厅桌上小憩",
+                "伸手接住飘落的樱花",
+                "用书本遮住半张脸",
+                "整理耳边的碎发",
+                "踮起脚尖伸手够东西",
+                "吹灭生日蜡烛瞬间",
+                "将耳机分给镜头一半",
+                "撑着透明雨伞转圈",
+                "喂食小鹿时被抓拍",
+                "滑滑板时失去平衡",
+                "弹钢琴时侧脸特写",
+                "荡秋千到最高点时",
+                "从门后突然跳出",
+                "用拍立得自拍时",
+                "手拿棉花糖正要吃",
+                "给植物浇水时专注",
+                "手指在玻璃上画心",
+                "背对镜头转身瞬间",
+                "整理和服袖子时",
+                "手举烟花绽放笑容",
+                "模仿雕像姿势",
+                "在书店踮脚取书",
+                "泡温泉时肩部特写",
+                "早上刚睡醒揉眼睛",
+                "收到惊喜礼物捂嘴",
+                "被玩笑逗得前仰后合",
+                "紧张地抓住裙摆",
+                "模仿经典电影海报",
+                "在厨房打鸡蛋失败",
+                "用望远镜观察远方",
+                "堆雪人时鼻尖冻红",
+                "在海边写名字被浪冲",
+                "吹蒲公英的种子",
+                "偷偷瞄喜欢的人",
+                "练习舞蹈时挥洒汗水",
+                "被突然的喷泉淋湿",
+            ],
+            # 表情 (50个选项)
+            "expressions": [
+                "露出八颗牙的标准偶像笑",
+                "微微抿嘴的害羞笑容",
+                "瞪大眼睛的惊讶表情",
+                "闭眼享受的幸福微笑",
+                "略带淘气的wink表情",
+                "充满感激的泪光闪闪",
+                "专注认真的凝视",
+                "恶作剧得逞的坏笑",
+                "困倦的眯眼打哈欠",
+                "吃到美味的陶醉神情",
+                "鼓起脸颊的假装生气",
+                "憧憬向往的星星眼",
+                "被夸奖后不好意思",
+                "胜利时得意的V字笑",
+                "思考时的微微皱眉",
+                "看到感人场景的落泪",
+                "寂寞孤单的侧脸",
+                "充满决心的坚毅眼神",
+                "撒娇时的嘟嘴表情",
+                "优雅从容的淡淡微笑",
+                "充满好奇的探头表情",
+                "满足的眯眼笑",
+                "略带伤感的微笑",
+                "充满活力的露齿笑",
+                "诱惑的舔嘴唇",
+                "天真的无邪大笑",
+                "温柔注视的母爱表情",
+                "挑衅的挑眉",
+                "恐惧的缩肩表情",
+                "惊喜的双手捂脸",
+                "疲惫但满足的微笑",
+                "困惑的歪头",
+                "感动的嘴唇颤抖",
+                "自信的直视镜头",
+                "害羞的低头浅笑",
+                "淘气的吐舌头",
+                "愤怒的瞪视",
+                "安心的放松表情",
+                "俏皮的皱鼻子",
+                "充满希望的笑容",
+                "狡猾的眯眼笑",
+                "心碎的眼眶泛红",
+                "骄傲的抬头挺胸",
+                "淘气的小恶魔笑",
+                "慈爱的温柔微笑",
+                "紧张的咬嘴唇",
+                "病弱的苍白微笑",
+                "女王般的高傲神情",
+                "恍惚的出神表情",
+                "热情似火的大笑",
+                "神秘的蒙娜丽莎微笑",
+            ],
+            # 情绪 (50个选项)
+            "moods": [
+                "初恋般的心动雀跃",
+                "毕业季的淡淡忧伤",
+                "圣诞节的温暖幸福",
+                "夏日祭的兴奋期待",
+                "清晨醒来的慵懒满足",
+                "雨天的宁静忧郁",
+                "获得奖项的激动喜悦",
+                "告别时的依依不舍",
+                "初次登台的紧张不安",
+                "收到情书的甜蜜羞涩",
+                "看到彩虹的惊喜感动",
+                "失败后的坚强振作",
+                "宠物离去的悲伤寂寞",
+                "梦想成真的难以置信",
+                "等待约会的小鹿乱撞",
+                "被误解的委屈难过",
+                "帮助他人的快乐充实",
+                "独自旅行的自由畅快",
+                "比赛前的斗志昂扬",
+                "深夜的孤独沉思",
+                "家庭团聚的温馨感动",
+                "突破自我的骄傲自豪",
+                "青春期的迷茫困惑",
+                "初雪的浪漫幻想",
+                "海边日出的希望重生",
+                "樱花飘落的物哀之美",
+                "游乐园的纯粹快乐",
+                "图书馆的安静专注",
+                "舞台上的光芒四射",
+                "后台的疲惫但幸福",
+                "离乡背井的思乡之情",
+                "重逢的欣喜若狂",
+                "秘密被发现的惊慌",
+                "挑战极限的刺激快感",
+                "被背叛的心碎痛苦",
+                "和解后的释然轻松",
+                "初为人师的紧张责任",
+                "创作灵感迸发的兴奋",
+                "照顾生病朋友的担忧",
+                "实现承诺的满足安心",
+                "发现真相的震惊愤怒",
+                "节日孤独的寂寞冷清",
+                "收到惊喜的感动流泪",
+                "改变形象的忐忑期待",
+                "引领团队的领导压力",
+                "放下过去的释怀平静",
+                "新环境的好奇探索",
+                "告白的勇敢决绝",
+                "被认可的价值感",
+                "失去方向的迷茫空虚",
+                "拯救他人的使命感",
+                "简单日常的小确幸",
+            ],
+            # 灯光 (50个选项)
+            "lighting": [
+                "清晨五点的金色侧逆光",
+                "正午树荫下的斑驳光点",
+                "下午三点的柔光箱补光",
+                "黄昏时分的暖色轮廓光",
+                "夜晚霓虹灯的彩色映照",
+                "烛光的摇曳温暖光照",
+                "舞台追光灯的强烈聚焦",
+                "窗边自然光的柔和漫射",
+                "阴天均匀的漫反射光",
+                "逆光形成的发丝光效果",
+                "路灯下的昏黄光影",
+                "闪光灯直射的硬朗光线",
+                "水面的波光粼粼反射",
+                "镜面反射的二次补光",
+                "篝火的跳跃橙色光芒",
+                "电子屏幕的冷色调光",
+                "车灯流线的动态轨迹",
+                "月光洒下的清冷银辉",
+                "雨滴折射的模糊光斑",
+                "雾气中的朦胧散射光",
+                "百叶窗的光影条纹",
+                "水晶吊灯的华丽折射",
+                "隧道口的明暗对比",
+                "迪斯科球旋转光点",
+                "荧光涂料的夜光效果",
+                "投影仪的画面映射",
+                "烟花瞬间的绚烂照明",
+                "极光的梦幻渐变色彩",
+                "海底的蓝色透射光线",
+                "晨曦中的耶稣光丁达尔",
+                "烛台的多重阴影",
+                "手术灯的无影效果",
+                "钨丝灯的温暖黄光",
+                "LED灯带的线性引导",
+                "火把的摇曳阴影",
+                "探照灯的强烈光束",
+                "彩色玻璃的斑斓投影",
+                "雪地的反光补脸",
+                "黑暗中的单一聚光",
+                "多云天气的柔和顶光",
+                "日落时分的粉紫色光",
+                "美术馆的重点照明",
+                "手机屏幕的微弱补光",
+                "闪电瞬间的惨白照亮",
+                "激光秀的线条切割",
+                "荧光棒的演唱会海洋",
+                "暗房红灯的特殊色调",
+                "宝石折射的七彩光芒",
+                "雾霾中的光晕效果",
+                "烈日下的高对比阴影",
+                "舞台染色灯的彩色氛围",
+                "荧光蘑菇的微弱生物光",
+                "朝阳穿透森林的晨雾",
+            ],
+        }
+
+    def _get_theme_vocab(self, theme: str) -> Dict[str, List[str]]:
+        """根据主题筛选词库"""
+        theme_keywords = {
+            "校园青春": {
+                "locations": ["东京", "镰仓", "学校", "教室", "图书馆", "操场"],
+                "outfits": ["水手服", "学院风", "针织衫", "格纹裙", "白衬衫"],
+                "moods": ["青春", "初恋", "毕业", "友情", "成长"],
+            },
+            "和风传统": {
+                "locations": ["京都", "奈良", "神社", "寺庙", "古街"],
+                "outfits": ["和服", "浴衣", "振袖", "袴", "草履"],
+                "moods": ["古典", "优雅", "宁静", "物哀", "传统"],
+            },
+            "夏日海滩": {
+                "locations": ["冲绳", "海滩", "海岛", "泳池", "海滨"],
+                "outfits": ["泳装", "沙滩裙", "草帽", "比基尼", "短裤"],
+                "moods": ["活力", "热情", "清凉", "度假", "阳光"],
+            },
+            "都市时尚": {
+                "locations": ["涩谷", "银座", "表参道", "商场", "咖啡厅"],
+                "outfits": ["设计师", "时尚", "潮流", "名牌", "混搭"],
+                "moods": ["酷炫", "自信", "成熟", "都市", "前卫"],
+            },
+        }
+
+        # 如果没有指定主题，返回全部词汇
+        if theme not in theme_keywords:
+            return self.vocab
+
+        # 根据主题关键词筛选
+        themed_vocab = {}
+        for category, words in self.vocab.items():
+            if (
+                category in ["locations", "outfits", "moods"]
+                and theme in theme_keywords
+            ):
+                # 筛选包含主题关键词的词汇
+                keywords = theme_keywords[theme].get(category, [])
+                filtered = [
+                    word
+                    for word in words
+                    if any(keyword in word for keyword in keywords)
+                ]
+                themed_vocab[category] = filtered if filtered else words[:20]
+            else:
+                themed_vocab[category] = words
+
+        return themed_vocab
+
+    def generate_photo_description(
+        self, num: int = 1, theme: Optional[str] = None
+    ) -> List[Dict]:
+        """生成写真描述，可选主题"""
+        results = []
+
+        # 获取对应主题的词库
+        current_vocab = self._get_theme_vocab(theme) if theme else self.vocab
+
+        for i in range(num):
+            description = {
+                "编号": i + 1,
+                "主题": theme if theme else "随机",
+                "地点": random.choice(current_vocab["locations"]),
+                "穿搭": random.choice(current_vocab["outfits"]),
+                "妆容": random.choice(current_vocab["makeup"]),
+                "动作": random.choice(current_vocab["poses"]),
+                "表情": random.choice(current_vocab["expressions"]),
+                "情绪": random.choice(current_vocab["moods"]),
+                "灯光": random.choice(current_vocab["lighting"]),
+            }
+            results.append(description)
+
+        return results
+
+    def generate_unique(
+        self, num: int = 1, theme: Optional[str] = None, unique_keys: List[str] = None
+    ) -> List[Dict]:
+        """
+        确保生成不重复的组合
+
+        Args:
+            num: 生成数量
+            theme: 主题名称
+            unique_keys: 用于判断唯一性的字段列表
+                        默认 ['地点', '动作', '表情']
+        """
+        if unique_keys is None:
+            unique_keys = ["地点", "动作", "表情"]
+
+        results = []
+        seen = set()
+        attempts = 0
+        max_attempts = num * 10  # 防止无限循环
+
+        while len(results) < num and attempts < max_attempts:
+            # 生成一个描述
+            desc = self.generate_photo_description(1, theme)[0]
+
+            # 创建唯一标识键
+            key_parts = [str(desc.get(field, "")) for field in unique_keys]
+            key = "|".join(key_parts)
+
+            # 检查是否重复
+            if key not in seen:
+                seen.add(key)
+                results.append(desc)
+
+            attempts += 1
+
+        # 如果无法生成足够的不重复组合
+        if len(results) < num:
+            print(f"⚠️  注意：只能生成 {len(results)} 个不重复组合")
+
+        return results
+
+    def generate_themed_series(
+        self, theme: str, num: int = 5, series_type: str = "story"
+    ) -> List[Dict]:
+        """
+        生成主题连贯的系列写真
+
+        Args:
+            theme: 主题名称
+            num: 生成数量
+            series_type: 系列类型
+                "story" - 故事连贯
+                "location" - 地点连贯
+                "outfit" - 服装连贯
+        """
+        # 获取主题词库
+        theme_vocab = self._get_theme_vocab(theme)
+
+        # 根据系列类型设置基础元素
+        base_elements = {}
+        if series_type == "location":
+            # 相同地点，不同角度/时间
+            base_location = random.choice(theme_vocab["locations"])
+            base_elements["地点"] = base_location
+        elif series_type == "outfit":
+            # 相同服装，不同场景
+            base_outfit = random.choice(theme_vocab["outfits"])
+            base_elements["穿搭"] = base_outfit
+        elif series_type == "story":
+            # 故事性：连贯的情绪变化
+            mood_progression = ["期待", "紧张", "进行", "高潮", "结束"]
+            if num <= len(mood_progression):
+                base_elements["情绪序列"] = mood_progression[:num]
+
+        results = []
+        for i in range(num):
+            desc = {
+                "编号": i + 1,
+                "主题": theme,
+                "系列类型": series_type,
+                "系列序号": f"{i + 1}/{num}",
+            }
+
+            # 设置系列基础元素
+            if "地点" in base_elements:
+                desc["地点"] = base_elements["地点"]
+            else:
+                desc["地点"] = random.choice(theme_vocab["locations"])
+
+            if "穿搭" in base_elements:
+                desc["穿搭"] = base_elements["穿搭"]
+            else:
+                desc["穿搭"] = random.choice(theme_vocab["outfits"])
+
+            # 如果是故事系列，使用预设的情绪序列
+            if "情绪序列" in base_elements:
+                desc["情绪"] = base_elements["情绪序列"][i]
+            else:
+                desc["情绪"] = random.choice(theme_vocab["moods"])
+
+            # 其他元素随机
+            desc["妆容"] = random.choice(theme_vocab["makeup"])
+            desc["动作"] = random.choice(theme_vocab["poses"])
+            desc["表情"] = random.choice(theme_vocab["expressions"])
+            desc["灯光"] = random.choice(theme_vocab["lighting"])
+
+            results.append(desc)
+
+        return results
+
+    def generate_tweet(self, description: Dict) -> str:
+        """为写真描述生成配套推文"""
+        # 从描述中提取关键元素
+        location_keywords = ["东京", "京都", "北海道", "冲绳", "镰仓", "大阪", "奈良"]
+        location = description["地点"]
+
+        # 确定话题标签
+        tags = [" #王道偶像写真", " #每日一照", " #偶像的日常", " #aifeifei "]
+
+        for keyword in location_keywords:
+            if keyword in location:
+                tags.append(f"#{keyword}摄影")
+                break
+
+        # 随机选择推文风格
+        tweet_templates = [
+            f"今天的拍摄在「{description['地点']}」完成～{description['情绪']}的心情，通过{description['动作']}传达给大家！{''.join(tags)}",
+            f"新写真花絮💖 {description['穿搭']}搭配{description['妆容']}，在{description['灯光']}下{description['表情']}。此刻感觉{description['情绪']}… {''.join(tags)}",
+            f"发现一个绝佳拍摄地：{description['地点']}！尝试了{description['动作']}，不知道效果如何？{description['表情']} {''.join(tags)}",
+            f"【幕后速报】{description['地点']}拍摄中。今天的主题是「{description['情绪']}」，{description['穿搭']}和{description['妆容']}都是为了营造这种氛围～ {''.join(tags)}",
+        ]
+
+        return random.choice(tweet_templates)
+
+
+# ============== 使用示例 ==============
+if __name__ == "__main__":
+    # 1. 创建生成器实例
+    generator = IdolPhotoGenerator()
+
+    print("🎀 增强版写真描述生成器 🎀")
+    print("=" * 60)
+
+    # 2. 使用 generate_unique 功能
+    print("\n🔹 示例1：生成3个不重复的写真描述")
+    print("-" * 40)
+
+    unique_photos = generator.generate_unique(
+        num=3,
+        theme="校园青春",  # 可选主题
+        unique_keys=["地点", "动作", "表情"],  # 可自定义唯一性判断字段
+    )
+
+    for photo in unique_photos:
+        print(f"📸 #{photo['编号']} | 主题:{photo['主题']}")
+        print(f"   地点: {photo['地点']}")
+        print(f"   动作: {photo['动作']}")
+        print(f"   表情: {photo['表情']}")
+        print(f"   情绪: {photo['情绪']}")
+
+    # 3. 使用 generate_themed_series 功能
+    print("\n\n🔹 示例2：生成'和风传统'主题的连贯系列")
+    print("-" * 40)
+
+    themed_series = generator.generate_themed_series(
+        theme="和风传统",
+        num=3,
+        series_type="story",  # 尝试: "location", "outfit", "story"
+    )
+
+    for photo in themed_series:
+        print(f"🎎 #{photo['编号']} | {photo['系列类型']}系列 {photo['系列序号']}")
+        print(f"   地点: {photo['地点']}")
+        print(f"   穿搭: {photo['穿搭']}")
+        print(f"   情绪: {photo['情绪']}")
+        print(f"   动作: {photo['动作']}")
+
+    # # 4. 批量生成并保存
+    # print("\n\n🔹 示例3：批量生成并保存到文件")
+    # print("-" * 40)
+    #
+    # # 生成20个不重复的都市时尚主题写真
+    # batch_results = generator.generate_unique(
+    #     num=20,
+    #     theme="都市时尚",
+    #     unique_keys=["地点", "穿搭", "动作"],  # 可根据需要调整
+    # )
+
+    # # 保存到文件
+    # with open("themed_photos.txt", "w", encoding="utf-8") as f:
+    #     for photo in batch_results:
+    #         f.write(f"写真 #{photo['编号']}\n")
+    #         f.write(f"主题: {photo.get('主题', '随机')}\n")
+    #         f.write(f"地点: {photo['地点']}\n")
+    #         f.write(f"穿搭: {photo['穿搭']}\n")
+    #         f.write(f"妆容: {photo['妆容']}\n")
+    #         f.write(f"动作: {photo['动作']}\n")
+    #         f.write(f"表情: {photo['表情']}\n")
+    #         f.write(f"情绪: {photo['情绪']}\n")
+    #         f.write(f"灯光: {photo['灯光']}\n")
+    #         # 生成推文
+    #         tweet = generator.generate_tweet(photo)
+    #         f.write(f"推文: {tweet}\n")
+    #         f.write("-" * 50 + "\n")
+    #
+    # print(f"✅ 已生成 {len(batch_results)} 个不重复的都市时尚写真描述")
+    # print("✅ 已保存到 'themed_photos.txt'")
+
+    # 5. 询问是否需要50个完整描述
+    print("\n\n🔹 示例4：生成50个随机写真描述")
+    print("-" * 40)
+
+    response = input("是否要生成50个完整写真描述？(y/n): ")
+    if response.lower() == "y":
+        all_photos = generator.generate_photo_description(50)
+        with open("idol_photo_descriptions.txt", "w", encoding="utf-8") as f:
+            for photo in all_photos:
+                f.write(f"写真 #{photo['编号']}\n")
+                f.write(f"地点: {photo['地点']}\n")
+                f.write(f"穿搭: {photo['穿搭']}\n")
+                f.write(f"妆容: {photo['妆容']}\n")
+                f.write(f"动作: {photo['动作']}\n")
+                f.write(f"表情: {photo['表情']}\n")
+                f.write(f"情绪: {photo['情绪']}\n")
+                f.write(f"灯光: {photo['灯光']}\n")
+                tweet = generator.generate_tweet(photo)
+                f.write(f"推文: {tweet}\n")
+                f.write("=" * 60 + "\n")
+        print("✅ 已保存50个完整描述到 idol_photo_descriptions.txt")