Update srv.py

srv.py

@@ -1,7 +1,6 @@
 from flask import Flask, request, jsonify, send_file, render_template
 from yt_dlp import YoutubeDL
 import os
-import subprocess
 
 app = Flask(__name__)
 
@@ -35,7 +34,7 @@ def get_info():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/download', methods=['POST'])
-def download_video():
+def download_audio():
     data = request.json
     url = data.get('url')
 
@@ -43,50 +42,38 @@ def download_video():
         return jsonify({'error': 'URL is required'}), 400
 
     try:
-        # Download video and audio streams
+        # Set up yt-dlp options to download only audio
         ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',  # Download the best video and audio
-            'outtmpl': '%(title)s.%(ext)s',  # Save file with title as filename
-            'cookiefile': 'www.youtube.com_cookies.txt',  # Use cookies if needed
-            'noplaylist': True,  # Avoid downloading playlists
+            'format': 'bestaudio/best',  # Download the best available audio
+            'outtmpl': '%(title)s.%(ext)s',  # Output filename format
+            'cookiefile': 'www.youtube.com_cookies.txt',  # Use cookies file if needed
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',  # Extract audio with FFmpeg
+                'preferredcodec': 'mp3',  # Convert to MP3
+                'preferredquality': '192',  # Set audio quality
+            }],
+            'noplaylist': True,  # Avoid downloading entire playlists
         }
 
         with YoutubeDL(ydl_opts) as ydl:
+            # Extract video info and download the audio
             info = ydl.extract_info(url, download=True)
-            video_file = ydl.prepare_filename(info)
+            file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + ".mp3"
 
-        # Set the output file for merged video and audio
-        output_file = video_file.rsplit('.', 1)[0] + '_merged.mp4'
-
-        # Use FFmpeg to merge audio and video files
-        video_file_path = video_file
-        audio_file_path = video_file.rsplit('.', 1)[0] + '.m4a'  # Assuming audio is in .m4a format
-
-        # Merge audio and video using FFmpeg
-        ffmpeg_command = [
-            'ffmpeg', '-i', video_file_path, '-i', audio_file_path,
-            '-c:v', 'copy', '-c:a', 'aac', '-strict', 'experimental', output_file
-        ]
-        subprocess.run(ffmpeg_command, check=True)
-
-        # Remove the original separate video and audio files to save space
-        os.remove(video_file_path)
-        os.remove(audio_file_path)
-
-        # Send the merged video file to the client
+        # Send the downloaded audio file to the client
         return send_file(
-            output_file,
+            file_name,
             as_attachment=True,
-            download_name=os.path.basename(output_file)
+            download_name=os.path.basename(file_name)
         )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
     finally:
-        # Clean up the merged file after sending it
-        if 'output_file' in locals() and os.path.exists(output_file):
-            os.remove(output_file)
+        # Clean up the file after sending
+        if 'file_name' in locals() and os.path.exists(file_name):
+            os.remove(file_name)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)