Update app.py

app.py

@@ -125,6 +125,14 @@ def resize_image(image, output_size=(672, 384)):
     return cropped_image
 
 with gr.Blocks() as demo:
+  aspect_ratio_variants = {
+        '16:9 (672x384)': '16:9', 
+        '9:16 (384x672)': '9:16', 
+        '1:1 (512x512)': '1:1', 
+        '1:2 (352x736)': '1:2', 
+        '2:1 (736x352)': '2:1'
+  }
+    
   gr.Markdown('''# Community demo for Kandinsky 4.0 Flash''')
   with gr.Row():
     with gr.Column():
@@ -144,14 +152,15 @@ with gr.Blocks() as demo:
             # gr.Markdown(
             #     "✨Upon pressing the enhanced prompt button, we will use [GigaChat Model](https://github.com/THUDM/GLM-4) to polish the prompt and overwrite the original one."
             # )
-            resolution = gr.Dropdown(
+            aspect_ratio = gr.Dropdown(
                 label="Aspect ratio",
-                choices=["1:1", "9:16", "16:9", "1:2", "2:1"],
-                value="16:9"
+                choices=["16:9 (672x384)", "9:16 (384x672)", "1:1 (512x512)", "1:2 (352x736)", "2:1 (736x352)"],
+                value="16:9 (672x384)"
             )
+            aspect_ratio = aspect_ratio_variants[aspect_ratio]
             with gr.Column():
                 enhance_button = gr.Button("Beautify prompt with GigaChat-Max")
-                generate_btn = gr.Button("Generate")
+                generate_btn = gr.Button("Generate Video")
 
   # with gr.Accordion("Advanced options", open=False):
   #     seed = gr.Slider(label="Seed", value=42, randomize=True, minimum=0, maximum=max_64_bit_int, step=1)
@@ -199,7 +208,7 @@ with gr.Blocks() as demo:
     return prompt
 
 
-  generate_btn.click(fn=sample, inputs=[prompt, resolution], outputs=[video], api_name="video")
+  generate_btn.click(fn=sample, inputs=[prompt, aspect_ratio], outputs=[video], api_name="video")