Update app.py

app.py

@@ -6,12 +6,10 @@ import tempfile
 import huggingface_hub
 import subprocess
 import threading
-from tqdm import tqdm
 
-def stream_output(pipe, progress_bar):
+def stream_output(pipe):
     for line in iter(pipe.readline, ''):
         print(line, end='')
-        progress_bar.update(1)
     pipe.close()
 
 HF_TKN = os.environ.get("GATED_HF_TOKEN")
@@ -80,7 +78,7 @@ def check_for_mp4_in_outputs():
     else:
         return None
 
-def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
+def infer(ref_image_in, ref_video_in):
     # check if 'outputs' dir exists and empty it if necessary
     check_outputs_folder('./outputs')
     
@@ -135,12 +133,9 @@ def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
         command = ['python', 'inference.py', '--inference_config', file_path]
         process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, bufsize=1)
     
-        # Create a tqdm progress bar
-        progress_bar = tqdm(total=100, desc="Inference Progress", unit="line", ncols=100, leave=True)
-    
         # Create threads to handle stdout and stderr
-        stdout_thread = threading.Thread(target=stream_output, args=(process.stdout, progress_bar))
-        stderr_thread = threading.Thread(target=stream_output, args=(process.stderr, progress_bar))
+        stdout_thread = threading.Thread(target=stream_output, args=(process.stdout,))
+        stderr_thread = threading.Thread(target=stream_output, args=(process.stderr,))
 
     
         # Start the threads
@@ -148,12 +143,9 @@ def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
         stderr_thread.start()
     
         # Wait for the process to complete and the threads to finish
+        process.wait()
         stdout_thread.join()
         stderr_thread.join()
-        process.wait()
-
-        # Close the progress bar
-        progress_bar.close()
     
         print("Inference script finished with return code:", process.returncode)
         
@@ -166,10 +158,19 @@ def infer(ref_video_in, ref_image_in, progress=gr.Progress(track_tqdm=True)):
     
     return mp4_file_path
 
-demo = gr.Interface(
-    fn = infer,
-    inputs = [gr.Video(), gr.Image(type="filepath")],
-    outputs = [gr.Video()]
-)
+with gr.Blocks() as demo:
+    with gr.Column():
+        with gr.Row():
+            with gr.Column():
+                with gr.Row():
+                    ref_image_in = gr.Image(type="filepath")
+                    ref_video_in = gr.Video()
+                submit_btn = gr.Button("Submit")
+            output_video = gr.Video()
+    submit_btn.click(
+        fn = infer,
+        inputs = [ref_image_in, ref_video_in],
+        outputs = [output_video]
+    )
 
 demo.launch()