update: logging and dynamic background color

app.py

@@ -19,7 +19,7 @@ supabase_key = os.getenv('SUPABASE_KEY')
 supabase = create_client(supabase_url, supabase_key)
 app = FastAPI()
 
-RESOURCES_DIR = "/app/resources"
+RESOURCES_DIR = "resources"
 
 os.makedirs(RESOURCES_DIR, exist_ok=True)
 TEMP_VIDEO_DIR = f"{RESOURCES_DIR}/temp_video"
@@ -202,12 +202,14 @@ class EachNecklaceVideoGeneratorRequest(BaseModel):
     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]]
 
 
 @app.post("/createcombinedvideo/")
-async def create_video(request: EachNecklaceVideoGeneratorRequest):
+async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
     logger.info(f"Creating video with request: {request.dict()}")
     start_time = time.time()
+    background_ = request.background_colors
     try:
         logger.info("Downloading images...")
 
@@ -227,8 +229,8 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
         def verify_files(files):
             logger.info(f"Verifying files: {files}")
             if isinstance(files, list):
-                return all(verify_files(item) for item in files)  # Recurse for nested lists
-            return files is not None  # Check single file
+                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(
@@ -236,7 +238,6 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
                 status_code=400
             )
 
-        # Prepare paths
         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}"
@@ -257,7 +258,8 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
             nto_title=request.nto_image_title,
 
             cto_title=request.nto_cto_image_title,
-            makeup_title=request.makeup_image_title
+            makeup_title=request.makeup_image_title,
+            backgrounds=background_
 
         )
         logger.info("Creating video...")

src/components/each_necklace_video_gen.py

@@ -14,11 +14,12 @@ from moviepy.video.io.VideoFileClip import VideoFileClip
 
 
 class EachVideoCreator:
-    def __init__(self, necklace_title, nto_title, cto_title, makeup_title, intro_video_path=None, necklace_image=None,
+    def __init__(self, necklace_title, backgrounds: list[tuple], nto_title, cto_title, makeup_title,
+                 intro_video_path=None, necklace_image=None,
                  nto_outputs=None,
                  nto_cto_outputs=None, makeup_outputs=None, font_path=None, output_path=None,
                  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):
+                 font_size=28, text_color="white", fps=1, ):
         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 []
@@ -37,19 +38,23 @@ class EachVideoCreator:
         self.nto_title = nto_title
         self.cto_title = cto_title
         self.makeup_title = makeup_title
+        self.backgrounds = backgrounds
+
+
 
     def create_necklace_clips(self, necklace_image, index, label):
         if not necklace_image:
             print(f"Skipping necklace {index + 1}: No image provided.")
             return []
 
-        backgrounds = [
-            (245, 245, 245),  # Soft White
-            (220, 245, 245),  # Light Blue
-            (230, 230, 235)  # Pearl Gray
-        ]
+        # backgrounds = [
+        #     (245, 245, 245),  # Soft White
+        #     (220, 245, 245),  # Light Blue
+        #     (230, 230, 235)  # Pearl Gray
+        # ]
+
         necklace_clips = []
-        for bg_color in backgrounds:
+        for bg_color in self.backgrounds:
             bg_clip = ColorClip((1080, 1080), col=bg_color, duration=self.image_display_duration)
             necklace = resize(ImageClip(necklace_image), width=650)
             necklace = necklace.set_duration(self.image_display_duration).set_position('center')