Update app.py

app.py

@@ -81,21 +81,10 @@ 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(
     prompt: str,
+    size_option: str = "竖版",
     negative_prompt: str = "",
     prompt_2: str = "",
     negative_prompt_2: str = "",
@@ -120,6 +109,16 @@ def generate(
     if not use_negative_prompt_2:
         negative_prompt_2 = None  # type: ignore
 
+    
+    if size_option == "横版":
+        width, height = 1024, 736
+    elif size_option == "竖版":
+        width, height = 736, 1024
+    elif size_option == "方形":
+        width, height = 736, 736
+    else:
+        width, height = 736, 1024  # 可以定义一个默认值
+        
     # process_text("里面做一个测试")
     # print("prompt是：", prompt)
     # print("negative_prompt是：", negative_prompt)
@@ -189,6 +188,8 @@ with gr.Blocks(css="style.css") as demo:
             run_button = gr.Button("生成", scale=0)
         result = gr.Image(label="生成结果", show_label=False)
 
+        # 使用 Radio 组件替代两个 Slider 组件
+        size_option = gr.Radio(choices=["横版", "竖版", "方形"], label="选择尺寸", value="竖版")
         
     with gr.Accordion("高级选项", open=False):
         with gr.Row():
@@ -222,9 +223,6 @@ with gr.Blocks(css="style.css") as demo:
             value=0,
         )
         randomize_seed = gr.Checkbox(label="随机种子数", value=True)
-
-        # 使用 Radio 组件替代两个 Slider 组件
-        size_option = gr.Radio(choices=["横版", "竖版", "方形"], label="选择尺寸", value="竖版")
         
         # with gr.Row():
         #     width = gr.Slider(
@@ -333,6 +331,7 @@ with gr.Blocks(css="style.css") as demo:
         fn=generate,
         inputs=[
             prompt,
+            size_option,
             negative_prompt,
             prompt_2,
             negative_prompt_2,