debugging

handler.py

@@ -258,6 +258,15 @@ class EndpointHandler():
             for f in files:
                 print(f"{subindent}{f}")
 
+    def print_directory_contents(self, path='.'):
+        for root, dirs, files in os.walk(path):
+            level = root.replace(path, '').count(os.sep)
+            indent = ' ' * 4 * level
+            print(f'{indent}{os.path.basename(root)}/')
+            sub_indent = ' ' * 4 * (level + 1)
+            for f in files:
+                print(f'{sub_indent}{f}')
+
     def __call__(self, data: Any) -> Dict[str, str]:
         inputs = data.get("inputs", {})
         ref_image_url = inputs.get("ref_image_url", "")
@@ -363,9 +372,11 @@ class EndpointHandler():
             torch.cuda.empty_cache()
 
             # Perform face swapping
+            self.print_directory_contents()
             swapped_face_video_path = os.path.join(save_dir, "swapped_face_output.mp4")
             
             # Subprocess call to facefusion for face swapping
+            self.print_directory_contents()
             facefusion_script_path = os.path.join(base_dir, 'facefusion', 'core.py')
             swap_command = f'python3 {facefusion_script_path} --source {cropped_face_path} --target {animation_path} --output {swapped_face_video_path}'
             swap_result = subprocess.run(swap_command, shell=True, capture_output=True, text=True)
@@ -382,6 +393,7 @@ class EndpointHandler():
 
 
             #remove background
+            self.print_directory_contents()
             removed_background_output_path = os.path.join(save_dir, "removed_background_result.mp4")
             remove_background_command = f'python3 ./rembg_video.py {swapped_face_video_path} {removed_background_output_path}'
             print("Command is " + remove_background_command)

sampler.py

@@ -3,8 +3,20 @@ import json
 from handler import EndpointHandler
 from PIL import Image
 import io
+import os
+import subprocess
 
 # Initialize the handler
+# base_dir = os.path.dirname(os.path.abspath(__file__))
+# video_root = os.path.join(base_dir, "dw_poses_videos")
+# extract_pose_path = os.path.join(base_dir, 'extract_dwpose_from_vid.py')
+# command = f'python3 {extract_pose_path} --video_root {video_root}'
+
+# # Run the command with shell=True
+# result = subprocess.run(command, shell=True, capture_output=True, text=True)
+# if result.returncode != 0:
+#     raise RuntimeError(f"Error running extract_dwpose_from_vid.py: {result.stderr}")
+
 handler = EndpointHandler()
 
 # Define sample inputs
@@ -12,7 +24,7 @@ inputs = {
     "inputs": {
         "ref_image_url": "https://media.discordapp.net/attachments/1237667104866697267/1246982998520496261/image.jpg?ex=665e5eea&is=665d0d6a&hm=46a719438e039655788336902d62419a2921a00720b21ac17e74755cb6de8cee&=&format=webp&width=976&height=1192",
         "video_url": "https://cdn.discordapp.com/attachments/1237667104866697267/1246950669689290752/pose.mp4?ex=665e40ce&is=665cef4e&hm=37a5f042a2d03f71c4ae5e9f56616a35d205b97df0ff205182cab28aa7784bbb&",
-        "length": 96,
+        "length": 12,
         "num_inference_steps": 25,
         "cfg": 3.5,
         "seed": 123