Spaces:
Build error
Build error
| import gradio as gr | |
| import random | |
| import re | |
| import json | |
| # ========== 可选：加载免费开源小模型（CPU本地推理，无需Key） ========== | |
| USE_LLM = True | |
| tokenizer = model = None | |
| MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct" | |
| try: | |
| import torch | |
| from transformers import AutoTokenizer, AutoModelForCausalLM | |
| tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True) | |
| model = AutoModelForCausalLM.from_pretrained( | |
| MODEL_NAME, | |
| trust_remote_code=True, | |
| torch_dtype=torch.float32 | |
| ) | |
| model.eval() | |
| except Exception: | |
| USE_LLM = False | |
| # ===== 名字合法性：拒绝拼音/奇怪符号，仅允许一个单词 ===== | |
| ALLOWLIST = {"Chloe","Charlotte","Xavier","Zach","Shawn","Shane","Zoe","Zoey","Chase","Charles","Xena","Phoenix"} | |
| PINYIN_FRAGMENTS = [ | |
| "zh","ch","sh","xue","xiao","ming","jing","qiang","qian","qin","qiu","xuan","xun", | |
| "yue","yu","yang","ying","yong","yan","jun","hao","lei","peng","lin","zhao","zhou","zhang", | |
| "wang","li","liu","chen","gao","guo","wu","he","luo","sun","tang","song" | |
| ] | |
| NAME_RE = re.compile(r"^[A-Z][a-z]{2,15}$") | |
| def looks_like_pinyin(name:str)->bool: | |
| return (name not in ALLOWLIST) and any(f in name.lower() for f in PINYIN_FRAGMENTS) | |
| def name_sanity(name:str)->bool: | |
| return bool(NAME_RE.match(name or "")) and (not looks_like_pinyin(name)) | |
| # ===== 基于中文的“主题/意象”提取 ===== | |
| # 主题 -> 候选英文“常规名”和“创意词名” | |
| THEME_POOL = { | |
| "bright": { | |
| "male": ["Leo","Lucian","Clarence","Ray"], | |
| "female": ["Clara","Lucy","Claire","Eleanor"], | |
| "word": ["Bright","Dawn","Halo","Glow","Solar"] | |
| }, | |
| "wisdom": { | |
| "male": ["Solomon","Hugo","Sage"], | |
| "female": ["Sophia","Sage","Minerva"], | |
| "word": ["Oracle","Logic","Riddle","Cipher"] | |
| }, | |
| "brave": { | |
| "male": ["Andrew","Conrad","Leon","Victor"], | |
| "female": ["Valerie","Andrea","Bridget"], | |
| "word": ["Valor","Brave","Vanguard","Grit","Steel"] | |
| }, | |
| "strong": { | |
| "male": ["Ethan","Arnold","Magnus"], | |
| "female": ["Matilda","Valerie"], | |
| "word": ["Atlas","Titan","Forge","Anchor"] | |
| }, | |
| "peace": { | |
| "male": ["Irenaeus","Pax","Felix"], | |
| "female": ["Irene","Serena","Pax"], | |
| "word": ["Truce","Calm","Zen","Peace"] | |
| }, | |
| "snow": { | |
| "male": ["Nevin","Frost"], | |
| "female": ["Neve","Elsa"], | |
| "word": ["Snow","Frost","Blanc","Crystal"] | |
| }, | |
| "pure": { | |
| "male": ["Aaron"], | |
| "female": ["Agnes","Catherine"], | |
| "word": ["Pure","Prism"] | |
| }, | |
| "dawn": { | |
| "male": ["Aiden"], | |
| "female": ["Aurora","Zora"], | |
| "word": ["Dawn","Eos","Sunrise","Firstlight"] | |
| }, | |
| "sunny": { | |
| "male": ["Sunny","Helios"], | |
| "female": ["Sunny","Solana"], | |
| "word": ["Sunny","Sunbeam","Sol"] | |
| }, | |
| "cosmic": { | |
| "male": ["Cosmo","Orion"], | |
| "female": ["Nova","Stella"], | |
| "word": ["Nova","Stellar","Comet","Nebula","Orbit"] | |
| }, | |
| "grand": { | |
| "male": ["August","Grant"], | |
| "female": ["Gloria","Augusta"], | |
| "word": ["Grand","Majesty","Prime"] | |
| }, | |
| "beauty": { | |
| "male": ["Beau"], | |
| "female": ["Bella"], | |
| "word": ["Beauty","Belle"] | |
| }, | |
| "serene": { | |
| "male": ["Irenaeus"], | |
| "female": ["Serena","Irene"], | |
| "word": ["Serene","Still","Quiet"] | |
| }, | |
| "talented": { | |
| "male": ["Aiden","Gavin","Kevin"], | |
| "female": ["Ava","Mia"], | |
| "word": ["Talent","Flair","Spark"] | |
| }, | |
| "positive": { | |
| "male": ["Felix"], | |
| "female": ["Joy"], | |
| "word": ["Joy","Lucky","Fortune"] | |
| }, | |
| # 通用兜底 | |
| "general": { | |
| "male": ["Henry","Daniel","Samuel","Arthur","Miles","James","William","Michael"], | |
| "female": ["Emma","Olivia","Emily","Grace","Lucy","Clara","Jane","Rachel"], | |
| "word": ["Echo","River","Willow","Raven","Skye","Blaze","Rider","Fox","Bolt"] | |
| } | |
| } | |
| # 中文字符到主题映射 | |
| CHAR2THEMES = { | |
| "明": ["bright","wisdom","dawn"], | |
| "文": ["wisdom"], | |
| "静": ["serene","peace"], | |
| "勇": ["brave"], | |
| "伟": ["grand","strong"], | |
| "强": ["strong"], | |
| "俊": ["talented"], | |
| "晨": ["dawn","bright"], | |
| "雪": ["snow","pure"], | |
| "宁": ["peace","serene"], | |
| "美": ["beauty"], | |
| "慧": ["wisdom"], | |
| "宇": ["cosmic","grand"], | |
| "阳": ["sunny","positive","bright"], | |
| } | |
| # 引用库（扩展版，含一些“词名”的流行文化对照） | |
| REFERENCE_DB = { | |
| "Elizabeth": ["Elizabeth Bennet（《傲慢与偏见》）"], | |
| "Darcy": ["Mr. Darcy（《傲慢与偏见》）"], | |
| "Emma": ["Emma Woodhouse（《爱玛》）"], | |
| "Jane": ["Jane Eyre（《简·爱》）"], | |
| "Alice": ["Alice（《爱丽丝梦游仙境》）"], | |
| "Hermione": ["Hermione Granger（《哈利·波特》）"], | |
| "Harry": ["Harry Potter（《哈利·波特》）"], | |
| "Ron": ["Ron Weasley（《哈利·波特》）"], | |
| "Luna": ["Luna Lovegood（《哈利·波特》）"], | |
| "Clara": ["Clara Oswald（《神秘博士》）"], | |
| "Lucy": ["Lucy Pevensie（《纳尼亚传奇》）"], | |
| "Peter": ["Peter Parker（蜘蛛侠）"], | |
| "Tony": ["Tony Stark（钢铁侠）"], | |
| "Bruce": ["Bruce Wayne（蝙蝠侠）"], | |
| "Arthur": ["Arthur（亚瑟王传说）"], | |
| "Victor": ["Victor Frankenstein（《弗兰肯斯坦》）"], | |
| "Magnus": ["Magnus Bane（《凡人圣器》）"], | |
| "Gatsby": ["Jay Gatsby（《了不起的盖茨比》）"], | |
| "Daisy": ["Daisy Buchanan（《了不起的盖茨比》）"], | |
| "Holden": ["Holden Caulfield（《麦田里的守望者》）"], | |
| "Atticus": ["Atticus Finch（《杀死一只知更鸟》）"], | |
| "Romeo": ["Romeo（《罗密欧与朱丽叶》）"], | |
| "Juliet": ["Juliet（《罗密欧与朱丽叶》）"], | |
| "Arya": ["Arya Stark（《权力的游戏》）"], | |
| "Sansa": ["Sansa Stark（《权力的游戏》）"], | |
| "Jon": ["Jon Snow（《权力的游戏》）"], | |
| "Daenerys": ["Daenerys Targaryen（《权力的游戏》）"], | |
| "Sam": ["Samwise Gamgee（《指环王》）"], | |
| "Frodo": ["Frodo Baggins（《指环王》）"], | |
| "Gandalf": ["Gandalf（《指环王》）"], | |
| "Katniss": ["Katniss Everdeen（《饥饿游戏》）"], | |
| "Tris": ["Tris Prior（《分歧者》）"], | |
| "Leo": ["Leo McGarry（《白宫风云》）"], | |
| "Felix": ["Felix Krull（《骗子费利克斯》）"], | |
| "Aurora": ["Aurora（《睡美人》）"], | |
| "Holly": ["Holly Golightly（《蒂凡尼的早餐》）"], | |
| "Henry": ["Henry Higgins（《窈窕淑女》）"], | |
| "Emily": ["Emily Dickinson（诗人）"], | |
| "Olivia": ["Olivia Pope（《丑闻》）"], | |
| "Rachel": ["Rachel Green（《老友记》）"], | |
| "Monica": ["Monica Geller（《老友记》）"], | |
| "Phoebe": ["Phoebe Buffay（《老友记》）"], | |
| "Chandler": ["Chandler Bing（《老友记》）"], | |
| "Ross": ["Ross Geller（《老友记》）"], | |
| "Joey": ["Joey Tribbiani（《老友记》）"], | |
| "Michael": ["Michael Corleone（《教父》）"], | |
| "James": ["James Bond（007）"], | |
| "Ethan": ["Ethan Hunt（《碟中谍》）"], | |
| "Liam": ["Liam Neeson（演员）"], | |
| "Ava": ["Ava（《机械姬》）"], | |
| "Mia": ["Mia Thermopolis（《公主日记》）"], | |
| "Chloe": ["Chloe Price（《奇异人生》）"], | |
| "Willow": ["Willow Rosenberg（《吸血鬼猎人巴菲》）"], | |
| "River": ["River Tam（《萤火虫》）"], | |
| "Raven": ["Raven（《少年泰坦》）"], | |
| "Echo": ["Echo / Maya Lopez（漫威）"], | |
| "Nova": ["Nova（漫威/多部作品）"], | |
| "Logan": ["Logan / Wolverine（《X战警》）"], | |
| "Edward": ["Edward Cullen（《暮光之城》）"], | |
| "Bella": ["Bella Swan（《暮光之城》）"], | |
| "Skye": ["Skye / Daisy Johnson（《神盾局特工》）"], | |
| "Atlas": ["Atlas（希腊神话）"], | |
| "Orion": ["Orion（猎户座/神话）"], | |
| "Comet": ["Comet（多部作品/角色名常见称呼）"], | |
| "Blaze": ["Blaze the Cat（《索尼克》）"], | |
| "Fox": ["Fox Mulder（《X档案》）"], | |
| } | |
| NORMAL_POOL_MALE = ["Ethan","Liam","Noah","Mason","Leo","Andrew","Victor","Felix","Grant","Conrad","Kevin","Gavin","Miles","Henry","Arthur","Daniel","Michael","James","Samuel","Benjamin","Alexander","Logan","Edward","William","Joseph"] | |
| NORMAL_POOL_FEMALE = ["Emma","Olivia","Emily","Grace","Lucy","Clara","Jane","Rachel","Ava","Mia","Chloe","Stella","Bella","Claire","Phoebe","Serena","Irene","Aurora","Zoe","Sophie","Hannah","Sarah","Natalie","Victoria","Willow"] | |
| MALE_SET = set(NORMAL_POOL_MALE) | |
| FEMALE_SET = set(NORMAL_POOL_FEMALE) | |
| FEMALE_HINT = set(list("芳娜丽敏静婷雪梅玲颖倩婧薇诗妍悦宁珂伊菲茜可璇欣美慧")) | |
| MALE_HINT = set(list("强军伟刚涛磊鹏杰阳龙峰超辉宁宇航晨鑫豪剑霖勇健俊")) | |
| def gender_hint_from_cn(name: str) -> str: | |
| score = 0 | |
| for ch in name: | |
| if ch in FEMALE_HINT: score -= 1 | |
| if ch in MALE_HINT: score += 1 | |
| return "male" if score > 0 else "female" | |
| # —— 从中文名提取主题列表（按频次和优先级） —— | |
| def themes_from_cn(cn_name: str): | |
| themes = [] | |
| for ch in cn_name: | |
| themes += CHAR2THEMES.get(ch, []) | |
| if not themes: | |
| themes = ["general"] | |
| # 去重保序 | |
| seen = set() | |
| out = [] | |
| for t in themes: | |
| if t not in seen: | |
| seen.add(t) | |
| out.append(t) | |
| if "general" not in out: | |
| out.append("general") | |
| return out | |
| # —— 动态含义模板（避免每次一样） —— | |
| MEANING_TEMPLATES = [ | |
| "取“{chars}”的意象（{themes_cn}），所以选了 {name}。", | |
| "参考“{chars}”带来的感觉（{themes_cn}），给你 {name}。", | |
| "从“{chars}”联想到 {themes_cn}，因此是 {name}。", | |
| "因为“{chars}”→{themes_cn}，所以用了 {name} 这个名字。", | |
| "由“{chars}”启发（{themes_cn}），名字定为 {name}。" | |
| ] | |
| THEME_ZH = { | |
| "bright":"明亮/光芒","wisdom":"智慧/学识","brave":"勇气/无畏","strong":"强壮/坚韧", | |
| "peace":"平和/宁静","snow":"雪意/纯净","pure":"纯洁","dawn":"黎明/新生","sunny":"阳光/积极", | |
| "cosmic":"宇宙/星空","grand":"宏大/气势","beauty":"美感/优雅","serene":"安静/从容", | |
| "talented":"才华/灵气","positive":"好运/积极","general":"通用/耐看" | |
| } | |
| def render_meaning(cn_name, used_themes, name): | |
| chars = "".join([ch for ch in cn_name if ch in CHAR2THEMES]) or cn_name | |
| zh = "、".join([THEME_ZH.get(t,t) for t in used_themes[:2]]) # 取前两个主题做说明 | |
| tpl = random.choice(MEANING_TEMPLATES) | |
| return tpl.format(chars=chars, themes_cn=zh, name=name) | |
| # —— 主题驱动选名：可能是常规英文名，也可能是“词名” —— | |
| def themed_pick(cn_name: str, gender: str, creative_ratio=0.4, seed=None): | |
| rnd = random.Random(seed) | |
| themes = themes_from_cn(cn_name) | |
| t = rnd.choice(themes[:2] or themes) # 更偏向前两个主题 | |
| pool = THEME_POOL.get(t, THEME_POOL["general"]) | |
| creative = rnd.random() < creative_ratio | |
| if creative: | |
| # 允许词名（不做男女白名单） | |
| cand = [w for w in pool["word"] if name_sanity(w)] | |
| if not cand: | |
| cand = [w for w in THEME_POOL["general"]["word"] if name_sanity(w)] | |
| pick = rnd.choice(cand) | |
| kind = "word" | |
| else: | |
| base = pool["male"] if gender=="male" else pool["female"] | |
| # 性别白名单 + 合法性 | |
| cand = [n for n in base if name_sanity(n) and ((n in MALE_SET) if gender=="male" else (n in FEMALE_SET))] | |
| if not cand: | |
| base2 = NORMAL_POOL_MALE if gender=="male" else NORMAL_POOL_FEMALE | |
| cand = [n for n in base2 if name_sanity(n)] | |
| pick = rnd.choice(cand) | |
| kind = "given" | |
| meaning = render_meaning(cn_name, [t], pick) | |
| # 引用：优先直接命中；如果词名没命中，用若干主题近似兜底 | |
| refs = REFERENCE_DB.get(pick, []) | |
| if not refs: | |
| approx = { | |
| "River":["River Tam（《萤火虫》）"], | |
| "Willow":["Willow Rosenberg（《吸血鬼猎人巴菲》）"], | |
| "Raven":["Raven（《少年泰坦》）"], | |
| "Echo":["Echo / Maya Lopez（漫威）"], | |
| "Nova":["Nova（漫威）》"], | |
| "Blaze":["Blaze the Cat（《索尼克》）"], | |
| "Fox":["Fox Mulder（《X档案》）"], | |
| "Atlas":["Atlas（希腊神话）"], | |
| "Orion":["Orion（神话/星座）"], | |
| "Comet":["Comet（常见角色/代号）"], | |
| } | |
| refs = approx.get(pick, []) | |
| if not refs: | |
| refs = ["未找到特别典型的文学/影视角色"] | |
| return {"name": pick, "meaning": meaning, "references": refs, "kind": kind} | |
| # —— LLM生成（仍保留，但确保和主题相关；失败则用 themed_pick） —— | |
| def llm_one(cn_name: str, gender: str, seed: int = None): | |
| if not USE_LLM or tokenizer is None or model is None: | |
| return themed_pick(cn_name, gender, seed=seed) | |
| themes = themes_from_cn(cn_name)[:3] | |
| sys_prompt = ( | |
| "你是英文起名专家。给用户生成1个名字，它可以是：\n" | |
| "A) 常规英文名（Given name，单词），需符合指定性别；或\n" | |
| "B) 一个英文单词式的“词名”（如 Nova, Raven, River, Echo, Blaze 等，单词）。\n" | |
| "要求：\n" | |
| "1) 名字必须与中文名的意象/主题“稍有关联”，主题如下：" + ", ".join(themes) + "；\n" | |
| "2) 名字必须是一个单词；禁止中文拼音/音译（Li, Ming, Xue 等）；\n" | |
| "3) 若选择常规英文名，需符合性别（male/female）；\n" | |
| "4) 输出JSON对象：{\"name\":..., \"meaning\":..., \"references\":[...], \"kind\":\"given|word\"}；\n" | |
| " meaning 用中文，简述与哪个汉字/主题有关；references 给2-4个文学/影视/名人同名（若无写“未找到特别典型的文学/影视角色”）。" | |
| ) | |
| user = {"cn_name": cn_name, "gender": gender} | |
| prompt = f"<|im_start|>system\n{sys_prompt}\n<|im_end|>\n<|im_start|>user\n{json.dumps(user, ensure_ascii=False)}\n<|im_end|>\n<|im_start|>assistant\n" | |
| if seed and hasattr(torch, "manual_seed"): | |
| torch.manual_seed(seed) | |
| def try_once(): | |
| inputs = tokenizer(prompt, return_tensors="pt") | |
| with torch.no_grad(): | |
| out = model.generate( | |
| **inputs, | |
| max_new_tokens=320, | |
| do_sample=True, | |
| top_p=0.9, | |
| temperature=0.9, | |
| eos_token_id=tokenizer.eos_token_id | |
| ) | |
| text = tokenizer.decode(out[0], skip_special_tokens=True) | |
| m = re.findall(r"\{.*\}", text, flags=re.S) | |
| if not m: | |
| return None | |
| try: | |
| obj = json.loads(m[-1]) | |
| name = (obj.get("name") or "").strip() | |
| kind = (obj.get("kind") or "").strip().lower() | |
| meaning = (obj.get("meaning") or "").strip() | |
| refs = obj.get("references") or [] | |
| if not name_sanity(name): | |
| return None | |
| if kind != "word": # 常规名要性别匹配 | |
| if gender=="male" and name not in MALE_SET: return None | |
| if gender=="female" and name not in FEMALE_SET: return None | |
| kind = "given" | |
| # 引用兜底 | |
| if not isinstance(refs, list) or not refs: | |
| refs = REFERENCE_DB.get(name, []) or ["未找到特别典型的文学/影视角色"] | |
| if not meaning: | |
| meaning = render_meaning(cn_name, themes[:1] or ["general"], name) | |
| return {"name": name, "meaning": meaning, "references": refs[:4], "kind": kind} | |
| except Exception: | |
| return None | |
| for _ in range(4): | |
| r = try_once() | |
| if r: | |
| return r | |
| if hasattr(torch, "manual_seed"): | |
| torch.manual_seed(random.randint(1, 10_000_000)) | |
| return themed_pick(cn_name, gender, seed=seed) | |
| # ======== 页面交互 ========= | |
| def do_generate(cn_name, gender, seed): | |
| cn_name = (cn_name or "").strip() | |
| if not cn_name: | |
| return gr.update(visible=False), gr.update(visible=True), "请输入中文名字。", "", "—" | |
| if any(not ('\u4e00' <= ch <= '\u9fff') for ch in cn_name): | |
| return gr.update(visible=False), gr.update(visible=True), "请输入**中文**名字（例如：李明 / 王静）。", "", "—" | |
| if gender not in ("male","female"): | |
| gender = "male" | |
| res = llm_one(cn_name, gender, seed=seed) | |
| name, meaning, refs = res["name"], res["meaning"], res["references"] | |
| refs_md = "、".join(refs) | |
| title = f"🎉 你的英文名字：**{name}**" | |
| return gr.update(visible=False), gr.update(visible=True), title, f"含义/灵感：{meaning}", f"同名角色：{refs_md}" | |
| def go_home(): | |
| return gr.update(visible=True), gr.update(visible=False), "", "", "" | |
| with gr.Blocks(title="给自己起一个英文名字", fill_height=True) as demo: | |
| seed_state = gr.State(value=random.randint(1, 10_000_000)) | |
| # 首页 | |
| gr.Markdown("## 给自己起一个英文名字") | |
| with gr.Group(visible=True) as home: | |
| gr.Markdown("输入中文名与性别，然后点击 **起名** 👇") | |
| with gr.Row(): | |
| cn_name = gr.Textbox(label="中文名字", placeholder="例如：李明 / 王静 / 陈宇", lines=1) | |
| with gr.Row(): | |
| gender = gr.Radio(choices=["male","female"], value="male", label="性别") | |
| btn_go = gr.Button("起名", variant="primary") | |
| # 结果页 | |
| with gr.Group(visible=False) as result: | |
| result_title = gr.Markdown("") | |
| result_meaning = gr.Markdown("") | |
| result_refs = gr.Markdown("") | |
| with gr.Row(): | |
| btn_shuffle = gr.Button("换名字") | |
| btn_back = gr.Button("返回首页", variant="secondary") | |
| # 绑定事件 | |
| btn_go.click( | |
| do_generate, | |
| inputs=[cn_name, gender, seed_state], | |
| outputs=[home, result, result_title, result_meaning, result_refs] | |
| ) | |
| def on_shuffle(cn, g, sd): | |
| new_seed = (sd or 0) + random.randint(7, 9999) | |
| return (*do_generate(cn, g, new_seed),) | |
| btn_shuffle.click( | |
| on_shuffle, | |
| inputs=[cn_name, gender, seed_state], | |
| outputs=[home, result, result_title, result_meaning, result_refs] | |
| ) | |
| btn_back.click( | |
| go_home, | |
| inputs=[], | |
| outputs=[home, result, result_title, result_meaning, result_refs] | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() |