Update app.py

app.py

@@ -1,6 +1,63 @@
+import os
+import subprocess
 import gradio as gr
 
-def infer(text):
-    return text
+# execute a CLI command
+def execute_command(command: str) -> None:
+    subprocess.run(command, check=True)
 
-gr.Interface(fn=infer,inputs=[gr.Textbox()], outputs=[gr.Textbox()]).launch()
+
+def infer(video_frames, masks_frames):
+
+    video_frames_folder = "inputs/object_removal/bmx-trees"
+    masks_folder = "inputs/object_removal/bmx-trees_mask"
+
+    # Create the "results" folder if it doesn't exist
+    output_folder = "results"
+    if not os.path.exists(results_folder):
+        os.makedirs(results_folder)
+    
+    command = [
+      f"python", 
+      f"inference_propainter.py",
+      f"--video={video_frames_folder}",
+      f"--mask={masks_folder}",
+      f"--output={output_folder}"
+    ]
+
+    execute_command(command)
+
+    return "done"
+
+css="""
+#col-container{
+    margin: 0 auto;
+    max-width: 840px;
+    text-align: left;
+}
+"""
+    
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.HTML("""
+        <h2 style="text-align: center;">ProPainter</h2>
+        <p style="text-align: center;">
+            
+        </p>
+                """)
+
+        with gr.Row():
+            with gr.Column():
+                video_frames = gr.Files(label="Video frames")
+                masks_frames = gr.Files(label="Masks frames")
+        
+                submit_btn = gr.Button("Submit")
+
+            with gr.Column():
+                result = gr.Textbox(label="Result")
+
+        
+            
+    submit_btn.click(fn=infer, inputs=[video_frames, masks_frames], outputs=[result])
+    
+demo.queue(max_size=12).launch()