fixing ifrbease update path

handler.py

@@ -290,10 +290,10 @@ class EndpointHandler():
         video_url = inputs.get("video_url", "")
         width = inputs.get("width", 512)
         height = inputs.get("height", 768)
-        length = inputs.get("length", 144)
-        num_inference_steps = inputs.get("num_inference_steps", 25)
+        length = inputs.get("length", 96)
+        num_inference_steps = inputs.get("num_inference_steps", 15)
         cfg = inputs.get("cfg", 3.5)
-        seed = inputs.get("seed", 123)
+        seed = inputs.get("seed", -1)
         firebase_doc_id = inputs.get("firebase_doc_id", "")
 
         base_dir = os.path.dirname(os.path.abspath(__file__))
@@ -323,6 +323,33 @@ class EndpointHandler():
             ref_image_no_bg_path = os.path.join(video_root, "ref_image_no_bg.png")
             ref_image_no_bg.save(ref_image_no_bg_path)
 
+                        # pose_output_path = os.path.join(temp_dir, "pose_videos")
+
+            # print("we are number 1")
+            # # Run the extract_dwpose_from_vid.py script
+            # 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}")
+            # print("we are number 2")
+
+            # # Locate the extracted pose video
+            # save_dir = video_root + "_dwpose"
+            # print(f"Expected save directory: {save_dir}")  # Debug statement
+            # pose_video_path = os.path.join(save_dir, "downloaded_video.mp4")
+
+            # if not os.path.exists(pose_video_path):
+            #     print("Contents of the temporary directory:")
+            #     self.print_directory_contents(temp_dir)
+            #     raise FileNotFoundError(f"The pose video was not found at: {pose_video_path}")
+
+            # Speed up the pose video by 4x
+            # sped_up_pose_video_path = os.path.join(temp_dir, "sped_up_pose_video.mp4")
+            # self.speed_up_video(downloaded_video_path, sped_up_pose_video_path, factor=2)
+
             torch.manual_seed(seed)
 
             pose_images = read_frames(downloaded_video_path)
@@ -361,15 +388,38 @@ class EndpointHandler():
             if swap_result.returncode != 0:
                 raise RuntimeError(f"Error running face swap: {swap_result.stderr}")
 
+            # Slow down the produced video by 4x
+            # self.print_directory_contents(temp_dir)
+            # slowed_down_animation_path = os.path.join(save_dir, "slowed_down_animation_output.mp4")
+            # self.slow_down_video(swapped_face_video_path, slowed_down_animation_path, factor=2)
+
+
             torch.cuda.empty_cache()
 
+            #remove background
+            # self.print_directory_contents()
+            # removed_background_output_path = os.path.join(save_dir, "removed_background_result.mp4")
+            # remove_background_script_path = os.path.join(base_dir, "rembg_video.py")
+            # remove_background_command = f'python3 {remove_background_script_path} {swapped_face_video_path} {removed_background_output_path}'
+            # print("Command is " + remove_background_command)
+            # remove_background_result = subprocess.run(remove_background_command, shell=True, capture_output=True, text=True)
+            # if remove_background_result.returncode != 0:
+            #     raise RuntimeError(f"Error running removing backgriund: {remove_background_result.stderr}")
+
+
+            # Perform RIFE interpolation
+            # self.print_directory_contents(temp_dir)
+            # rife_output_path = os.path.join(save_dir, "completed_result.mp4")
+            # self.run_rife_interpolation(swapped_face_video_path, rife_output_path, multi=2, scale=0.5)
+
+
             with open(swapped_face_video_path, "rb") as video_file:
                 video_base64 = base64.b64encode(video_file.read()).decode("utf-8")
 
             # Upload video to Firebase Storage
             bucket = storage.bucket()
             blob = bucket.blob(f"videos/{firebase_doc_id}/swapped_face_output.mp4")
-            blob.upload_from_filename(swapped_face_video_path)
+            blob.upload_from_filename(slowed_down_animation_path)
             
             # Make the file publicly accessible
             blob.make_public()
@@ -379,6 +429,6 @@ class EndpointHandler():
             # Update Firestore document
             db = firestore.client()
             doc_ref = db.collection('danceResults').document(firebase_doc_id)
-            doc_ref.update({"videoURL": video_url})
+            doc_ref.update({"videoResultUrl": video_url})
 
             return {"video": video_base64}

sampler.py

@@ -22,12 +22,12 @@ handler = EndpointHandler()
 # Define sample inputs
 inputs = {
     "inputs": {
-        "ref_image_url": "https://media.discordapp.net/attachments/1237667074210267217/1247311431028703242/image.png?ex=665f90cb&is=665e3f4b&hm=89767149275a6ef40e39d4a73d77019ab30ab665512398816bc614cfecdddb34&=&format=webp&quality=lossless&width=590&height=1194",
-        "video_url": "https://firebasestorage.googleapis.com/v0/b/quiz-app-edffe.appspot.com/o/creeper_shortneed_pose.mp4?alt=media&token=f87f6e8c-6ab2-445c-8b55-205f08f1f5d5",
-        "length": 6,
-        "num_inference_steps": 25,
+        "ref_image_url": "https://media.discordapp.net/attachments/1237667074210267217/1247398713228263494/image.jpg?ex=665fe214&is=665e9094&hm=ba3863d8149e4e43c6ec271b09fcd4027b6a58cc65b7a658797ab9fbc74b4808&=&format=webp&width=896&height=1194",
+        "video_url": "https://firebasestorage.googleapis.com/v0/b/quiz-app-edffe.appspot.com/o/tell_my_bf_shortened_pose.mp4?alt=media&token=d3b7cb47-6fbb-435c-94a7-2cbd62705efb",
+        "length": 48,
+        "num_inference_steps": 15,
         "cfg": 3.5,
-        "seed": 123,
+        "seed": -1,
         "firebase_doc_id" : "0gkm91Ao0GsmQ2WdLP9c"
     }
 }