add video resolution

app.py

@@ -43,6 +43,7 @@ max_64_bit_int = 2**63 - 1
 def sample(
     # image: Image,
     prompt,
+    resolution,
     seed: Optional[int] = 42,
     # randomize_seed: bool = True,
     # motion_bucket_id: int = 127,
@@ -64,14 +65,23 @@ def sample(
     os.makedirs(output_folder, exist_ok=True)
     base_count = len(glob(os.path.join(output_folder, "*.mp4")))
     video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
+    res_variants = {
+            '1:1': '512x512',
+            '9:16': '384x672',
+            '16:9': '672x384',
+            '1:2': '352x736',
+            '2:1': '736x352'
+    }
+    width = int(res_variants[resolution].split('x')[0])
+    height = int(res_variants[resolution].split('x')[1])
 
     # frames = pipe(image, decode_chunk_size=decoding_t, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=0.1, num_frames=25).frames[0]
     # prompt = "The camera follows behind a white vintage SUV with a black roof rack as it speeds up a steep dirt road surrounded by pine trees on a steep mountain slope, dust kicks up from it’s tires, the sunlight shines on the SUV as it speeds along the dirt road, casting a warm glow over the scene. The dirt road curves gently into the distance, with no other cars or vehicles in sight. The trees on either side of the road are redwoods, with patches of greenery scattered throughout. The car is seen from the rear following the curve with ease, making it seem as if it is on a rugged drive through the rugged terrain. The dirt road itself is surrounded by steep hills and mountains, with a clear blue sky above with wispy clouds."
     frames = pipe(
         seed=seed,
         time_length=12,
-        width = 672,
-        height = 384,
+        width = width,
+        height = height,
         save_path=video_path,
         text=prompt,
     )
@@ -124,6 +134,13 @@ with gr.Blocks() as demo:
             placeholder="Enter your prompt",
             container=False,
         )
+        #TODO ? можно так принять входное значение 
+        resolution = gr.Dropdown(
+            label="Video resolution",
+            choices=["1:1", "9:16", "16:9", "1:2", "2:1"],
+            value="16:9"
+        )
+
         generate_btn = gr.Button("Generate")
 
   # with gr.Accordion("Advanced options", open=False):
@@ -133,7 +150,8 @@ with gr.Blocks() as demo:
   #     fps_id = gr.Slider(label="Frames per second", info="The length of your video in seconds will be 25/fps", value=6, minimum=5, maximum=30)
 
   # image.upload(fn=resize_image, inputs=image, outputs=image, queue=False)
-  generate_btn.click(fn=sample, inputs=[prompt], outputs=[video], api_name="video")
+  #TODO функция так принимает аргументы?
+  generate_btn.click(fn=sample, inputs=[prompt, resolution], outputs=[video], api_name="video")
   # gr.Examples(
   #   examples=[
   #       "images/blink_meme.png",