Update srv.py

srv.py

@@ -1,107 +1,78 @@
-"""
-------------------------------------------------------------
-    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
+from flask import Flask, request, jsonify, send_file
+import yt_dlp
 import os
 
 app = Flask(__name__)
 
-@app.route('/')
-def home():
-    return render_template('index.html')
-
-@app.route('/get-info', methods=['GET', 'POST'])
-def get_info():
+@app.route('/download', methods=['GET', 'POST'])
+def download_video():
     data = request.json
     url = data.get('url')
-
-    if not url:
-        return jsonify({'error': 'URL is required'}), 400
+    quality = data.get('quality')
+    cookies = data.get('cookies')
+
+    if not url or not quality:
+        return jsonify({'error': 'URL and quality are required'}), 400
+
+    # Define quality options
+    quality_options = {
+        '144': '144p',
+        '240': '240p',
+        '360': '360p',
+        '480': '480p',
+        '720': '720p',
+        '1080': '1080p'
+    }
+
+    if quality not in quality_options:
+        return jsonify({'error': 'Invalid quality option'}), 400
+
+    # Set cookies for yt-dlp
+    cookie_file = 'cookies.txt'
+    with open(cookie_file, 'w') as f:
+        f.write(cookies)
+
+    # Download video options
+    ydl_opts = {
+        'format': f'bestvideo[height<={quality}]',
+        'outtmpl': '%(title)s.%(ext)s',
+        'cookiefile': cookie_file,
+        'noplaylist': True,
+    }
 
     try:
-        ydl_opts = {
-            'cookiefile': 'www.youtube.com_cookies.txt'
-        }
-        
-        with YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=False)
-            return jsonify({
-                'title': info['title'],
-                'thumbnail': info.get('thumbnail'),
-                'duration': info.get('duration'),
-                'channel': info.get('channel')
-            })
-
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            filename = ydl.prepare_filename(info)
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-@app.route('/download', methods=['GET', 'POST'])
-def download_media():
-    data = request.json
-    url = data.get('url')
-    format_type = data.get('format_type', 'mp3') 
-    quality = str(data.get('quality', 'best'))  
+    return send_file(filename, as_attachment=True)
+
+@app.route('/info', methods=['GET', 'POST'])
+def get_video_info():
+    url = request.args.get('url')
 
     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 = {
+            'cookiefile': 'cookies.txt'  # Optional: Use cookies if needed
+        }
 
-            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 yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=False)
+            video_info = {
+                'title': info.get('title'),
+                'thumbnail': info.get('thumbnail'),
+                'duration': info.get('duration'),
+                'channel': info.get('channel')
             }
-
-        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)
-        )
+            return jsonify(video_info)
 
     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)
-
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True, threaded=True)
+    app.run(debug=True)