Update srv.py

srv.py

@@ -1,78 +1,76 @@
-from flask import Flask, request, jsonify, send_file
-import yt_dlp
+from flask import Flask, request, jsonify, send_file, render_template
+from yt_dlp import YoutubeDL
 import os
 
 app = Flask(__name__)
 
-@app.route('/download', methods=['GET', 'POST'])
-def download_video():
+@app.route('/')
+def home():
+    return render_template('index.html')
+
+@app.route('/info', methods=['POST'])
+def info():
     data = request.json
     url = data.get('url')
-    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,
-    }
+    if not url:
+        return jsonify({'error': 'URL is required'}), 400
 
     try:
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=True)
-            filename = ydl.prepare_filename(info)
+        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')
+            })
+
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-    return send_file(filename, as_attachment=True)
-
-@app.route('/info', methods=['GET', 'POST'])
-def get_video_info():
-    url = request.args.get('url')
+@app.route('/ytmp3', methods=['POST'])
+def ytmp3():
+    data = request.json
+    url = data.get('url')
 
     if not url:
         return jsonify({'error': 'URL is required'}), 400
 
     try:
         ydl_opts = {
-            'cookiefile': 'cookies.txt'  # Optional: Use cookies if needed
+            'format': '251/bestaudio',  # Format untuk audio terbaik
+            'outtmpl': '%(title)s.%(ext)s',
+            'cookiefile': 'www.youtube.com_cookies.txt',
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '128',
+            }],
         }
 
-        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')
-            }
-            return jsonify(video_info)
+        with YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + ".mp3"
+
+        # Kirim file ke client
+        return send_file(
+            file_name,
+            as_attachment=True,
+            download_name=os.path.basename(file_name)
+        )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
+    finally:
+        # Bersihkan file setelah dikirim
+        if 'file_name' in locals() and os.path.exists(file_name):
+            os.remove(file_name)
+
 if __name__ == '__main__':
-    app.run(debug=True)
+       app.run(host='0.0.0.0', port=7860, debug=True)