Update app.py

app.py

@@ -81,6 +81,17 @@ def process_text(prompt):
         print("prompt: \n", prompt_trans)
         return prompt_trans
 
+# 定义一个函数来处理图片尺寸的选择
+def set_size(option):
+    if option == "横版":
+        width, height = 1024, 736
+    elif option == "竖版":
+        width, height = 736, 1024
+    elif option == "方形":
+        width, height = 736, 736
+    else:
+        width, height = 736, 1024  # 可以定义一个默认值
+    return width, height
 
 @spaces.GPU
 def generate(
@@ -119,8 +130,8 @@ def generate(
         return pipe(
             prompt=process_text(prompt),
             negative_prompt=process_text(negative_prompt),
-            prompt_2=prompt_2,
-            negative_prompt_2=negative_prompt_2,
+            prompt_2=process_text(prompt_2),
+            negative_prompt_2=process_text(negative_prompt_2),
             width=width,
             height=height,
             guidance_scale=guidance_scale_base,
@@ -132,8 +143,8 @@ def generate(
         latents = pipe(
             prompt=process_text(prompt),
             negative_prompt=process_text(negative_prompt),
-            prompt_2=prompt_2,
-            negative_prompt_2=negative_prompt_2,
+            prompt_2=process_text(prompt_2),
+            negative_prompt_2=process_text(negative_prompt_2),
             width=width,
             height=height,
             guidance_scale=guidance_scale_base,
@@ -144,8 +155,8 @@ def generate(
         image = refiner(
             prompt=process_text(prompt),
             negative_prompt=process_text(negative_prompt),
-            prompt_2=prompt_2,
-            negative_prompt_2=negative_prompt_2,
+            prompt_2=process_text(prompt_2),
+            negative_prompt_2=process_text(negative_prompt_2),
             guidance_scale=guidance_scale_refiner,
             num_inference_steps=num_inference_steps_refiner,
             image=latents,
@@ -177,6 +188,8 @@ with gr.Blocks(css="style.css") as demo:
             )
             run_button = gr.Button("生成", scale=0)
         result = gr.Image(label="生成结果", show_label=False)
+
+        
     with gr.Accordion("高级选项", open=False):
         with gr.Row():
             use_negative_prompt = gr.Checkbox(label="使用反向提示词", value=False)
@@ -209,22 +222,28 @@ with gr.Blocks(css="style.css") as demo:
             value=0,
         )
         randomize_seed = gr.Checkbox(label="随机种子数", value=True)
-        with gr.Row():
-            width = gr.Slider(
-                label="宽",
-                minimum=256,
-                maximum=MAX_IMAGE_SIZE,
-                step=32,
-                value=736,
-            )
-            height = gr.Slider(
-                label="高",
-                minimum=256,
-                maximum=MAX_IMAGE_SIZE,
-                step=32,
-                value=1024,
-            )
+
+        # 使用 Radio 组件替代两个 Slider 组件
+        size_option = gr.Radio(choices=["横版", "竖版", "方形"], label="选择尺寸", value="竖版")
+        
+        # with gr.Row():
+        #     width = gr.Slider(
+        #         label="宽度",
+        #         minimum=256,
+        #         maximum=MAX_IMAGE_SIZE,
+        #         step=32,
+        #         value=736,
+        #     )
+        #     height = gr.Slider(
+        #         label="高度",
+        #         minimum=256,
+        #         maximum=MAX_IMAGE_SIZE,
+        #         step=32,
+        #         value=1024,
+        #     )
+        
         apply_refiner = gr.Checkbox(label="增加精炼模型（refiner）", value=False, visible=ENABLE_REFINER)
+        
         with gr.Row():
             guidance_scale_base = gr.Slider(
                 label="提示词相关性",
@@ -256,7 +275,11 @@ with gr.Blocks(css="style.css") as demo:
                 value=25,
             )
 
+    # 使用 size_option 的选择结果来动态设置 width 和 height 的值
+    size_option.change(fn=set_size, inputs=size_option, outputs=[width, height])
+    
     gr.Examples(
+        label="例子"
         examples=examples,
         inputs=prompt,
         outputs=result,