Update srv.py

srv.py

@@ -34,62 +34,68 @@ def get_info():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/download', methods=['POST'])
-def download_media():
+def download():
     data = request.json
     url = data.get('url')
-    format_type = data.get('format', 'mp4')  # Default format: MP4
-    quality = str(data.get('quality', 'best'))  # Default: best quality
+    format_type = data.get('format', 'mp3')
+    quality = data.get('quality', 'best')
 
     if not url:
         return jsonify({'error': 'URL is required'}), 400
 
     try:
-        if format_type == 'mp4':
-            quality_map = {
-                '144': 'bv*[height=144][ext=mp4]+ba[ext=m4a]/b[height=144][ext=mp4]',
-                '360': 'bv*[height=360][ext=mp4]+ba[ext=m4a]/b[height=360][ext=mp4]',
-                '480': 'bv*[height=480][ext=mp4]+ba[ext=m4a]/b[height=480][ext=mp4]',
-                '720': 'bv*[height=720][ext=mp4]+ba[ext=m4a]/b[height=720][ext=mp4]',
-                '1080': 'bv*[height=1080][ext=mp4]+ba[ext=m4a]/b[height=1080][ext=mp4]',
-                'best': 'bv*[ext=mp4]+ba[ext=m4a]/b[ext=mp4]'  # Best available
-            }
-
-            ydl_opts = {
-                'format': quality_map.get(quality, quality_map['best']),
-                'merge_output_format': 'mp4',
-                'outtmpl': '%(title)s.%(ext)s',
-                'cookiefile': 'www.youtube.com_cookies.txt'
-            }
-        else:  # Default: MP3
+        if format_type == 'mp3':
             ydl_opts = {
-                  'format': 'bestaudio',
+                'format': 'bestaudio/best',
                 'outtmpl': '%(title)s.%(ext)s',
                 'cookiefile': 'www.youtube.com_cookies.txt',
                 'postprocessors': [{
                     'key': 'FFmpegExtractAudio',
                     'preferredcodec': 'mp3',
-                    'preferredquality': '64',
+                    'preferredquality': '192',
                 }],
             }
+        else: 
+            quality_map = {
+                    '144': 'bv*[height=144][ext=mp4]+ba[ext=m4a]/b[height=144][ext=mp4]',
+                    '240': 'bv*[height=240][ext=mp4]+ba[ext=m4a]/b[height=240][ext=mp4]',
+                    '360': 'bv*[height=360][ext=mp4]+ba[ext=m4a]/b[height=360][ext=mp4]',
+                    '480': 'bv*[height=480][ext=mp4]+ba[ext=m4a]/b[height=480][ext=mp4]',
+                    '720': 'bv*[height=720][ext=mp4]+ba[ext=m4a]/b[height=720][ext=mp4]',
+                    '1080': 'bv*[height=1080][ext=mp4]+ba[ext=m4a]/b[height=1080][ext=mp4]',
+                    'best': 'bv*[ext=mp4]+ba[ext=m4a]/b[ext=mp4]'
+                }
+
+            selected_quality = quality_map.get(quality, quality_map['best'])
+            
+            ydl_opts = {
+                'format': selected_quality,
+                'outtmpl': '%(title)s.%(ext)s',
+                'merge_output_format': 'mp4',
+                'cookiefile': 'www.youtube.com_cookies.txt',
+            }
+
         with YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=True)
-            file_ext = "mp4" if format_type == "mp4" else "mp3"
-            file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + f".{file_ext}"
+            
+            if format_type == 'mp3':
+                file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + ".mp3"
+            else:
+                file_name = ydl.prepare_filename(info)
 
-        # Send file to client
         return send_file(
             file_name,
             as_attachment=True,
-            download_name=os.path.basename(file_name)
+            download_name=os.path.basename(file_name),
+            mimetype='video/mp4' if format_type == 'mp4' else 'audio/mp3'
         )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
     finally:
-        # Delete 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)
+       app.run(host='0.0.0.0', port=7860, debug=True)