Update srv.py

srv.py

@@ -1,27 +1,9 @@
-"""
-------------------------------------------------------------
-    YouTube Video & Audio Downloader API
-------------------------------------------------------------
-    ✨ Developed by: Mr. Asitha
-    ✅ Contact: +94743381623
-    📅 Created: 2025-03-18
-    🔗 Join WhatsApp Channel: https://whatsapp.com/channel/0029VaeyMWv3QxRu4hA6c33Z
-    🚀 Program: Flask API for Downloading YouTube Videos & MP3
-------------------------------------------------------------
-"""
-
 from flask import Flask, request, jsonify, send_file, render_template
 from yt_dlp import YoutubeDL
 import os
-import uuid
-import logging
-import time
 
 app = Flask(__name__)
 
-# Set up logging
-logging.basicConfig(level=logging.DEBUG)
-
 @app.route('/')
 def home():
     return render_template('index.html')
@@ -49,77 +31,65 @@ 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', 'mp4')  # Default format: MP4
+    quality = str(data.get('quality', 'best'))  # Default: best quality
 
     if not url:
         return jsonify({'error': 'URL is required'}), 400
 
-    max_retries = 3
-    retry_delay = 5  # seconds
-
-    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]'
-                }
-
-                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}"
+    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
+            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}"
 
-            return send_file(
-                file_name,
-                as_attachment=True,
-                download_name=os.path.basename(file_name)
-            )
+        # Send file to client
+        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
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
 
-        finally:
-            if 'file_name' in locals() and os.path.exists(file_name):
-                os.remove(file_name)
+    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, threaded=True)
+    app.run(host='0.0.0.0', port=7860, debug=True)