Update app.py

app.py

@@ -7,7 +7,47 @@ def execute_command(command: str) -> None:
     subprocess.run(command, check=True)
 
 
-def infer(video_frames, masks_frames):
+def infer(video_frames, masks_frames, project_name):
+
+
+    # Create the directory if it doesn't exist
+    my_video_directory = f"{project_name}"
+    if not os.path.exists(my_video_directory):
+        os.makedirs(my_video_directory)
+
+    # Assuming 'images' is a list of image file paths
+    for idx, image in enumerate(video_frames):
+        # Get the base file name (without path) from the original location
+        image_name = os.path.basename(image.name)
+    
+        # Construct the destination path in the working directory
+        destination_path = os.path.join(my_video_directory, image_name)
+    
+        # Copy the image from the original location to the working directory
+        shutil.copy(image.name, destination_path)
+    
+        # Print the image name and its corresponding save path
+        print(f"Image {idx + 1}: {image_name} copied to {destination_path}")
+
+
+    # Create the directory if it doesn't exist
+    my_masks_directory = f"{project_name}_masks"
+    if not os.path.exists(my_masks_directory):
+        os.makedirs(my_masks_directory)
+
+    # Assuming 'images' is a list of image file paths
+    for idx, image in enumerate(video_frames):
+        # Get the base file name (without path) from the original location
+        image_name = os.path.basename(image.name)
+    
+        # Construct the destination path in the working directory
+        destination_path = os.path.join(my_masks_directory, image_name)
+    
+        # Copy the image from the original location to the working directory
+        shutil.copy(image.name, destination_path)
+    
+        # 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"