Update app.py

app.py

@@ -1,70 +1,75 @@
-# 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:
-    # 目前使用占位，未来可用 insightface 替代
-    return "微笑的18岁青年"
-
-def generate_initial_avatar(image, description):
-    global session_prompt, latest_avatar
-    session_prompt = f"{description}，风格为吉卜力风格人物头像"
-    avatar = generate_ghibli_avatar(session_prompt, reference_image=image)
-    latest_avatar = avatar
-    return avatar
-
-def apply_user_command(command):
-    global session_prompt, latest_avatar
-    if not latest_avatar:
-        return None, "请先上传图片并生成初始头像"
-
-    new_prompt = interpret_command(command, session_prompt)
-    session_prompt = new_prompt
-    avatar = generate_ghibli_avatar(new_prompt, reference_image=None)
-    latest_avatar = avatar
-    return avatar, f"提示词已更新为：{new_prompt}"
-
-def finalize():
-    return "感谢使用！希望你喜欢这个吉卜力头像 😊"
-
-with gr.Blocks(title="吉卜力头像生成器") as demo:
-    gr.Markdown("# 🎨 吉卜力头像生成器\n上传照片，生成你的动漫形象，并用中文指令调整细节！")
-
-    with gr.Row():
-        image_input = gr.Image(type="pil", label="上传照片或拍照")
-        description_output = gr.Textbox(label="分析结果", interactive=False)
-
-    with gr.Row():
-        gen_btn = gr.Button("生成初始头像")
-        avatar_output = gr.Image(label="生成的头像")
-
-    with gr.Row():
-        user_command = gr.Textbox(label="输入你的修改指令（如：给我加一个红色头发）")
-        command_btn = gr.Button("应用指令")
-        feedback = gr.Textbox(label="系统反馈", interactive=False)
-
-    final_btn = gr.Button("我满意啦！")
-
-    # 功能绑定
-    gen_btn.click(
-        fn=lambda img: (analyze_emotion_age(img), generate_initial_avatar(img, analyze_emotion_age(img))),
-        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()
+# 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:
+    # 目前使用占位，未来可用 insightface 替代
+    return "微笑的18岁青年"
+
+def generate_initial_avatar(image, description):
+    global session_prompt, latest_avatar
+    session_prompt = f"{description}，风格为吉卜力风格人物头像"
+    avatar = generate_ghibli_avatar(session_prompt, reference_image=image)
+    latest_avatar = avatar
+    return avatar
+
+def apply_user_command(command):
+    global session_prompt, latest_avatar
+    if not latest_avatar:
+        return None, "请先上传图片并生成初始头像"
+
+    new_prompt = interpret_command(command, session_prompt)
+    session_prompt = new_prompt
+    avatar = generate_ghibli_avatar(new_prompt, reference_image=None)
+    latest_avatar = avatar
+    return avatar, f"提示词已更新为：{new_prompt}"
+
+def finalize():
+    return "感谢使用！希望你喜欢这个吉卜力头像 😊"
+
+with gr.Blocks(title="吉卜力头像生成器") as demo:
+    gr.Markdown("# 🎨 吉卜力头像生成器\n上传照片，生成你的动漫形象，并用中文指令调整细节！")
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="上传照片或拍照")
+        description_output = gr.Textbox(label="分析结果", interactive=False)
+
+    with gr.Row():
+        gen_btn = gr.Button("生成初始头像")
+        avatar_output = gr.Image(label="生成的头像")
+
+    with gr.Row():
+        user_command = gr.Textbox(label="输入你的修改指令（如：给我加一个红色头发）")
+        command_btn = gr.Button("应用指令")
+        feedback = gr.Textbox(label="系统反馈", interactive=False)
+
+    final_btn = gr.Button("我满意啦！")
+
+    # 功能绑定
+    gen_btn.click(
+        fn=lambda img: (analyze_emotion_age(img), generate_initial_avatar(img, analyze_emotion_age(img))),
+        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()
+# app.py 文件末尾
+if __name__ == "__main__":
+    gr.close_all()
+    demo.launch()
+