Update srv.py

srv.py

@@ -34,7 +34,7 @@ def get_info():
         return jsonify({'error': str(e)}), 500
 
 @app.route('/download', methods=['POST'])
-def download_audio():
+def download_video():
     data = request.json
     url = data.get('url')
 
@@ -42,22 +42,18 @@ def download_audio():
         return jsonify({'error': 'URL is required'}), 400
 
     try:
+        # Update the options to download video only
         ydl_opts = {
-            'format': '251/bestaudio',  # Format untuk audio terbaik
-            'outtmpl': '%(title)s.%(ext)s',
-            'cookiefile': 'www.youtube.com_cookies.txt',
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
+            'format': 'bestvideo/best',  # Download the best available video
+            'outtmpl': '%(title)s.%(ext)s',  # Save the file with title as filename
+            'cookiefile': 'www.youtube.com_cookies.txt',  # Use cookies (if needed)
         }
 
         with YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=True)
-            file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + ".mp3"
+            file_name = ydl.prepare_filename(info)
 
-        # Kirim file ke client
+        # Send the video file to the client
         return send_file(
             file_name,
             as_attachment=True,
@@ -68,9 +64,9 @@ def download_audio():
         return jsonify({'error': str(e)}), 500
 
     finally:
-        # Bersihkan file setelah dikirim
+        # Clean up the file after sending it
         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)
+       app.run(host='0.0.0.0', port=7860, debug=True)