Update Akeno/plugins/youtube.py

Akeno/plugins/youtube.py

@@ -1,7 +1,8 @@
 import os
 import time
+
 import requests
-from pyrogram.types import *
+from pyrogram.types import Message
 from pyrogram import Client, filters
 from youtube_search import YoutubeSearch
 from yt_dlp import YoutubeDL
@@ -16,16 +17,6 @@ from Akeno.utils.prefixprem import command
 from config import *
 
 custom_loading = "<emoji id=5974235702701853774>🗿</emoji>"
-custom_head = "<emoji id=5316919120149619748>🤔</emoji>"
-custom_mata = "<emoji id=5253959125838090076>🤔</emoji>"
-custom_uptime = "<emoji id=5451732530048802485>🤔</emoji>"
-
-def time_to_seconds(time):
-    stringt = str(time)
-    return sum(
-        int(x) * 60**i
-        for i, x in enumerate(reversed(stringt.split(":")))
-    )
 
 @Akeno(
     ~filters.scheduled
@@ -64,21 +55,21 @@ async def youtube_search_audio(client: Client, message: Message):
         response = requests.get(f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg")
         with open(f"{yt_file}.jpg", "wb") as f:
             f.write(response.content)
-        if client.me.is_premium:
-            caption = f"**{custom_head}𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**{custom_mata}𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**{custom_uptime}𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        else:
-            caption = f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        duration_rc = time_to_seconds(yt_data["duration"])
-        await pro.edit_media(
-            media=InputMediaAudio(
-                f"{yt_file}.mp3",
-                caption=caption,
-                duration=duration_rc,
-                performer="[Akeno UB]",
-                title=yt_data["title"],
-                thumb=f"{yt_file}.jpg",
-            )
+        await message.reply_audio(
+            f"{yt_file}.mp3",
+            caption=f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`",
+            duration=int(yt_data["duration"]),
+            performer="[Akeno UB]",
+            title=yt_data["title"],
+            thumb=f"{yt_file}.jpg",
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text,
+            ),
         )
+        await pro.delete()
     except Exception as e:
         return await pro.edit_text(f"**🍀 Audio not Downloaded:** `{e}`")
     try:
@@ -120,20 +111,21 @@ async def youtube_audio(client: Client, message: Message):
         response = requests.get(f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg")
         with open(f"{yt_file}.jpg", "wb") as f:
             f.write(response.content)
-        if client.me.is_premium:
-            caption = f"**{custom_head}𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**{custom_mata}𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**{custom_uptime}𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        else:
-            caption = f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        await pro.edit_media(
-            media=InputMediaAudio(
-                f"{yt_file}.mp3",
-                caption=caption,
-                duration=int(yt_data["duration"]),
-                performer="[Akeno UB]",
-                title=yt_data["title"],
-                thumb=f"{yt_file}.jpg",
-            )
+        await message.reply_audio(
+            f"{yt_file}.mp3",
+            caption=f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`",
+            duration=int(yt_data["duration"]),
+            performer="[Akeno UB]",
+            title=yt_data["title"],
+            thumb=f"{yt_file}.jpg",
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text,
+            ),
         )
+        await pro.delete()
     except Exception as e:
         return await pro.edit_text(f"**🍀 Audio not Downloaded:** `{e}`")
     try:
@@ -179,18 +171,19 @@ async def ytvideo_search(client: Client, message: Message):
         response = requests.get(f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg")
         with open(f"{yt_file}.jpg", "wb") as f:
             f.write(response.content)
-        if client.me.is_premium:
-            caption = f"**{custom_head}𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**{custom_mata}𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**{custom_uptime}𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        else:
-            caption = f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        await pro.edit_media(
-            media=InputMediaVideo(
-                f"{yt_file}.mp4",
-                caption=caption,
-                duration=int(yt_data["duration"]),
-                thumb=f"{yt_file}.jpg",
-            )
+        await message.reply_video(
+            f"{yt_file}.mp4",
+            caption=f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`",
+            duration=int(yt_data["duration"]),
+            thumb=f"{yt_file}.jpg",
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text,
+            ),
         )
+        await pro.delete()
     except Exception as e:
         return await pro.edit_text(f"**🍀 Video not Downloaded:** `{e}`")
     try:
@@ -232,18 +225,19 @@ async def ytvideo(client: Client, message: Message):
         response = requests.get(f"https://i.ytimg.com/vi/{yt_data['id']}/hqdefault.jpg")
         with open(f"{yt_file}.jpg", "wb") as f:
             f.write(response.content)
-        if client.me.is_premium:
-            caption = f"**{custom_head}𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**{custom_mata}𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**{custom_uptime}𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        else:
-            caption = f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`"
-        await pro.edit_media(
-            media=InputMediaVideo(
-                f"{yt_file}.mp4",
-                caption=caption,
-                duration=int(yt_data["duration"]),
-                thumb=f"{yt_file}.jpg",
-            )
+        await message.reply_video(
+            f"{yt_file}.mp4",
+            caption=f"**🎧 𝖳𝗂𝗍𝗅𝖾:** {yt_data['title']} \n\n**👀 𝖵𝗂𝖾𝗐𝗌:** `{yt_data['view_count']}` \n**⌛ 𝖣𝗎𝗋𝖺𝗍𝗂𝗈𝗇:** `{secs_to_mins(int(yt_data['duration']))}`",
+            duration=int(yt_data["duration"]),
+            thumb=f"{yt_file}.jpg",
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text,
+            ),
         )
+        await pro.delete()
     except Exception as e:
         return await pro.edit_text(f"**🍀 Video not Downloaded:** `{e}`")
     try: