Add information and fix video format (#22)

- Add information and fix video format (7e266454113338159a0424722f876bd76384e002)


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

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import torch
 import os
 import random
+import time
+import math
 import spaces
 from glob import glob
 from pathlib import Path
@@ -36,6 +38,7 @@ def animate(
     version: str = "auto",
     output_folder: str = "outputs",
 ):
+    start = time.time()
     if image.mode == "RGBA":
         image = image.convert("RGB")
         
@@ -57,8 +60,20 @@ def animate(
     video_path = os.path.join(output_folder, f"{base_count:06d}." + video_format)
 
     export_to_video(frames, video_path, fps=fps_id)
+    end = time.time()
+    secondes = int(end - start)
+    minutes = math.floor(secondes / 60)
+    secondes = secondes - (minutes * 60)
+    hours = math.floor(minutes / 60)
+    minutes = minutes - (hours * 60)
+    information = ("Start the process again if you want a different result. " if randomize_seed else "") + \
+    "Wait 2 min before a new run to avoid quota penalty or use another computer. " + \
+    "The video has been generated in " + \
+    ((str(hours) + " h, ") if hours != 0 else "") + \
+    ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + \
+    str(secondes) + " sec."
     
-    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
+    return gr.update(value=video_path, format=video_format), gr.update(value=video_path, visible=True), gr.update(label="Generated frames in *." + frame_format + " format", format = frame_format, value = frames, visible=True), seed, gr.update(value = information, visible = True)
 
 @spaces.GPU(duration=120)
 def animate_on_gpu(
@@ -136,10 +151,11 @@ with gr.Blocks() as demo:
       with gr.Column():
           video = gr.Video(label="Generated video", autoplay=True)
           download_button = gr.DownloadButton(label="💾 Download video", visible=False)
+          information_msg = gr.HTML(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=animate, inputs=[image, seed, randomize_seed, motion_bucket_id, fps_id, noise_aug_strength, decoding_t, video_format, frame_format, version], outputs=[video, download_button, gallery, seed], api_name="video")
+  generate_btn.click(fn=animate, inputs=[image, seed, randomize_seed, motion_bucket_id, fps_id, noise_aug_strength, decoding_t, video_format, frame_format, version], outputs=[video, download_button, gallery, seed, information_msg], api_name="video")
     
   gr.Examples(
     examples=[
@@ -148,7 +164,7 @@ with gr.Blocks() as demo:
         ["Examples/Town.jpeg", 42, True, 127, 25, 0.1, 3, "mp4", "png", "auto"]
     ],
     inputs=[image, seed, randomize_seed, motion_bucket_id, fps_id, noise_aug_strength, decoding_t, video_format, frame_format, version],
-    outputs=[video, download_button, gallery, seed],
+    outputs=[video, download_button, gallery, seed, information_msg],
     fn=animate,
     run_on_click=True,
     cache_examples=False,