Update app.py

app.py

@@ -8,11 +8,12 @@ MODEL_NAME = "gpt-4o-mini"
 client = OpenAI(api_key=API_KEY)
 
 SYSTEM_PROMPT = """You are a creative pet name generator.
-Given species, appearance and personality, propose exactly ONE concise, pronounceable name.
+Given species, appearance, personality and gender, propose exactly ONE concise, pronounceable name.
 Rules:
 - Output exactly one name with a short explanation.
 - Avoid offensive or trademarked terms.
 - Prefer 2–3 syllables.
+- Consider the pet's gender vibe if provided.
 - Provide a short meaning/explanation (≤12 words).
 - If language is Chinese/Russian/Spanish, output in that language, with romanization if helpful.
 - No numbers, no long phrases, no public-figure names.
@@ -20,18 +21,19 @@ Output strictly as:
 Name — short meaning
 """
 
-def build_prompt(species, look, personality, language):
+def build_prompt(species, look, personality, gender, language):
     return f"""
 Target language: {language}
 Species: {species}
 Appearance: {look}
 Personality: {personality}
+Gender: {gender}
 Generate exactly ONE creative pet name.
 """
 
-def generate_name(species, look, personality, language):
+def generate_name(species, look, personality, gender, language):
     try:
-        prompt = build_prompt(species, look, personality, language)
+        prompt = build_prompt(species, look, personality, gender, language)
         resp = client.chat.completions.create(
             model=MODEL_NAME,
             temperature=0.7,
@@ -45,18 +47,19 @@ def generate_name(species, look, personality, language):
     except Exception as e:
         return f"⚠️ 出错：{html.escape(str(e))}"
 
-with gr.Blocks(title="PetName Genie") as demo:
-    gr.Markdown("## 🐾 PetName Genie｜宠物起名机\n输入宠物信息，一键生成一个名字 + 简短含义。")
+with gr.Blocks(title="Pawfect Match") as demo:
+    gr.Markdown("## 🐾 Pawfect Match｜宠物起名机\n输入宠物信息，一键生成一个名字 + 简短含义。")
 
     species = gr.Textbox(label="宠物种类", placeholder="例：金毛 / 英短 / 仓鼠 / 鹦鹉")
     look = gr.Textbox(label="外表特征", placeholder="例：橘色短毛，大眼睛，白爪")
     personality = gr.Textbox(label="性格特点", placeholder="例：粘人、活泼、聪明")
+    gender = gr.Radio(["unknown","male","female"], value="unknown", label="宠物性别")
     language = gr.Radio(["English","中文","Русский","Español"], value="English", label="输出语言")
 
     btn = gr.Button("✨ 起名！")
     out = gr.Markdown()
 
-    btn.click(generate_name, inputs=[species, look, personality, language], outputs=out)
+    btn.click(generate_name, inputs=[species, look, personality, gender, language], outputs=out)
 
 if __name__ == "__main__":
     demo.launch()