Update app.py

app.py

@@ -31,16 +31,19 @@ def check_for_mp4_in_outputs(given_folder):
         return None
 
 
-def infer():
+def infer(input_video, cropped_and_aligned):
 
-    filepath = "./assets/examples/synthetic_1.mp4"
+    filepath = input_video
     # Get the current timestamp
     timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
 
     output_folder_name = f"results_{timestamp}"
 
-    # Example: Run the inference script (replace with your actual command)
-    run_command(f"{sys.executable} inference_keep.py -i={filepath} -o={output_folder_name} --has_aligned --save_video -s=1")
+    if cropped_and_aligned is True:
+        # Example: Run the inference script (replace with your actual command)
+        run_command(f"{sys.executable} inference_keep.py -i={filepath} -o={output_folder_name} --has_aligned --save_video -s=1")
+    else:
+        run_command(f"{sys.executable} inference_keep.py -i={filepath} -o={output_folder_name} --draw_box --save_video -s=1 --bg_upsampler=realesrgan")
 
     # Call the function and print the result
     this_infer_folder = os.path.splitext(os.path.basename(filepath))[0]
@@ -52,19 +55,43 @@ def infer():
     
     return mp4_file_path
 
+
+result_video = gr.Video()
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# KEEP")
+        gr.Markdown("## Kalman-Inspired Feature Propagation for Video Face Super-Resolution")
         with gr.Row():
             with gr.Column():
+                input_video = gr.Video(label="Input Video")
+                is_cropped_and_aligned = gr.Checkbox(label="Synthetic data", info="Is your input video ready with cropped and aligned faces ?", value=False)
                 submit_btn = gr.Button("Submit")
+                gr.Examples(
+                    examples = [
+                        ["./assets/examples/synthetic_1.mp4", True],
+                        ["./assets/examples/synthetic_2.mp4", True],
+                        ["./assets/examples/synthetic_3.mp4", True],
+                        ["./assets/examples/synthetic_4.mp4", True],
+                        ["./assets/examples/real_1.mp4", False],
+                        ["./assets/examples/real_2.mp4", False],
+                        ["./assets/examples/real_3.mp4", False],
+                        ["./assets/examples/real_4.mp4", False]
+                    ],
+                    fn = infer,
+                    inputs = [input_video, is_cropped_and_aligned],
+                    outputs = [result_video],
+                    run_on_click = False,
+                    cache_examples = False
+                )
+            
             with gr.Column():
-                result = gr.Video()
+                result_video.render()
+                
 
     submit_btn.click(
         fn = infer, 
-        inputs = None,
-        outputs = [result],
+        inputs = [input_video, is_cropped_and_aligned],
+        outputs = [result_video],
         show_api=False
     )