add: outro page, bug fix- title mismatched

app.py

@@ -195,14 +195,18 @@ class EachNecklaceVideoGeneratorRequest(BaseModel):
     image_display_duration: float = 2.5
     fps: int = 10
     necklace_title: List[str]
-    nto_image_title: List[str]
-    nto_cto_image_title: List[str]
-    makeup_image_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]]
+    outro_title: str = "Reach out to us for more information"
+    address: str = "123, ABC Street, XYZ City"
+    phone_numbers: str = "1234567890"
+    logo_url: str = "https://lvuhhlrkcuexzqtsbqyu.supabase.co/storage/v1/object/public/MagicMirror/FullImages/default.png"
 
 
 @app.post("/createcombinedvideo/")
@@ -222,7 +226,8 @@ async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
             'necklaces': process_images(request.necklace_images),
             'necklace_tryon': process_images(request.necklace_try_on_output_images),
             'clothing': process_images(request.clothing_output_images),
-            'makeup': process_images(request.makeup_output_images)
+            'makeup': process_images(request.makeup_output_images),
+            'logo': download_image(request.logo_url)
         }
         file_download_time = time.time() - start_time
 
@@ -256,19 +261,20 @@ async def create_combined_video(request: EachNecklaceVideoGeneratorRequest):
             fps=request.fps,
             necklace_title=request.necklace_title,
             nto_title=request.nto_image_title,
-
             cto_title=request.nto_cto_image_title,
             makeup_title=request.makeup_image_title,
-            backgrounds=background_
+            backgrounds=background_,
+            outro_title=request.outro_title,
+            address=request.address,
+            phone_numbers=request.phone_numbers,
+            logo_image=temp_files['logo']
 
         )
         logger.info("Creating video...")
 
-        # Generate video
         video_creator.create_final_video()
         video_creation_time = time.time() - start_time - file_download_time
         logger.info("Video created successfully.")
-        # Upload video to Supabase
 
         url = upload_to_supabase(video_path=output_path)
         supabase_upload_time = time.time() - start_time - file_download_time - video_creation_time

src/components/each_necklace_video_gen.py

