Update srv.py

srv.py

@@ -2,7 +2,6 @@ from flask import Flask, request, jsonify, send_file, render_template
 from yt_dlp import YoutubeDL
 import os
 
-
 app = Flask(__name__)
 
 @app.route('/')
@@ -34,7 +33,6 @@ def get_info():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-@app.route('/download', methods=['POST'])
 @app.route('/download', methods=['POST'])
 def download():
     data = request.json
@@ -45,6 +43,8 @@ def download():
     if not url:
         return jsonify({'error': 'URL is required'}), 400
 
+    file_name = None
+
     try:
         if format_type == 'mp3':
             ydl_opts = {
@@ -57,7 +57,7 @@ def download():
                     'preferredquality': '192',
                 }],
             }
-        else: 
+        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]',
@@ -66,9 +66,9 @@ def download():
                 '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]',
-            }  # Added missing comma here
+            }
 
-            selected_quality = quality_map.get(quality, quality_map['best']) + '?'  # Added '?' here
+            selected_quality = quality_map.get(quality, quality_map['best']) + '?'
             
             ydl_opts = {
                 'format': selected_quality,
@@ -89,15 +89,15 @@ def download():
             file_name,
             as_attachment=True,
             download_name=os.path.basename(file_name),
-            mimetype='video/mp4' if format_type == 'mp4' else 'audio/mp3'
+            mimetype='audio/mp3' if format_type == 'mp3' else 'video/mp4'
         )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
     finally:
-        if 'file_name' in locals() and os.path.exists(file_name):
+        if file_name and os.path.exists(file_name):
             os.remove(file_name)
 
 if __name__ == '__main__':
-       app.run(host='0.0.0.0', port=7860, debug=True,threaded=True)
+    app.run(host='0.0.0.0', port=7860, debug=True, threaded=True)