Update srv.py

srv.py

@@ -1,6 +1,7 @@
 from flask import Flask, request, jsonify, send_file, render_template
 from yt_dlp import YoutubeDL
 import os
+import subprocess
 
 app = Flask(__name__)
 
@@ -42,31 +43,50 @@ def download_video():
         return jsonify({'error': 'URL is required'}), 400
 
     try:
-        # Update the options to download video only
+        # Download video and audio streams
         ydl_opts = {
-            'format': 'bestvideo/best',  # Download the best available video
-            'outtmpl': '%(title)s.%(ext)s',  # Save the file with title as filename
-            'cookiefile': 'www.youtube.com_cookies.txt',  # Use cookies (if needed)
+            '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
         }
 
         with YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=True)
-            file_name = ydl.prepare_filename(info)
+            video_file = ydl.prepare_filename(info)
 
-        # Send the video file to the client
+        # 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
         return send_file(
-            file_name,
+            output_file,
             as_attachment=True,
-            download_name=os.path.basename(file_name)
+            download_name=os.path.basename(output_file)
         )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
     finally:
-        # Clean up the file after sending it
-        if 'file_name' in locals() and os.path.exists(file_name):
-            os.remove(file_name)
+        # Clean up the merged file after sending it
+        if 'output_file' in locals() and os.path.exists(output_file):
+            os.remove(output_file)
 
 if __name__ == '__main__':
-       app.run(host='0.0.0.0', port=7860, debug=True)
+    app.run(host='0.0.0.0', port=7860, debug=True)