Update srv.py

srv.py

@@ -84,5 +84,43 @@ def download_audio():
         return jsonify({'error': str(e)}), 500
 
 
+@app.route('/audio', methods=['POST'])
+def download_high_audio():
+    data = request.json
+    url = data.get('url')
+
+    if not url:
+        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': 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': '320',  # Set audio quality
+            }],
+            'noplaylist': True,  # Avoid downloading entire playlists
+        }
+
+        with YoutubeDL(ydl_opts) as ydl:
+            # Extract video info and download the audio
+            ydl.extract_info(url, download=True)
+
+        # 返回可下载的地址
+        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
+
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)