Update app.py

app.py

@@ -50,9 +50,12 @@ def infer(video_frames, masks_frames, project_name):
         # Print the image name and its corresponding save path
         print(f"Image {idx + 1}: {image_name} copied to {destination_path}")
 
-    video_frames_folder = "inputs/object_removal/bmx-trees"
-    masks_folder = "inputs/object_removal/bmx-trees_mask"
+    #video_frames_folder = "inputs/object_removal/bmx-trees"
+    #masks_folder = "inputs/object_removal/bmx-trees_mask"
 
+    video_frames_folder = f"{my_video_directory}"
+    masks_folder = f"{my_masks_directory}"
+    
     # Create the "results" folder if it doesn't exist
     output_folder = "results"
     if not os.path.exists(output_folder):
@@ -80,14 +83,14 @@ def infer(video_frames, masks_frames, project_name):
         print(item)
 
     # List the content of the "bmx-trees" folder within "results"
-    bmx_trees_folder = os.path.join(output_folder, "bmx-trees")
-    bmx_trees_content = [os.path.join(bmx_trees_folder, item) for item in os.listdir(bmx_trees_folder)]
+    results_folder = os.path.join(output_folder, f"{project_name}")
+    results_folder_content = [os.path.join(results_folder, item) for item in os.listdir(results_folder)]
 
-    print(f"Contents of the {bmx_trees_folder} folder:")
-    for item in bmx_trees_content:
+    print(f"Contents of the {results_folder} folder:")
+    for item in results_content:
         print(item)
 
-    return "done", bmx_trees_content[0], bmx_trees_content[1]
+    return "done", results_content[0], results_content[1]
 
 css="""
 #col-container{