Update srv.py

srv.py

@@ -43,8 +43,6 @@ def download():
     if not url:
         return jsonify({'error': 'URL is required'}), 400
 
-    file_name = None
-
     try:
         if format_type == 'mp3':
             ydl_opts = {
@@ -57,18 +55,18 @@ 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]',
-                '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']) + '?'
+                    '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,
@@ -89,15 +87,15 @@ def download():
             file_name,
             as_attachment=True,
             download_name=os.path.basename(file_name),
-            mimetype='audio/mp3' if format_type == 'mp3' else 'video/mp4'
+            mimetype='video/mp4' if format_type == 'mp4' else 'audio/mp3'
         )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
     finally:
-        if file_name and os.path.exists(file_name):
+        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, threaded=True)
+       app.run(host='0.0.0.0', port=7860, debug=True,threaded=True)