Upload main.py

main.py

@@ -195,15 +195,17 @@ async def youtube_api(payload: YouTubeBase):
             status="False",
             randydev={"error": url}
         )
+    
     try:
+        time_second = time.time()
         if payload.only_audio:
             with YoutubeDL(YoutubeDriver.song_options()) as ytdl:
                 yt_data = ytdl.extract_info(url, download=True)
-                yt_file = ytdl.prepare_filename(yt_data)
+                yt_file = ytdl.prepare_filename(yt_data).replace('.webm', '.mp3')
                 ytdl.process_info(yt_data)
-            time_second = time.time()
-            with open(f"{yt_file}.mp3", "rb") as audio_file:
-                encoded_string = base64.b64encode(audio_file.read())
+            with open(yt_file, "rb") as audio_file:
+                encoded_string = base64.b64encode(audio_file.read()).decode('utf-8')
+            os.remove(yt_file)
             return SuccessResponse(
                 status="True",
                 randydev={
@@ -219,10 +221,10 @@ async def youtube_api(payload: YouTubeBase):
         else:
             with YoutubeDL(YoutubeDriver.video_options()) as ytdl:
                 yt_data = ytdl.extract_info(url, download=True)
-                yt_file = yt_data["id"]
-            time_second = time.time()
-            with open(f"{yt_file}.mp4", "rb") as video_file:
-                encoded_string = base64.b64encode(video_file.read())
+                yt_file = ytdl.prepare_filename(yt_data).replace('.webm', '.mp4')
+            with open(yt_file, "rb") as video_file:
+                encoded_string = base64.b64encode(video_file.read()).decode('utf-8')
+            os.remove(yt_file)
             return SuccessResponse(
                 status="True",
                 randydev={