@@ -3,6 +3,7 @@ project @ images_to_video
 created @ 2024-12-17
 author  @ github.com/ishworrsubedii
 """
+
 import os
 
 from moviepy.audio.io.AudioFileClip import AudioFileClip
@@ -19,7 +20,8 @@ class EachVideoCreator:
                  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, outro_title=None, logo_image=None, address=None,
+                 phone_numbers=None):
         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 []
@@ -39,8 +41,10 @@ class EachVideoCreator:
         self.cto_title = cto_title
         self.makeup_title = makeup_title
         self.backgrounds = backgrounds
-
-
+        self.outro_title = outro_title
+        self.logo_image = logo_image
+        self.address = address
+        self.phone_numbers = phone_numbers
 
     def create_necklace_clips(self, necklace_image, index, label):
         if not necklace_image:
@@ -70,7 +74,7 @@ class EachVideoCreator:
                 if os.path.exists(img_path) and img_path.lower().endswith(('.png', '.jpg', '.jpeg')):
                     print(f"Processing {label} image: {img_path}")
                     img_clip = resize(ImageClip(img_path), (1080, 1080))
-                    txt_overlay = self.create_text_overlay(f"{label} {idx + 1}")
+                    txt_overlay = self.create_text_overlay(f"{label}")
                     final_clip = CompositeVideoClip([
                         img_clip.set_duration(self.image_display_duration),
                         txt_overlay.set_position(('center', 'bottom'))
@@ -84,6 +88,35 @@ class EachVideoCreator:
         txt = TextClip(text, font=self.font_path, fontsize=self.font_size, color=self.text_color)
         return CompositeVideoClip([box, txt.set_position('center')])
 
+    def create_last_clip(self, title, logo_image, address, phone_numbers, font_path):
+        # Background clip (1080x1080, light gray color)
+        bg_clip = ColorClip((1080, 1080), col=(245, 245, 245), duration=self.image_display_duration)
+
+        # Resize logo to fit well within the 1080x1080 frame
+        logo = resize(ImageClip(logo_image), width=400)
+        logo = logo.set_duration(self.image_display_duration)
+        logo = logo.set_position(lambda t: ('center', 200))  # Place logo near top
+
+        # Title overlay text
+        txt_overlay_title = TextClip(title, fontsize=50, color='black', font=font_path)
+        txt_overlay_title = txt_overlay_title.set_duration(self.image_display_duration)
+        txt_overlay_title = txt_overlay_title.set_position(lambda t: ('center', 600))  # Below logo
+
+        # Address overlay text
+        txt_overlay_address = TextClip(address, fontsize=30, color='black', font=font_path)
+        txt_overlay_address = txt_overlay_address.set_duration(self.image_display_duration)
+        txt_overlay_address = txt_overlay_address.set_position(lambda t: ('center', 680))  # Below title
+
+        # Phone number overlay text
+        txt_overlay_phone = TextClip(phone_numbers, fontsize=30, color='black', font=font_path)
+        txt_overlay_phone = txt_overlay_phone.set_duration(self.image_display_duration)
+        txt_overlay_phone = txt_overlay_phone.set_position(lambda t: ('center', 730))  # Below address
+
+        # Combine everything into the final clip
+        final_clip = CompositeVideoClip([bg_clip, logo, txt_overlay_title, txt_overlay_address, txt_overlay_phone])
+
+        return final_clip
+
     def create_final_video(self):
         try:
             print("Starting video creation...")
@@ -109,31 +142,40 @@ class EachVideoCreator:
 
                 # NTO outputs
                 if idx < len(self.nto_outputs):
-                    print(f"Adding NTO outputs for Necklace {idx + 1}")
-                    nto_clips = self.create_grouped_clips([self.nto_outputs[idx]], self.nto_title[idx])
-                    if nto_clips:
+                    for idj in range(len(self.nto_outputs[idx])):
+                        print(f"Total NTO outputs{len(self.nto_outputs)}")
+                        print(f"Adding NTO outputs for Necklace {idx + 1}")
+                        nto_clips = self.create_grouped_clips([self.nto_outputs[idx]], self.nto_title[idx][idj])
+                        if nto_clips:
+                            clips.extend(nto_clips)
 
-                        clips.extend(nto_clips)
-                    else:
-                        print(f"No valid NTO clips for Necklace {idx + 1}")
-
-                # CTO outputs
                 if idx < len(self.nto_cto_outputs):
-                    print(f"Adding CTO outputs for Necklace {idx + 1}")
-                    cto_clips = self.create_grouped_clips([self.nto_cto_outputs[idx]], self.cto_title[idx])
-                    if cto_clips:
-                        clips.extend(cto_clips)
-                    else:
-                        print(f"No valid CTO clips for Necklace {idx + 1}")
-
-                # Makeup outputs
+                    for idj in range(len(self.nto_cto_outputs[idx])):
+                        print(f"Total CTO outputs{len(self.nto_cto_outputs)}")
+                        print(f"Adding CTO outputs for Necklace {idx + 1}")
+                        cto_clips = self.create_grouped_clips([self.nto_cto_outputs[idx]], self.cto_title[idx][idj])
+                        if cto_clips:
+                            clips.extend(cto_clips)
+                        else:
+                            print(f"No valid CTO clips for Necklace {idx + 1}")
+
                 if idx < len(self.makeup_outputs):
-                    print(f"Adding Makeup outputs for Necklace {idx + 1}")
-                    makeup_clips = self.create_grouped_clips([self.makeup_outputs[idx]], self.makeup_title[idx])
-                    if makeup_clips:
-                        clips.extend(makeup_clips)
-                    else:
-                        print(f"No valid Makeup clips for Necklace {idx + 1}")
+                    for idj in range(len(self.makeup_outputs[idx])):
+                        print(f"Total Makeup outputs{len(self.makeup_outputs)}")
+                        print(f"Adding Makeup outputs for Necklace {idx + 1}")
+                        makeup_clips = self.create_grouped_clips([self.makeup_outputs[idx]],
+                                                                 self.makeup_title[idx][idj])
+                        if makeup_clips:
+                            clips.extend(makeup_clips)
+                        else:
+                            print(f"No valid Makeup clips for Necklace {idx + 1}")
+
+            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:
@@ -172,16 +214,3 @@ class EachVideoCreator:
 
         except Exception as e:
             print(f"An error occurred during video creation: {e}")
-
-# if __name__ == "__main__":
-#     creator = EachVideoCreator(
-#         intro_video_path="intro.mp4",
-#         necklace_image=["necklace.jpg", "necklace2.png"],
-#         nto_outputs=[["nto1.jpg", "nto2.jpg"], ["nto3.jpg", "nto4.jpg"]],
-#         nto_cto_outputs=[["cto1.jpg", "cto2.jpg"], ["cto3.jpg", "cto4.jpg"]],
-#         makeup_outputs=[["makeup1.jpg", "makeup2.jpg"], ["makeup3.jpg", "makeup4.jpg"]],
-#         font_path="font.ttf",
-#         output_path="output/final_video.mp4",
-#         audio_path="audio.mp3"
-#     )
-#     creator.create_final_video()