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 traceback
 
 app = Flask(__name__)
 
@@ -44,31 +43,32 @@ def download():
     if not url:
         return jsonify({'error': 'URL is required'}), 400
 
-    file_name = None  # Initialize file_name to avoid reference issues
-
     try:
         if format_type == 'mp3':
             ydl_opts = {
                 'format': 'bestaudio/best',
                 'outtmpl': '%(title)s.%(ext)s',
                 'cookiefile': 'www.youtube.com_cookies.txt',
+                'headers': {
+                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'
+                },
                 'postprocessors': [{
                     'key': 'FFmpegExtractAudio',
                     'preferredcodec': 'mp3',
-                    'preferredquality': '192',
-                    'nopostoverwrites': True,  # Prevent overwriting existing files
+                    'preferredquality': '192',                
+                    'nopostoverwrites': True,
                 }],
             }
-        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]'
-            }
+                    '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'])
             
@@ -77,6 +77,10 @@ def download():
                 'outtmpl': '%(title)s.%(ext)s',
                 'merge_output_format': 'mp4',
                 'cookiefile': 'www.youtube.com_cookies.txt',
+                'headers': {
+                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'
+                },
+                'nopostoverwrites': True,
             }
 
         with YoutubeDL(ydl_opts) as ydl:
@@ -95,10 +99,10 @@ def download():
         )
 
     except Exception as e:
-        return jsonify({'error': str(e), 'traceback': traceback.format_exc()}), 500
+        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__':