Update DragMusic/utils/thumbnails.py

DragMusic/utils/thumbnails.py

@@ -7,6 +7,8 @@ import os
 import re
 import aiofiles
 import aiohttp
+import traceback
+import tempfile
 from PIL import Image, ImageDraw, ImageEnhance, ImageFilter, ImageFont
 from youtubesearchpython.__future__ import VideosSearch
 
@@ -109,8 +111,13 @@ def draw_text_with_shadow(background, draw, position, text, font, fill, shadow_o
 
 async def gen_thumb(videoid: str):
     try:
-        if os.path.isfile(f"cache/{videoid}_v4.png"):
-            return f"cache/{videoid}_v4.png"
+        # Create cache directory if it doesn't exist
+        cache_dir = "cache"
+        if not os.path.exists(cache_dir):
+            os.makedirs(cache_dir)
+            
+        if os.path.isfile(f"{cache_dir}/{videoid}_v4.png"):
+            return f"{cache_dir}/{videoid}_v4.png"
 
         url = f"https://www.youtube.com/watch?v={videoid}"
         results = VideosSearch(url, limit=1)
@@ -146,8 +153,6 @@ async def gen_thumb(videoid: str):
         
         async with aiohttp.ClientSession() as session:
             async with session.get(thumbnail) as resp:
-        
-                content = await resp.read()
                 if resp.status == 200:
                     content_type = resp.headers.get('Content-Type')
                     if 'jpeg' in content_type or 'jpg' in content_type:
@@ -158,14 +163,14 @@ async def gen_thumb(videoid: str):
                         logging.error(f"Unexpected content type: {content_type}")
                         return None
 
-                    filepath = f"cache/thumb{videoid}.png"
+                    filepath = f"{cache_dir}/thumb{videoid}.png"
                     f = await aiofiles.open(filepath, mode="wb")
                     await f.write(await resp.read())
                     await f.close()
                     # os.system(f"file {filepath}")
                     
         
-        image_path = f"cache/thumb{videoid}.png"
+        image_path = f"{cache_dir}/thumb{videoid}.png"
         youtube = Image.open(image_path)
         image1 = changeImageSize(1280, 720, youtube)
         
@@ -181,9 +186,9 @@ async def gen_thumb(videoid: str):
         background = Image.blend(background, gradient_image, alpha=0.2)
         
         draw = ImageDraw.Draw(background)
-        arial = ImageFont.truetype("AviaxMusic/assets/font2.ttf", 30)
-        font = ImageFont.truetype("AviaxMusic/assets/font.ttf", 30)
-        title_font = ImageFont.truetype("AviaxMusic/assets/font3.ttf", 45)
+        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)
 
 
         circle_thumbnail = crop_center_circle(youtube, 400, 20, start_gradient_color)
@@ -233,13 +238,13 @@ 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("AviaxMusic/assets/play_icons.png")
+        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)
 
-        os.remove(f"cache/thumb{videoid}.png")
+        os.remove(f"{cache_dir}/thumb{videoid}.png")
 
-        background_path = f"cache/{videoid}_v4.png"
+        background_path = f"{cache_dir}/{videoid}_v4.png"
         background.save(background_path)
         
         return background_path