Update srv.py

srv.py

@@ -13,9 +13,14 @@
 from flask import Flask, request, jsonify, send_file, render_template
 from yt_dlp import YoutubeDL
 import os
+import uuid
+import logging
 
 app = Flask(__name__)
 
+# Set up logging
+logging.basicConfig(level=logging.DEBUG)
+
 @app.route('/')
 def home():
     return render_template('index.html')
@@ -43,64 +48,77 @@ def get_info():
             })
 
     except Exception as e:
+        app.logger.error(f"Error fetching video info: {str(e)}")
         return jsonify({'error': str(e)}), 500
 
 @app.route('/download', methods=['POST'])
 def download_media():
     data = request.json
     url = data.get('url')
-    format_type = data.get('format_type', 'mp3') 
-    quality = str(data.get('quality', 'best'))  
+    format_type = data.get('format_type', 'mp3')
+    quality = str(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]',
-                '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]'  
-            }
-
-            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: 
-            ydl_opts = {
-                  'format': 'bestaudio',
-                'outtmpl': '%(title)s.%(ext)s',
-                'cookiefile': 'www.youtube.com_cookies.txt',
-                'postprocessors': [{
-                    'key': 'FFmpegExtractAudio',
-                    'preferredcodec': 'mp3',
-                    'preferredquality': '64',
-                }],
-            }
-        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}"
+    max_retries = 3
+    retry_delay = 5  # seconds
 
-        return send_file(
-            file_name,
-            as_attachment=True,
-            download_name=os.path.basename(file_name)
-        )
+    for attempt in range(max_retries):
+        try:
+            unique_id = str(uuid.uuid4())  # Generate a unique ID for the file
+            if format_type == 'mp4':
+                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]'
+                }
 
-    except Exception as e:
-        return jsonify({'error': str(e)}), 500
+                ydl_opts = {
+                    'format': quality_map.get(quality, quality_map['best']),
+                    'merge_output_format': 'mp4',
+                    'outtmpl': f'%(title)s_{unique_id}.%(ext)s',  # Unique filename
+                    'cookiefile': 'www.youtube.com_cookies.txt'
+                }
+            else:
+                ydl_opts = {
+                    'format': 'bestaudio',
+                    'outtmpl': f'%(title)s_{unique_id}.%(ext)s',  # Unique filename
+                    'cookiefile': 'www.youtube.com_cookies.txt',
+                    'postprocessors': [{
+                        'key': 'FFmpegExtractAudio',
+                        'preferredcodec': 'mp3',
+                        'preferredquality': '64',
+                    }],
+                }
+
+            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}"
+
+            return send_file(
+                file_name,
+                as_attachment=True,
+                download_name=os.path.basename(file_name)
+            )
+
+        except Exception as e:
+            if attempt < max_retries - 1:
+                app.logger.warning(f"Attempt {attempt + 1} failed. Retrying in {retry_delay} seconds...")
+                time.sleep(retry_delay)
+                continue
+            else:
+                app.logger.error(f"Error downloading media: {str(e)}")
+                return jsonify({'error': str(e)}), 500
 
-    finally:
-        if 'file_name' in locals() and os.path.exists(file_name):
-            os.remove(file_name)
+        finally:
+            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)