import os, html, random, re import gradio as gr from openai import OpenAI # ===== 基础配置 ===== API_KEY = os.getenv("OPENAI_API_KEY") MODEL_NAME = "gpt-4o-mini" client = OpenAI(api_key=API_KEY) # ===== 系统提示：短、好念、可自创、真实名言 ===== SYSTEM_PROMPT = """You are a playful, high-creativity pet name generator. Goal: Produce EXACTLY ONE short, cute, kid-pronounceable pet name + one authentic quote line. NAME rules (STRICT): - The name must be VERY short: • Chinese/Japanese/Korean: 1–2 characters/syllables (e.g., “豆豆”, “Momo”, “Bori”). • English/French: 1–2 syllables only (e.g., “Pip”, “Lulu”, “Nori”, “Chou”). - You MAY invent new words if they sound natural and easy. - NO numbers, NO hyphens, NO long phrases, NO public-figure names. - Output only ONE name. - Add a micro-meaning (≤5 words) after the name. QUOTE rules (must be real): - After the name line, provide ONE authentic, warm animal-related quote with attribution. - Keep the quote short (≤20 words). Attribution required (e.g., — George Eliot). - Translate the quote text into the requested language if needed; attribution kept. STRICT OUTPUT FORMAT (2 lines only): Name — short meaning Quote: “...” — Attribution """ def build_prompt(species, look, personality, gender, language, inspiration): return f""" Requested language: {language} Inspiration source (primary): {inspiration} Refinements (secondary only, may be empty): - Species: {species or '(unspecified)'} - Appearance: {look or '(unspecified)'} - Personality: {personality or '(unspecified)'} - Gender: {gender} Task: 1) Create ONE ultra-short, cute, kid-pronounceable pet name (length limits above). 2) Add a micro meaning (≤5 words). 3) Output ONE authentic animal-related quote with attribution. Remember: you may invent new names; do NOT include any digits or hyphens in the name. """ # ==== 名言兜底 ==== FALLBACK_QUOTES = { "English": [ ('"Until one has loved an animal, a part of one\'s soul remains unawakened."', "Anatole France"), ('"An animal\'s eyes have the power to speak a great language."', "Martin Buber"), ('"Animals are such agreeable friends—they ask no questions; they pass no criticisms."', "George Eliot"), ], "中文": [ ('“一个人若未曾爱过动物，他的灵魂尚未苏醒。”', "Anatole France"), ('“动物的眼睛自有它们的语言。”', "Martin Buber"), ('“动物是极好的朋友——它们不发问，也不挑剔。”', "George Eliot"), ], "日本語": [ ('「動物を愛したことのない者の魂は、まだ目覚めていない。」', "Anatole France"), ('「動物の瞳は偉大な言葉を語る力を持つ。」', "Martin Buber"), ('「動物は実に快い友である——問いも批判もない。」', "George Eliot"), ], "한국어": [ ('“동물을 사랑해 보지 않은 영혼은 아직 깨어나지 않았다.”', "Anatole France"), ('“동물의 눈에는 위대한 언어가 담겨 있다.”', "Martin Buber"), ('“동물은 참으로 상냥한 친구다—묻지도 비난하지도 않는다.”', "George Eliot"), ], "Français": [ ("« Jusqu'à ce qu'on ait aimé un animal, une partie de notre âme reste endormie. »", "Anatole France"), ("« Les yeux d’un animal ont le pouvoir de parler un grand langage. »", "Martin Buber"), ("« Les animaux sont de si agréables amis — ils ne posent aucune question et ne font aucune critique. »", "George Eliot"), ], } def _fallback_quote(language: str) -> str: lst = FALLBACK_QUOTES.get(language, FALLBACK_QUOTES["English"]) text, author = random.choice(lst) return f'Quote: {text} — {author}' def _ensure_two_lines_and_strip_digits(content: str, language: str): if not content: return None, None lines = [ln.strip() for ln in content.splitlines() if ln.strip()] name_line = lines[0] if lines else "" quote_line = next((ln for ln in lines[1:] if ln.lower().startswith("quote:")), None) if not quote_line: quote_line = _fallback_quote(language) sep = "—" if "—" in name_line else ("-" if "-" in name_line else None) if sep: parts = name_line.split(sep, 1) name_part = re.sub(r"[0-9\\-]", "", parts[0].strip()) meaning_part = parts[1].strip() if len(parts) > 1 else "" if not name_part: name_part = "Momo" if language in ["English", "Français"] else ("豆豆" if language == "中文" else "モモ") name_line = f"{name_part} — {meaning_part}" if meaning_part else name_part else: cleaned = re.sub(r"[0-9\\-]", "", name_line).strip() if not cleaned: cleaned = "Momo" if language in ["English", "Français"] else ("豆豆" if language == "中文" else "モモ") name_line = cleaned return name_line, quote_line def generate_modal_html(species, look, personality, gender, language, inspiration): try: prompt = build_prompt( (species or "").strip(), (look or "").strip(), (personality or "").strip(), gender, language, inspiration ) resp = client.chat.completions.create( model=MODEL_NAME, temperature=0.95, max_tokens=200, messages=[ {"role": "system", "content": SYSTEM_PROMPT}, {"role": "user", "content": prompt} ] ) raw = (resp.choices[0].message.content or "").strip() name_line, quote_line = _ensure_two_lines_and_strip_digits(raw, language) name_html, quote_html = html.escape(name_line), html.escape(quote_line) return f"""
Pawfect Match
{name_html}
{quote_html}
""" except Exception as e: err = html.escape(str(e)) return f"
⚠️ 出错
{err}
" # ===== 样式与界面 ===== theme = gr.themes.Soft(primary_hue="pink", secondary_hue="cyan", neutral_hue="slate") CUSTOM_CSS = """ :root{--pm-bg:#faf7fb;--pm-surface:#fff;--pm-border:#eddcf3;--pm-text:#111;--pm-muted:#6b7280;--pm-accent:#ff4fa3;} body,.gradio-container{background:var(--pm-bg)!important;} .pm-topbar{display:flex;align-items:center;justify-content:space-between;padding:12px;background:linear-gradient(90deg,#ffe8f3,#e8fbff);border-radius:12px;border:1px solid var(--pm-border);} .pm-brand{display:flex;align-items:center;gap:10px;} .pm-logo{font-size:24px;} .pm-title{font-weight:800;font-size:20px;color:var(--pm-text);} .pm-overlay{position:fixed;inset:0;background:rgba(0,0,0,.35);display:flex;align-items:center;justify-content:center;z-index:9999;} .pm-modal{width:min(560px,92vw);background:var(--pm-surface);border:1px solid var(--pm-border);border-radius:14px;box-shadow:0 20px 60px rgba(0,0,0,.18);} .pm-modal-title{padding:14px 16px;font-weight:800;border-bottom:1px solid var(--pm-border);} .pm-modal-body{padding:16px;} .pm-name{font-weight:800;font-size:22px;margin-bottom:8px;color:#111;} .pm-quote{font-weight:600;font-size:14px;color:#374151;opacity:.95;} .pm-modal-actions{padding:12px 16px;display:flex;justify-content:flex-end;border-top:1px solid var(--pm-border);} .pm-btn{padding:10px 14px;border-radius:10px;border:1px solid var(--pm-border);background:var(--pm-surface);cursor:pointer;font-weight:700;} .pm-btn:hover{border-color:var(--pm-accent);color:var(--pm-accent);} """ with gr.Blocks(title="Pawfect Match", theme=theme, css=CUSTOM_CSS) as demo: gr.HTML("""
Pawfect Match
""") with gr.Group(): species = gr.Textbox(label="宠物种类") look = gr.Textbox(label="外表特征") personality = gr.Textbox(label="性格特点") with gr.Row(): gender = gr.Radio(["male","female"], value="male", label="宠物性别") language = gr.Radio(["中文","English","日本語","한국어","Français"], value="English", label="输出语言") inspiration = gr.Radio(["山海经","迪士尼","哈利波特","生活"], value="生活", label="灵感来源") with gr.Row(): btn = gr.Button("✨ 起名", variant="primary") clear_btn = gr.Button("清空", variant="secondary") modal = gr.HTML(visible=True, value="") btn.click(generate_modal_html, inputs=[species, look, personality, gender, language, inspiration], outputs=modal) clear_btn.click(lambda: ("","","","male","English","生活",""), outputs=[species, look, personality, gender, language, inspiration, modal]) if __name__ == "__main__": demo.launch(ssr_mode=False, share=True)