add: music list, logger

.gitignore

@@ -2,3 +2,4 @@ resourcesssss
 temp_video
 __pycache__
 .idea
+/logs

app.py

@@ -12,6 +12,7 @@ from supabase import create_client
 
 from src.components.each_necklace_video_gen import EachVideoCreator
 from src.components.vidgen import VideoCreator
+from src.utils.logs import logger
 
 supabase_url = os.getenv('SUPABASE_URL')
 supabase_key = os.getenv('SUPABASE_KEY')
@@ -27,6 +28,8 @@ os.environ['MOVIEPY_TEMP_DIR'] = '/tmp/moviepy'
 
 
 def upload_to_supabase(video_path, bucket_name="JewelmirrorVideoGeneration"):
+    logger.info(f"Uploading video to Supabase: {video_path}")
+
     try:
         if not os.path.exists(video_path):
             raise FileNotFoundError(f"Video file not found: {video_path}")
@@ -39,6 +42,7 @@ def upload_to_supabase(video_path, bucket_name="JewelmirrorVideoGeneration"):
             "x-upsert": "true",
             "cache-control": "max-age=3600"
         }
+        logger.info(f"Uploading video to Supabase: {video_path}")
 
         with open(video_path, 'rb') as f:
             file_data = f.read()
@@ -52,17 +56,21 @@ def upload_to_supabase(video_path, bucket_name="JewelmirrorVideoGeneration"):
             public_url = supabase.storage.from_(bucket_name).get_public_url(file_name)
 
             response = requests.head(public_url)
+
             if response.status_code != 200:
+                logger.error(f"Upload verification failed: {response.status_code}")
                 raise Exception(f"Upload verification failed: {response.status_code}")
 
+            logger.info(f"Video uploaded successfully: {public_url}")
             return public_url
 
     except Exception as e:
-        print(f"Error uploading to Supabase: {str(e)}")
+        logger.error(f"Error uploading to Supabase: {str(e)}")
         return None
 
 
 def download_image(url):
+    logger.info(f"Downloading image from {url}")
     try:
         response = requests.get(url, stream=True)
         response.raise_for_status()
@@ -71,8 +79,11 @@ def download_image(url):
         with open(temp_file.name, 'wb') as f:
             for chunk in response.iter_content(chunk_size=8192):
                 f.write(chunk)
+
+        logger.info(f"Image downloaded successfully: {temp_file.name}")
         return temp_file.name
     except Exception as e:
+        logger.error(f"Error downloading image from {url}: {str(e)}")
         print(f"Error downloading image from {url}: {str(e)}")
         return None
 
@@ -98,8 +109,10 @@ class VideoGenerator(BaseModel):
 
 @app.post("/createvideo/")
 async def create_video(request: VideoGenerator):
+    logger.info(f"Creating video with request: {request.dict()}")
     start_time = time.time()
     try:
+        logger.info("Downloading images...")
         temp_files = {
             'necklace': download_image(request.necklace_image),
             'necklace_tryon': [download_image(url) for url in request.necklace_try_on_output_images],
@@ -117,6 +130,7 @@ async def create_video(request: VideoGenerator):
         output_path = f"{TEMP_VIDEO_DIR}/video_{os.urandom(8).hex()}.mp4"
         font_path = f"{RESOURCES_DIR}/fonts/{request.font_path}"
         audio_path = f"{RESOURCES_DIR}/audio/{request.background_audio_path}"
+        logger.info(f"Creating video with paths: {intro_path}, {output_path}, {font_path}, {audio_path}")
 
         video_creator = VideoCreator(
             intro_video_path=intro_path,
@@ -135,12 +149,15 @@ async def create_video(request: VideoGenerator):
             makeup_image_title=request.makeup_image_title
 
         )
+        logger.info("Creating video...")
 
         video_creator.create_final_video()
         video_creation_time = time.time() - start_time - file_download_time
 
         url = upload_to_supabase(video_path=output_path)
+        logger.info(f"Video created successfully: {url}")
         supabase_upload_time = time.time() - start_time - file_download_time - video_creation_time
+        logger.info(f"Video uploaded to Supabase: {url}")
 
         response = {
             "status": "success",
@@ -152,6 +169,7 @@ async def create_video(request: VideoGenerator):
                 "supabase_upload_time": supabase_upload_time
             }
         }
