Update srv.py

srv.py

@@ -1,12 +1,14 @@
-from flask import Flask, request, jsonify, send_file, render_template
+from flask import Flask, request, jsonify
 from yt_dlp import YoutubeDL
 import os
+import uuid
 
 app = Flask(__name__)
 
-@app.route('/')
-def home():
-    return render_template('index.html')
+# 存储下载文件的临时目录
+DOWNLOAD_DIR = 'downloads'
+if not os.path.exists(DOWNLOAD_DIR):
+    os.makedirs(DOWNLOAD_DIR)
 
 @app.route('/get-info', methods=['POST'])
 def get_info():
@@ -20,7 +22,7 @@ def get_info():
         ydl_opts = {
             'cookiefile': 'www.youtube.com_cookies.txt'
         }
-        
+
         with YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=False)
             return jsonify({
@@ -42,38 +44,34 @@ def download_audio():
         return jsonify({'error': 'URL is required'}), 400
 
     try:
+        # 生成唯一的文件名
+        unique_id = str(uuid.uuid4())
+        file_name = os.path.join(DOWNLOAD_DIR, f'{unique_id}.mp3')
+
         # Set up yt-dlp options to download only audio
         ydl_opts = {
             'format': 'bestaudio/best',  # Download the best available audio
-            'outtmpl': '%(title)s.%(ext)s',  # Output filename format
+            'outtmpl': file_name.rsplit('.', 1)[0] + '.%(ext)s',  # Output filename format
             'cookiefile': 'www.youtube.com_cookies.txt',  # Use cookies file if needed
             'postprocessors': [{
                 'key': 'FFmpegExtractAudio',  # Extract audio with FFmpeg
                 'preferredcodec': 'mp3',  # Convert to MP3
-                'preferredquality': '192',  # Set audio quality
+                'preferredquality': '128',  # Set audio quality
             }],
             'noplaylist': True,  # Avoid downloading entire playlists
         }
 
         with YoutubeDL(ydl_opts) as ydl:
             # Extract video info and download the audio
-            info = ydl.extract_info(url, download=True)
-            file_name = ydl.prepare_filename(info).rsplit(".", 1)[0] + ".mp3"
+            ydl.extract_info(url, download=True)
 
-        # Send the downloaded audio file to the client
-        return send_file(
-            file_name,
-            as_attachment=True,
-            download_name=os.path.basename(file_name)
-        )
+        # 返回可下载的地址
+        download_url = f'{request.host_url}{DOWNLOAD_DIR}/{os.path.basename(file_name)}'
+        return jsonify({'download_url': download_url})
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-    finally:
-        # Clean up the 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)
+    app.run(host='0.0.0.0', port=7860, debug=True)