Update app.py

app.py

@@ -1,4 +1,4 @@
-import os, html
+import os, html, random
 import gradio as gr
 from openai import OpenAI
 
@@ -6,28 +6,34 @@ API_KEY = os.getenv("OPENAI_API_KEY")
 MODEL_NAME = "gpt-4o-mini"
 client = OpenAI(api_key=API_KEY)
 
-# --- 系统提示，强化“可爱名字 + 真实温馨名言” ---
-SYSTEM_PROMPT = """You are a playful yet creative pet name generator.
-Goal: Produce EXACTLY ONE cute, pet-friendly name + one authentic quote line.
+# ===== 可爱名字 + 真实名言：系统提示 =====
+SYSTEM_PROMPT = """You are a playful, high-creativity pet name generator.
+Goal: Produce EXACTLY ONE cute, kid-pronounceable pet name + one authentic quote line.
 
 Creative balance:
-- Inspiration source (山海经 / 迪士尼 / 哈利波特 / 生活) drives the theme (~70%).
-- User inputs (species / appearance / personality / gender) refine the vibe (~30%).
-
-NAME rules:
-- Must feel CUTE and easy to call out loud (pet style).
-- Prefer 2–3 syllables; round and soft sounds are good.
-- Draw imagery from the inspiration source, but keep tone playful and adorable.
-- Output in the requested language (Chinese/English/Japanese/Korean/French).
-- Provide ONE short meaning/explanation (≤12 words).
-
-QUOTE rules:
-- After the name line, provide ONE real existing quote about animals, love, companionship, or playfulness.
-- The quote must be authentic (from literature, films, proverbs, or famous people).
-- Keep it short, warm, and not too formal.
-- Translate the quote into the requested language if needed, attribution kept.
-
-STRICT OUTPUT: exactly two lines
+- Inspiration source (山海经 / 迪士尼 / 哈利波特 / 生活) drives the concept (~70%).
+- User inputs (species/appearance/personality/gender) refine the vibe (~30%).
+- Keep outputs imaginative; do NOT overfit to traits.
+
+NAME rules (must be easy for kids):
+- Keep it cute, round, and easy to call.
+- Length guidance by language:
+  • Chinese: 1–2 simple characters (avoid rare characters).
+  • English: 1–2 syllables (e.g., “Momo”, “Pip”, “Nori”).
+  • Japanese: 2–3 mora, open vowels preferred (e.g., “Momo”, “Koko”).
+  • Korean: 2 simple syllables (e.g., “Bori”, “Toto”).
+  • French: 1–2 syllables with soft endings (-ou, -i, -ette).
+- You MAY coin a new word if it sounds cute and natural.
+- Output in requested language (Chinese/English/Japanese/Korean/French).
+- Provide ONE short meaning/explanation (≤12 words), theme-first, traits-second.
+- No numbers; no long phrases.
+
+QUOTE rules (must be real):
+- After the name line, provide ONE short, warm, authentic quote about animals/love/companionship/playfulness.
+- Use a real, verifiable author/work/proverb; include attribution (e.g., — Anatole France).
+- Translate the quote text into the requested language if needed, attribution kept in Latin script.
+
+STRICT OUTPUT (exactly two lines):
 Name — short meaning
 Quote: “...” — Attribution
 """
@@ -35,18 +41,73 @@ Quote: “...” — Attribution
 def build_prompt(species, look, personality, gender, language, inspiration):
     return f"""
 Requested language: {language}
-Inspiration source (primary): {inspiration}
-Refinement inputs (secondary):
+Primary inspiration source: {inspiration}
+Secondary refinements (may be empty):
 - Species: {species or '(unspecified)'}
 - Appearance: {look or '(unspecified)'}
 - Personality: {personality or '(unspecified)'}
 - Gender: {gender}
 
 Task:
-1) Create EXACTLY ONE cute pet name (≤12-word meaning).
-2) Then add ONE authentic warm quote about animals/companionship/playfulness.
+1) Create EXACTLY ONE cute, kid-pronounceable pet name (≤12-word meaning).
+2) Then output ONE authentic, warm animal-related quote with attribution.
+Remember: inspiration first (~70%), traits second (~30%); keep both lines in {language}.
 """
 
+# ===== 真实名言兜底（多语言） =====
+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 get_fallback_quote(language: str):
+    lang = language if language in FALLBACK_QUOTES else "English"
+    text, author = random.choice(FALLBACK_QUOTES[lang])
+    return f'Quote: {text} — {author}'
+
+def ensure_two_lines_and_quote(content: str, language: str):
+    """
+    确保严格两行输出；若缺失或格式不当，为第二行注入真实名言兜底。
+    """
+    if not content:
+        return None
+    lines = [ln.strip() for ln in content.splitlines() if ln.strip()]
+    # 尽量抓取首行作为名字
+    name_line = lines[0] if lines else ""
+    # 找到包含 Quote: 的行
+    quote_line = None
+    for ln in lines[1:]:
+        if ln.lower().startswith("quote:"):
+            quote_line = ln
+            break
+    if not quote_line:
+        quote_line = get_fallback_quote(language)
+    # 只返回两行
+    return f"{name_line}\n{quote_line}"
+
 def generate_name(species, look, personality, gender, language, inspiration):
     try:
         prompt = build_prompt(
@@ -57,7 +118,7 @@ def generate_name(species, look, personality, gender, language, inspiration):
         )
         resp = client.chat.completions.create(
             model=MODEL_NAME,
-            temperature=0.9,  # 提高创造力
+            temperature=0.95,   # 强创意
             top_p=0.95,
             max_tokens=280,
             messages=[
@@ -65,13 +126,15 @@ def generate_name(species, look, personality, gender, language, inspiration):
                 {"role": "user",   "content": prompt}
             ]
         )
-        return resp.choices[0].message.content.strip()
+        raw = (resp.choices[0].message.content or "").strip()
+        final_out = ensure_two_lines_and_quote(raw, language)
+        return final_out or "（没有生成内容，请稍后再试）"
     except Exception as e:
         return f"⚠️ 出错：{html.escape(str(e))}"
 
-# --- Gradio 界面 ---
+# ===== 界面（去掉那句介绍，只留标题） =====
 with gr.Blocks(title="Pawfect Match") as demo:
-    gr.Markdown("## 🐾 Pawfect Match｜宠物起名机\n选择灵感来源 + 输入信息 → 一键生成一个可爱的名字 + 一句真实名言。")
+    gr.Markdown("## 🐾 Pawfect Match｜宠物起名机")
 
     species = gr.Textbox(label="宠物种类", placeholder="例：金毛 / 英短 / 仓鼠 / 鹦鹉（可留空）")
     look = gr.Textbox(label="外表特征", placeholder="例：橘色短毛，大眼睛，白爪（可留空）")