vikarshana commited on
Commit
5833d54
·
verified ·
1 Parent(s): 83c91dd

Update srv.py

Browse files
Files changed (1) hide show
  1. srv.py +2 -4
srv.py CHANGED
@@ -1,7 +1,5 @@
1
  from flask import Flask, request, jsonify, send_file, render_template
2
  from yt_dlp import YoutubeDL
3
- import random
4
- import string
5
  import os
6
 
7
  app = Flask(__name__)
@@ -81,9 +79,9 @@ def download():
81
  info = ydl.extract_info(url, download=True)
82
 
83
  if format_type == 'mp3':
84
- file_name = ''.join(random.choices(string.ascii_letters + string.digits, k=19))
85
  else:
86
- file_name = ''.join(random.choices(string.ascii_letters + string.digits, k=20))
87
 
88
  return send_file(
89
  file_name,
 
1
  from flask import Flask, request, jsonify, send_file, render_template
2
  from yt_dlp import YoutubeDL
 
 
3
  import os
4
 
5
  app = Flask(__name__)
 
79
  info = ydl.extract_info(url, download=True)
80
 
81
  if format_type == 'mp3':
82
+ file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + ".mp3"
83
  else:
84
+ file_name = ydl.prepare_filename(info)
85
 
86
  return send_file(
87
  file_name,