From 4612c9fbe79fd464487177ec4fbb188adf8bc531 Mon Sep 17 00:00:00 2001
From: Tanjirotg <156563417+Tanjirotg@users.noreply.github.com>
Date: Sat, 19 Oct 2024 14:19:29 +0530
Subject: [PATCH] Done

---
 plugins/helpers/song.py | 54 +++++++++++++++++++++++++----------------
 1 file changed, 33 insertions(+), 21 deletions(-)

diff --git a/plugins/helpers/song.py b/plugins/helpers/song.py
index bc0ad1a..16888dc 100644
--- a/plugins/helpers/song.py
+++ b/plugins/helpers/song.py
@@ -1,32 +1,44 @@
-# credit by @Mrz_bots
-
-from HorridAPI import api
+from HorridAPI import Songmrz 
 import os
+from info import HORRI_API_KEY
 from pyrogram import Client, filters
 import aiohttp
 
+api_key = HORRI_API_KEY
+
+Horrid = Songmrz(api_key)
+
 @Client.on_message(filters.command("song"))
 async def song(client, message):
-    if len(message.text.split()) < 2:
+    if len(message.command) < 2:  # Changed to message.command to properly check for command arguments
         await message.reply("Give An Any Song Name!")
         return
 
-    query = " ".join(message.command[1:])    
-    data = api.song(query)
+    query = " ".join(message.command[1:])  
     m = await message.reply_text("**📥 Downloading...**")
-    url = data['url']
-    thumb_url = data['thumb']
-    title = data['title']
-    dura = data['duration']
-    songs = f"Title: {title}\nDuration: {dura}\nProvide by @Mrz_bots"
-    await m.edit("**📤 Uploading...**") 
-    await message.reply_photo(photo=thumb_url, caption=songs)
-    async with aiohttp.ClientSession() as session:
-        async with session.get(url) as resp:
-            with open(f"{title}.mp3", "wb") as f:
-                f.write(await resp.content.read())
-        async with session.get(thumb_url) as resp:
-            with open("thumb.jpg", "wb") as f:
-                f.write(await resp.content.read())
-    await message.reply_audio(f"{title}.mp3", thumb="thumb.jpg", title=title, caption=songs)
+
+    try:
+        data = Horrid.download(query)  # Ensure this line doesn't throw an error (handle exceptions if necessary)
+        url = data.url
+        thumb_url = data.thumb
+        title = data.title
+        dura = data.duration
+        songs = f"Title: {title}\nDuration: {dura}\nProvided by @Mrz_bots"
+
+        await m.edit("**📤 Uploading...**") 
+        await message.reply_photo(photo=thumb_url, caption=songs)
+
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url) as resp:
+                with open(f"{title}.mp3", "wb") as f:
+                    f.write(await resp.read())  # Use await resp.read() instead of await resp.content.read()
+            async with session.get(thumb_url) as resp:
+                with open("thumb.jpg", "wb") as f:
+                    f.write(await resp.read())  # Same change here
+
+        await message.reply_audio(f"{title}.mp3", thumb="thumb.jpg", title=title, caption=songs)
+    except Exception as e:
+        await m.edit("An error occurred while processing your request.")
+        print(f"Error: {e}")  # Log the error for debugging
+
     await m.delete()