merge pull request #144 from Xtao-Labs/thumbs

[feat] Show tg audio's thumbnail

driver/design/thumbnail.py

@@ -9,6 +9,14 @@ from PIL import (
 
 
 def changeImageSize(maxWidth, maxHeight, image):
+    if image.size[0] == image.size[1]:
+        # Does not change the scale of the orientation image and displays it centered.
+        # It may look even better
+        newImage = image.resize((maxHeight, maxHeight))
+        img = Image.new("RGBA", (maxWidth, maxHeight))
+        img.paste(newImage, (int((maxWidth - maxHeight) / 2), 0))
+        return img
+    else:
         widthRatio = maxWidth / image.size[0]
         heightRatio = maxHeight / image.size[1]
         newWidth = int(widthRatio * image.size[0])
@@ -18,13 +26,17 @@ def changeImageSize(maxWidth, maxHeight, image):
 
 
 async def thumb(thumbnail, title, userid, ctitle):
+    img_path = f"search/thumb{userid}.png"
+    if 'http' in thumbnail:
         async with aiohttp.ClientSession() as session:
             async with session.get(thumbnail) as resp:
                 if resp.status == 200:
-                f = await aiofiles.open(f"search/thumb{userid}.png", mode="wb")
+                    f = await aiofiles.open(img_path, mode="wb")
                     await f.write(await resp.read())
                     await f.close()
-    image1 = Image.open(f"search/thumb{userid}.png")
+    else:
+        img_path = thumbnail
+    image1 = Image.open(img_path)
     image2 = Image.open("driver/source/LightGreen.png")
     image3 = changeImageSize(1280, 720, image1)
     image4 = changeImageSize(1280, 720, image2)
@@ -49,6 +61,6 @@ async def thumb(thumbnail, title, userid, ctitle):
     )
     img.save(f"search/final{userid}.png")
     os.remove(f"search/temp{userid}.png")
-    os.remove(f"search/thumb{userid}.png")
+    os.remove(img_path)
     final = f"search/final{userid}.png"
     return final

program/music.py

@@ -147,16 +147,22 @@ async def play(c: Client, m: Message):
             suhu = await replied.reply("📥 downloading audio...")
             dl = await replied.download()
             link = replied.link
+            songname = "Audio"
+            thumbnail = f"{IMG_5}"
             try:
                 if replied.audio:
+                    if replied.audio.title:
                         songname = replied.audio.title[:70]
+                    else:
                         songname = replied.audio.file_name[:70]
+                    if replied.audio.thumbs:
+                        thumbnail = await c.download_media(replied.audio.thumbs[0].file_id)
                     duration = convert_seconds(replied.audio.duration)
                 elif replied.voice:
                     songname = "Voice Note"
                     duration = convert_seconds(replied.voice.duration)
             except BaseException:
-                songname = "Audio"
+                pass
 
             if chat_id in QUEUE:
                 await suhu.edit("🔄 Queueing Track...")
@@ -164,7 +170,6 @@ async def play(c: Client, m: Message):
                 ctitle = await CHAT_TITLE(gcname)
                 title = songname
                 userid = m.from_user.id
-                thumbnail = f"{IMG_5}"
                 image = await thumb(thumbnail, title, userid, ctitle)
                 pos = add_to_queue(chat_id, songname, dl, link, "Audio", 0)
                 requester = f"[{m.from_user.first_name}](tg://user?id={m.from_user.id})"
@@ -182,7 +187,6 @@ async def play(c: Client, m: Message):
                     ctitle = await CHAT_TITLE(gcname)
                     title = songname
                     userid = m.from_user.id
-                    thumbnail = f"{IMG_5}"
                     image = await thumb(thumbnail, title, userid, ctitle)
                     await suhu.edit("🔄 Joining Group Call...")
                     await music_on(chat_id)