Update DragMusic/utils/thumbnails.py

DragMusic/utils/thumbnails.py

@@ -151,6 +151,10 @@ async def gen_thumb(videoid: str):
                 channel = "Unknown Channel"
 
         
+        if not thumbnail:
+            logging.error(f"No thumbnail URL found for video {videoid}")
+            return None
+            
         async with aiohttp.ClientSession() as session:
             async with session.get(thumbnail) as resp:
                 if resp.status == 200:
@@ -168,10 +172,22 @@ async def gen_thumb(videoid: str):
                     await f.write(await resp.read())
                     await f.close()
                     # os.system(f"file {filepath}")
+                else:
+                    logging.error(f"Failed to download thumbnail: HTTP {resp.status}")
+                    return None
                     
-        
         image_path = f"{cache_dir}/thumb{videoid}.png"
-        youtube = Image.open(image_path)
+        
+        # Verify the downloaded file exists and has content
+        if not os.path.exists(image_path) or os.path.getsize(image_path) == 0:
+            logging.error(f"Thumbnail file not found or empty: {image_path}")
+            return None
+            
+        try:
+            youtube = Image.open(image_path)
+        except Exception as img_error:
+            logging.error(f"Failed to open thumbnail image: {img_error}")
+            return None
         image1 = changeImageSize(1280, 720, youtube)
         
         image2 = image1.convert("RGBA")
@@ -186,9 +202,29 @@ async def gen_thumb(videoid: str):
         background = Image.blend(background, gradient_image, alpha=0.2)
         
         draw = ImageDraw.Draw(background)
-        arial = ImageFont.truetype("DragMusic/assets/font2.ttf", 30)
-        font = ImageFont.truetype("DragMusic/assets/font.ttf", 30)
-        title_font = ImageFont.truetype("DragMusic/assets/font3.ttf", 45)
+        
+        # Load fonts with error handling and fallbacks
+        try:
+            # Try to use absolute paths first
+            current_dir = os.path.dirname(os.path.abspath(__file__))
+            assets_dir = os.path.join(current_dir, "..", "assets")
+            
+            arial = ImageFont.truetype(os.path.join(assets_dir, "font2.ttf"), 30)
+            font = ImageFont.truetype(os.path.join(assets_dir, "font.ttf"), 30)
+            title_font = ImageFont.truetype(os.path.join(assets_dir, "font3.ttf"), 45)
+        except Exception as font_error:
+            logging.warning(f"Failed to load custom fonts: {font_error}, using default fonts")
+            try:
+                # Try relative paths as fallback
+                arial = ImageFont.truetype("DragMusic/assets/font2.ttf", 30)
+                font = ImageFont.truetype("DragMusic/assets/font.ttf", 30)
+                title_font = ImageFont.truetype("DragMusic/assets/font3.ttf", 45)
+            except Exception as fallback_error:
+                logging.warning(f"Failed to load fonts with relative paths: {fallback_error}, using system defaults")
+                # Use system default fonts as last resort
+                arial = ImageFont.load_default()
+                font = ImageFont.load_default()
+                title_font = ImageFont.load_default()
 
 
         circle_thumbnail = crop_center_circle(youtube, 400, 20, start_gradient_color)
@@ -238,9 +274,17 @@ async def gen_thumb(videoid: str):
         draw_text_with_shadow(background, draw, (text_x_position, 400), "00:00", arial, (255, 255, 255))
         draw_text_with_shadow(background, draw, (1080, 400), duration, arial, (255, 255, 255))
         
-        play_icons = Image.open("DragMusic/assets/play_icons.png")
-        play_icons = play_icons.resize((580, 62))
-        background.paste(play_icons, (text_x_position, 450), play_icons)
+        # Load play icons with error handling
+        try:
+            if 'assets_dir' in locals():
+                play_icons = Image.open(os.path.join(assets_dir, "play_icons.png"))
+            else:
+                play_icons = Image.open("DragMusic/assets/play_icons.png")
+            play_icons = play_icons.resize((580, 62))
+            background.paste(play_icons, (text_x_position, 450), play_icons)
+        except Exception as icon_error:
+            logging.warning(f"Failed to load play icons: {icon_error}, skipping icon")
+            # Continue without the play icons if they can't be loaded
 
         os.remove(f"{cache_dir}/thumb{videoid}.png")