Autoplay the generated video and add a download button (#19)

- Autoplay the generated video and add a download button (a452b5852a3919f3f735c9f3dc2271f15b1aa280)


Co-authored-by: Fabrice TIERCELIN <[email protected]>

app.py

@@ -55,7 +55,7 @@ def sample(
         frames = fps14Pipe(image, decode_chunk_size=decoding_t, generator=generator, motion_bucket_id=motion_bucket_id, noise_aug_strength=noise_aug_strength, num_frames=25).frames[0]
     export_to_video(frames, video_path, fps=fps_id)
     
-    return video_path, gr.update(label="Generated frames in *." + frame_format + " format", format = frame_format, value = frames), seed
+    return video_path, gr.update(value=video_path, visible=True), gr.update(label="Generated frames in *." + frame_format + " format", format = frame_format, value = frames, visible=True), seed
 
 def resize_image(image, output_size=(1024, 576)):
     # Calculate aspect ratios
@@ -105,18 +105,19 @@ with gr.Blocks() as demo:
               noise_aug_strength = gr.Slider(label="Noise strength", info="The noise to add", value=0.1, minimum=0, maximum=1, step=0.1)
               decoding_t = gr.Slider(label="Decoding", info="Number of frames decoded at a time; this eats more VRAM; reduce if necessary", value=3, minimum=1, maximum=5, step=1)
               frame_format = gr.Radio([["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"]], label="Image format for result", info="File extention", value="webp", interactive=True)
-              version = gr.Radio([["Auto", "auto"], ["SVD (trained on 14 f/s)", "svd"], ["SVD-XT (trained on 25 f/s)", "svdxt"]], label="Model", info="Trained model", value="auto", interactive=True)
+              version = gr.Radio([["Auto", "auto"], ["🏃🏻‍♀️ SVD (trained on 14 f/s)", "svd"], ["🏃🏻‍♀️💨 SVD-XT (trained on 25 f/s)", "svdxt"]], label="Model", info="Trained model", value="auto", interactive=True)
               seed = gr.Slider(label="Seed", value=42, randomize=True, minimum=0, maximum=max_64_bit_int, step=1)
               randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
-          generate_btn = gr.Button(value="Animate", variant="primary")
+          generate_btn = gr.Button(value="🚀 Animate", variant="primary")
 
       with gr.Column():
-          video = gr.Video(label="Generated video")
-          gallery = gr.Gallery(label="Generated frames")
+          video = gr.Video(label="Generated video", autoplay=True)
+          download_button = gr.DownloadButton(label="💾 Download video", visible=False)
+          gallery = gr.Gallery(label="Generated frames", visible=False)
       
   image.upload(fn=resize_image, inputs=image, outputs=image, queue=False)
-  generate_btn.click(fn=sample, inputs=[image, seed, randomize_seed, motion_bucket_id, fps_id, noise_aug_strength, decoding_t, frame_format, version], outputs=[video, gallery, seed], api_name="video")
+  generate_btn.click(fn=sample, inputs=[image, seed, randomize_seed, motion_bucket_id, fps_id, noise_aug_strength, decoding_t, frame_format, version], outputs=[video, download_button, gallery, seed], api_name="video")
     
   gr.Examples(
     examples=[
@@ -125,7 +126,7 @@ with gr.Blocks() as demo:
         ["Examples/Town.jpeg", 42, True, 127, 25, 0.1, 3, "png", "auto"]
     ],
     inputs=[image, seed, randomize_seed, motion_bucket_id, fps_id, noise_aug_strength, decoding_t, frame_format, version],
-    outputs=[video, gallery, seed],
+    outputs=[video, download_button, gallery, seed],
     fn=sample,
     run_on_click=True,
     cache_examples=False,