Update app.py

app.py

@@ -11,12 +11,12 @@ def generate_artwork(description, style, enhance_details):
         prompt += " with enhanced details"
     
     # 根据用户选择的风格修改提示
-    if style == "Fantasy":
-        prompt += " in a fantasy style"
+    if style == "梦幻":
+        prompt += "以奇幻风格"
     elif style == "Cyberpunk":
-        prompt += " in a cyberpunk style"
+        prompt += "赛博朋克"
     else:
-        prompt += " in a realistic style"
+        prompt += "写实风格"
 
     # 生成图像
     image = generate_image(prompt)
@@ -25,15 +25,15 @@ def generate_artwork(description, style, enhance_details):
 # 创建 Gradio 界面
 with gr.Blocks() as demo:
     with gr.Row():
-        description_input = gr.Textbox(label="Enter a description", placeholder="e.g., A magical treehouse in the sky")
-        style_selector = gr.Dropdown(choices=["Fantasy", "Cyberpunk", "Realistic"], label="Select Style")
-        detail_checkbox = gr.Checkbox(label="Enhance Details")
+        description_input = gr.Textbox(label="输入描述", placeholder="e.g., A magical treehouse in the sky")
+        style_selector = gr.Dropdown(choices=["梦幻", "赛博朋克", "写实"], label="选择风格")
+        detail_checkbox = gr.Checkbox(label="增强细节")
         
     with gr.Row():
-        output_prompt = gr.Textbox(label="Generated Prompt", interactive=False)
-        output_image = gr.Image(label="Generated Image", interactive=False)
+        output_prompt = gr.Textbox(label="生成提示", interactive=False)
+        output_image = gr.Image(label="生成图像", interactive=False)
 
-    generate_button = gr.Button("Generate Artwork")
+    generate_button = gr.Button("生成作品")
     generate_button.click(generate_artwork, 
                           inputs=[description_input, style_selector, detail_checkbox], 
                           outputs=[output_prompt, output_image])