Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import subprocess
+from share_btn import community_icon_html, loading_icon_html, share_js
 import cv2
 import numpy as np
 from moviepy.editor import VideoFileClip, concatenate_videoclips
@@ -119,7 +120,7 @@ def change_video_fps(input_path):
 
     return 'output_video.mp4'
 
-def run_inference(prompt, video_path, condition, video_length, seed, steps):
+def run_inference(prompt, video_path, condition, video_length, seed):
     
     seed = math.floor(seed)
     o_width = get_video_dimension(video_path)[0]
@@ -166,9 +167,9 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
 
     print(f"RUNNING INFERENCE ...")
     if video_length > 12:
-        command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --condition '{condition}' --video_path '{resized}' --output_path '{output_path}' --temp_chunk_path 'result' --width {r_width} --height {r_height} --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20 --is_long_video"
+        command = f"python inference.py --prompt '{prompt}' --inference_steps 50 --condition '{condition}' --video_path '{resized}' --output_path '{output_path}' --temp_chunk_path 'result' --width {r_width} --height {r_height} --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20 --is_long_video"
     else:
-        command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --condition '{condition}' --video_path '{resized}' --output_path '{output_path}' --temp_chunk_path 'result'  --width {r_width} --height {r_height} --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20"
+        command = f"python inference.py --prompt '{prompt}' --inference_steps 50 --condition '{condition}' --video_path '{resized}' --output_path '{output_path}' --temp_chunk_path 'result'  --width {r_width} --height {r_height} --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20"
     
     try:
         subprocess.run(command, shell=True)
@@ -190,11 +191,64 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
     print(f"GEN VIDEO FPS: {gen_fps}")
     final = change_video_fps(video_path_output)
     print(f"FINISHED !")
-    return "done", final
+    
+    return final
  
 
 css="""
 #col-container {max-width: 810px; margin-left: auto; margin-right: auto;}
+.animate-spin {
+  animation: spin 1s linear infinite;
+}
+@keyframes spin {
+  from {
+      transform: rotate(0deg);
+  }
+  to {
+      transform: rotate(360deg);
+  }
+}
+#share-btn-container {
+  display: flex; 
+  padding-left: 0.5rem !important; 
+  padding-right: 0.5rem !important; 
+  background-color: #000000; 
+  justify-content: center; 
+  align-items: center; 
+  border-radius: 9999px !important; 
+  max-width: 13rem;
+}
+#share-btn-container:hover {
+  background-color: #060606;
+}
+#share-btn {
+  all: initial; 
+  color: #ffffff;
+  font-weight: 600; 
+  cursor:pointer; 
+  font-family: 'IBM Plex Sans', sans-serif; 
+  margin-left: 0.5rem !important; 
+  padding-top: 0.5rem !important; 
+  padding-bottom: 0.5rem !important;
+  right:0;
+}
+#share-btn * {
+  all: unset;
+}
+#share-btn-container div:nth-child(-n+2){
+  width: auto !important;
+  min-height: 0px !important;
+}
+#share-btn-container .wrap {
+  display: none !important;
+}
+#share-btn-container.hidden {
+  display: none!important;
+}
+img[src*='#center'] { 
+    display: block;
+    margin: auto;
+}
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
@@ -216,7 +270,10 @@ with gr.Blocks(css=css) as demo:
                 submit_btn = gr.Button("Submit")
             with gr.Column():
                 video_res = gr.Video(label="result", elem_id="video-out")
-                status = gr.Textbox(label="result")
+                with gr.Group(elem_id="share-btn-container", visible=False) as share_group:
+                    community_icon = gr.HTML(community_icon_html)
+                    loading_icon = gr.HTML(loading_icon_html)
+                    share_button = gr.Button("Share to community", elem_id="share-btn")
                 gr.Examples(
                     examples=[["Indiana Jones moonwalk in the Jungle", "./examples/moonwalk.mp4", 'depth', 12, 424242, 50]],
                     fn=run_inference,
@@ -224,10 +281,9 @@ with gr.Blocks(css=css) as demo:
                              video_path,
                              condition,
                              video_length,
-                             seed,
-                             inference_steps
+                             seed
                             ],
-                    outputs=[status, video_res],
+                    outputs=[video_res],
                     cache_examples=True
                 )
     video_path.change(fn=get_frame_count,
@@ -240,9 +296,8 @@ with gr.Blocks(css=css) as demo:
                              video_path,
                              condition,
                              video_length,
-                             seed,
-                             inference_steps
+                             seed
                             ],
-                    outputs=[status, video_res])
+                    outputs=[video_res])
 
 demo.queue(max_size=12).launch()