add: logic for skipping necklace portion from the video

app.py

@@ -237,15 +237,15 @@ class EachNecklaceVideoGeneratorRequest(BaseModel):
     background_audio_path: str = "TraditionalIndianVlogMusic.mp3"
     image_display_duration: float = 2.5
     fps: int = 10
-    necklace_title: List[str]
-    nto_image_title: List[List[str]]
-    nto_cto_image_title: List[List[str]]
-    makeup_image_title: List[List[str]]
-    necklace_images: List[str]
-    necklace_try_on_output_images: List[List[str]]
-    clothing_output_images: List[List[str]]
-    makeup_output_images: List[List[str]]
-    background_colors: list[tuple[int, int, int]]
+    necklace_title: List[str] = None
+    nto_image_title: List[List[str]] = None
+    nto_cto_image_title: List[List[str]] = None
+    makeup_image_title: List[List[str]] = None
+    necklace_images: List[str] = None
+    necklace_try_on_output_images: List[List[str]] = None
+    clothing_output_images: List[List[str]] = None
+    makeup_output_images: List[List[str]] = None
+    background_colors: list[tuple[int, int, int]] = None
     outro_title: str = "Reach out to us for more information"
     address: str = "123, ABC Street, XYZ City"
     phone_numbers: str = "1234567890"
@@ -254,6 +254,7 @@ class EachNecklaceVideoGeneratorRequest(BaseModel):
     transition_type: str = "None"
     direction: str = "left"
     outro_video_path: str = "JewelMirror_outro.mp4"
+    skip_necklace_video_portion: bool = False
 
 
 @app.post("/createcombinedvideo/")
@@ -284,12 +285,6 @@ async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
                 return all(verify_files(item) for item in files)
             return files is not None
 
-        if not all(verify_files(files) for files in temp_files.values()):
-            return JSONResponse(
-                content={"status": "error", "message": "Failed to download all required images."},
-                status_code=400
-            )
-
         intro_path = f"{RESOURCES_DIR}/intro/{request.intro_video_path}"
         output_path = f"{TEMP_VIDEO_DIR}/video_{os.urandom(8).hex()}.mp4"
         font_path = f"{RESOURCES_DIR}/fonts/{request.font_path}"
@@ -297,7 +292,6 @@ async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
         outro_path = f"{RESOURCES_DIR}/outro/{request.outro_video_path}"
         try:
             model_name = request.necklace_try_on_output_images[0][0].split("/")[-1].split(".")[0].split("-")[1]
-
         except:
             model_name = "MOD"
 
@@ -324,7 +318,8 @@ async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
             transition_duration=request.transition_duration,
             transition_type=request.transition_type,
             direction=request.direction,
-            outro_video_path=outro_path
+            outro_video_path=outro_path,
+            skip_necklace_video_portion=request.skip_necklace_video_portion
 
         )
         logger.info("Creating video...")
@@ -364,7 +359,6 @@ async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
         return JSONResponse(content=response, status_code=200)
 
     except Exception as e:
-        raise e
         return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
 
 

src/components/each_necklace_video_gen.py

@@ -23,7 +23,7 @@ class EachVideoCreator:
                  audio_path=None, image_display_duration=2.5, box_color=(131, 42, 48), box_opacity=0.8,
                  font_size=28, text_color="white", fps=1, outro_title=None, logo_image=None, address=None,
                  phone_numbers=None, transition_duration=0.5, transition_type='None', direction='left',
-                 outro_video_path: str = ""):
+                 outro_video_path: str = "", skip_necklace_video_portion=False):
         self.intro_video_path = intro_video_path
         self.necklace_images = necklace_image if necklace_image else []
         self.nto_outputs = nto_outputs if nto_outputs else []
@@ -51,6 +51,7 @@ class EachVideoCreator:
         self.transition_type = transition_type
         self.direction = direction
         self.outro_video_path = outro_video_path
+        self.skip_necklace_video_portion = skip_necklace_video_portion
 
     def create_necklace_clips(self, necklace_image, index, label):
         if not necklace_image:
@@ -156,14 +157,18 @@ class EachVideoCreator:
                 print("Skipping intro video: Path not provided or invalid.")
 
             # Step 2: Process Necklaces and Associated Outputs
+
             for idx, necklace_image in enumerate(self.necklace_images):
-                print(f"Processing Necklace {idx + 1}...")
-                # Necklace preview clips
-                necklace_clips = self.create_necklace_clips(necklace_image, idx, self.necklace_title[idx])
-                if necklace_clips:
-                    clips.extend(necklace_clips)
+                if self.skip_necklace_video_portion:
+                    pass
                 else:
-                    print(f"Skipping Necklace {idx + 1} preview: No valid clips created.")
+                    print(f"Processing Necklace {idx + 1}...")
+                    # Necklace preview clips
+                    necklace_clips = self.create_necklace_clips(necklace_image, idx, self.necklace_title[idx])
+                    if necklace_clips:
+                        clips.extend(necklace_clips)
+                    else:
+                        print(f"Skipping Necklace {idx + 1} preview: No valid clips created.")
 
                 # NTO outputs
                 if idx < len(self.nto_outputs):
@@ -202,13 +207,6 @@ class EachVideoCreator:
                 outro_clip = resize(VideoFileClip(self.outro_video_path), (1080, 1080))
                 clips.append(outro_clip)
 
-            # if self.logo_image or self.address or self.phone_numbers is not None:
-            #     outro_clip = self.create_last_clip(title=self.outro_title, logo_image=self.logo_image,
-            #                                        address=self.address,
-            #                                        phone_numbers=self.phone_numbers,
-            #                                        font_path=self.font_path)
-            #     clips.extend([outro_clip])
-
             final_clips = []
             for clip in clips:
                 clip = clip.set_duration(self.image_display_duration)