Update app.py

app.py

@@ -107,6 +107,24 @@ def run_inference(temp_dir, removed_bg_path):
         
         # Retrieve the file name without the extension
         removed_bg_file_name = os.path.splitext(os.path.basename(removed_bg_path))[0]
+
+        # List objects in the "out" folder
+        out_folder_path = "out"
+        out_folder_objects = os.listdir(out_folder_path)
+        print(f"Objects in '{out_folder_path}':")
+        for obj in out_folder_objects:
+            print(f" - {obj}")
+        
+        # List objects in the "out/{removed_bg_file_name}" folder
+        specific_out_folder_path = os.path.join(out_folder_path, removed_bg_file_name)
+        if os.path.exists(specific_out_folder_path) and os.path.isdir(specific_out_folder_path):
+            specific_out_folder_objects = os.listdir(specific_out_folder_path)
+            print(f"\nObjects in '{specific_out_folder_path}':")
+            for obj in specific_out_folder_objects:
+                print(f" - {obj}")
+        else:
+            print(f"\nThe folder '{specific_out_folder_path}' does not exist.")
+        
         output_videos = glob(os.path.join(f"out/{removed_bg_file_name}", "*.mp4"))
         return output_videos
     except subprocess.CalledProcessError as e: