Update DragMusic/utils/thumbnails.py

DragMusic/utils/thumbnails.py

@@ -1,13 +1,14 @@
+# ATLEAST GIVE CREDITS IF YOU STEALING :(((((((((((((((((((((((((((((((((((((
+# ELSE NO FURTHER PUBLIC THUMBNAIL UPDATES
+
 import random
 import logging
 import os
 import re
 import aiofiles
 import aiohttp
-import traceback
 from PIL import Image, ImageDraw, ImageEnhance, ImageFilter, ImageFont
 from youtubesearchpython.__future__ import VideosSearch
-import tempfile
 
 logging.basicConfig(level=logging.INFO)
 
@@ -108,9 +109,8 @@ def draw_text_with_shadow(background, draw, position, text, font, fill, shadow_o
 
 async def gen_thumb(videoid: str):
     try:
-        temp_dir = tempfile.gettempdir()
-        if os.path.isfile(os.path.join(temp_dir, f"{videoid}_v4.png")):
-            return os.path.join(temp_dir, f"{videoid}_v4.png")
+        if os.path.isfile(f"cache/{videoid}_v4.png"):
+            return f"cache/{videoid}_v4.png"
 
         url = f"https://www.youtube.com/watch?v={videoid}"
         results = VideosSearch(url, limit=1)
@@ -144,12 +144,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:
+        
+                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:
@@ -160,26 +158,16 @@ async def gen_thumb(videoid: str):
                         logging.error(f"Unexpected content type: {content_type}")
                         return None
 
-                    filepath = os.path.join(temp_dir, f"thumb{videoid}.png")
+                    filepath = f"cache/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 = os.path.join(temp_dir, f"thumb{videoid}.png")
-        
-        # Check if the thumbnail 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)
-            image1 = changeImageSize(1280, 720, youtube)
-        except Exception as e:
-            logging.error(f"Error opening or processing thumbnail image: {e}")
-            return None
+        image_path = f"cache/thumb{videoid}.png"
+        youtube = Image.open(image_path)
+        image1 = changeImageSize(1280, 720, youtube)
         
         image2 = image1.convert("RGBA")
         background = image2.filter(filter=ImageFilter.BoxBlur(20))
@@ -193,18 +181,9 @@ async def gen_thumb(videoid: str):
         background = Image.blend(background, gradient_image, alpha=0.2)
         
         draw = ImageDraw.Draw(background)
-        
-        # Load fonts with error handling
-        try:
-            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 e:
-            logging.error(f"Error loading fonts: {e}")
-            # Use default font as fallback
-            arial = ImageFont.load_default()
-            font = ImageFont.load_default()
-            title_font = ImageFont.load_default()
+        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)
 
 
         circle_thumbnail = crop_center_circle(youtube, 400, 20, start_gradient_color)
@@ -254,13 +233,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("DragMusic/assets/play_icons.png")
+        play_icons = Image.open("AviaxMusic/assets/play_icons.png")
         play_icons = play_icons.resize((580, 62))
         background.paste(play_icons, (text_x_position, 450), play_icons)
 
-        os.remove(os.path.join(temp_dir, f"thumb{videoid}.png"))
+        os.remove(f"cache/thumb{videoid}.png")
 
-        background_path = os.path.join(temp_dir, f"{videoid}_v4.png")
+        background_path = f"cache/{videoid}_v4.png"
         background.save(background_path)
         
         return background_path