+        logger.info(f"Response: {response}")
 
         for files in temp_files.values():
             if isinstance(files, list):
@@ -162,9 +180,11 @@ async def create_video(request: VideoGenerator):
                 os.unlink(files)
 
         os.remove(output_path)
+        logger.info("Files cleaned up successfully.")
         return JSONResponse(content=response, status_code=200)
 
     except Exception as e:
+        logger.error(f"Error creating video: {str(e)}")
         return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
 
 
@@ -186,8 +206,11 @@ class EachNecklaceVideoGeneratorRequest(BaseModel):
 
 @app.post("/createcombinedvideo/")
 async def create_video(request: EachNecklaceVideoGeneratorRequest):
+    logger.info(f"Creating video with request: {request.dict()}")
     start_time = time.time()
     try:
+        logger.info("Downloading images...")
+
         def process_images(image_list):
             if isinstance(image_list, list):
                 return [process_images(item) for item in image_list]
@@ -202,6 +225,7 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
         file_download_time = time.time() - start_time
 
         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
@@ -236,14 +260,17 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
             makeup_title=request.makeup_image_title
 
         )
+        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
+        logger.info(f"Video uploaded to Supabase: {url}")
 
         response = {
             "status": "success",
@@ -255,6 +282,7 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
                 "supabase_upload_time": supabase_upload_time
             }
         }
+        logger.info(f"Response: {response}")
 
         def cleanup_files(files):
             if isinstance(files, list):
@@ -266,6 +294,8 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
         if os.path.exists(output_path):
             os.remove(output_path)
 
+        logger.info("Files cleaned up successfully.")
+
         return JSONResponse(content=response, status_code=200)
 
     except Exception as e:
@@ -274,9 +304,11 @@ async def create_video(request: EachNecklaceVideoGeneratorRequest):
 
 @app.get("/resources")
 async def get_infromation():
+    logger.info("Getting resources information")
     music = os.listdir(RESOURCES_DIR + "/audio")
     fonts = os.listdir(RESOURCES_DIR + "/fonts")
     intro = os.listdir(RESOURCES_DIR + "/intro")
+    logger.info(f"Resources: {music}, {fonts}, {intro}")
 
     json = {"music": music, "fonts": fonts, "intro": intro}
     return JSONResponse(content=json, status_code=200)

resources/audio/ChrisHaugenMirage.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:230d11502724a50fe8d3b5d9d7e2c43d8432f970e6d185e8bfbfe7cac6d3e6fc
+size 7170226

resources/audio/ElectroIndianROYALASHUTOSH.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68127acdea914d66eca0e1e0bff00f4d525c64d7407b8cceaf3d3596dde90230
+size 8547402

resources/audio/IndianElectronicFUSIONFLOWASHUTOSH.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b90760624a483fcafa661626abe9553e03271cc96ea01823098663e5e5ec5091
+size 6686439

resources/audio/IndianInstrumentalKeralaMandolin.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e06e32f7af96a3be67e21f2ee925822af2bbecaf39c1fde7a61943c545263ec8
+size 5045949

resources/audio/LoveIndianCinematicBGM.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70707dedb298f220ebbc8869fb8cc9e49a3e72473808e3a20f17cc053e2f5a59
+size 11458488

resources/audio/SitarDholakIndianInstrumentalBackgroundMusic.mp3

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c66b23af609df9f54444b38b81343d46736e18ef76bccf29ae9a008d722b962e
+size 5053263

src/utils/logs.py

@@ -0,0 +1,27 @@
+"""
+project @ images_to_video
+created @ 2024-12-18
+author  @ github.com/ishworrsubedii
+"""
+import logging
+import os
+
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.INFO)
+
+log_dir = os.path.join(os.getcwd(), "logs")
+os.makedirs(log_dir, exist_ok=True)
+
+LOG_FILE = os.path.join(log_dir, "running_logs.log")
+
+logFormat = "[%(asctime)s: %(levelname)s: %(module)s: %(message)s]"
+logFormatter = logging.Formatter(fmt=logFormat, style="%")
+
+streamHandler = logging.StreamHandler()
+streamHandler.setFormatter(logFormatter)
+
+fileHandler = logging.FileHandler(filename=LOG_FILE)
+fileHandler.setFormatter(logFormatter)
+
+logger.addHandler(streamHandler)
+logger.addHandler(fileHandler)