Update app.py

app.py

@@ -1,44 +1,41 @@
-# app.py
-
 import gradio as gr
 from PIL import Image
 from generate_image import generate_ghibli_avatar
-from prompt_engineering import interpret_command
 
-# 初始状态记录
+# 会话状态记录
 session_prompt = ""
 latest_avatar = None
 
+# 分析情绪与年龄（占位函数）
 def analyze_emotion_age(img: Image.Image) -> str:
-    # 占位描述（可替换成真实年龄/情绪识别）
     return "微笑的18岁青年"
 
-def translate_description_to_prompt(description):
-    # 简单手动映射（你可以用 OpenCC、百度翻译API 或 LLM 自动翻译）
-    return "smiling 18-year-old boy"  # 替换为真实翻译后的内容
-
-def generate_initial_avatar(image, description):
+# 初次生成头像（仅基于描述）
+def generate_initial_avatar(image):
     global session_prompt, latest_avatar
-    translated_prompt = translate_description_to_prompt(description)
-    session_prompt = f"A portrait of a {translated_prompt}, ghibli style, anime, headshot, studio ghibli"
-    avatar = generate_ghibli_avatar(session_prompt, reference_image=image)
+    description = analyze_emotion_age(image)
+    session_prompt = description
+    avatar = generate_ghibli_avatar(session_prompt)
     latest_avatar = avatar
-    return avatar
+    return description, avatar
 
+# 用户输入修改指令（中文）后更新头像
 def apply_user_command(command):
     global session_prompt, latest_avatar
-    if latest_avatar is None:
-        return None, "请先上传图片并生成初始头像"
-    session_prompt = interpret_command(command, session_prompt)
+    if not latest_avatar:
+        return None, "请先上传照片并生成初始头像！"
+    session_prompt += f"，{command}"
     avatar = generate_ghibli_avatar(session_prompt)
     latest_avatar = avatar
-    return avatar, f"提示词已更新为：{session_prompt}"
+    return avatar, f"已更新描述为：{session_prompt}"
 
+# 完成
 def finalize():
-    return "感谢使用！希望你喜欢这个吉卜力头像 😊"
+    return "感谢使用吉卜力头像生成器！希望你喜欢～"
 
+# 构建 Gradio 界面
 with gr.Blocks(title="吉卜力头像生成器") as demo:
-    gr.Markdown("# 🎨 吉卜力头像生成器\n上传照片，生成你的动漫形象，并用中文指令调整细节！")
+    gr.Markdown("## 🎨 吉卜力头像生成器\n上传照片，生成你的动漫形象，并用中文指令调整细节！")
 
     with gr.Row():
         image_input = gr.Image(type="pil", label="上传照片（仅用于分析年龄与情绪）")
@@ -49,24 +46,17 @@ with gr.Blocks(title="吉卜力头像生成器") as demo:
         avatar_output = gr.Image(label="生成的头像")
 
     with gr.Row():
-        user_command = gr.Textbox(label="输入你的修改指令（如：给我加一个红色头发）")
+        user_command = gr.Textbox(label="输入你的修改指令（例如：换一个红色头发）")
         command_btn = gr.Button("应用指令")
         feedback = gr.Textbox(label="系统反馈", interactive=False)
 
     final_btn = gr.Button("我满意啦！")
 
-    gen_btn.click(
-        fn=generate_initial_avatar,
-        inputs=image_input,
-        outputs=[description_output, avatar_output]
-    )
-
-    command_btn.click(
-        fn=apply_user_command,
-        inputs=user_command,
-        outputs=[avatar_output, feedback]
-    )
-
+    # 绑定逻辑
+    gen_btn.click(fn=generate_initial_avatar, inputs=image_input, outputs=[description_output, avatar_output])
+    command_btn.click(fn=apply_user_command, inputs=user_command, outputs=[avatar_output, feedback])
     final_btn.click(fn=finalize, outputs=feedback)
 
-demo.launch()
+# 启动应用（仅用于本地调试；Spaces 会自动运行）
+if __name__ == "__main__":
+    demo.launch()