mirror of
https://github.com/Nezumi-2711/json-server-vercel.git
synced 2026-09-22 13:38:40 +00:00
Implement youtube-api
This commit is contained in:
@@ -0,0 +1,53 @@
|
||||
from flask import Flask, request, jsonify
|
||||
from pytube import YouTube
|
||||
from flask_cors import CORS
|
||||
import youtube_dl
|
||||
|
||||
app = Flask(__name__)
|
||||
CORS(app, origins='*')
|
||||
|
||||
@app.route('/api/flask/convert', methods=['GET'])
|
||||
def convert():
|
||||
youtubeId = request.args.get('youtubeId')
|
||||
|
||||
if not youtubeId:
|
||||
return jsonify({"status": False, "error": "YouTube ID not specified"})
|
||||
|
||||
try:
|
||||
# Get YouTube video info using pytube
|
||||
video_url = f'https://www.youtube.com/watch?v={youtubeId}'
|
||||
yt = YouTube(video_url)
|
||||
|
||||
options = {
|
||||
'format': 'bestaudio/best',
|
||||
'extractaudio': True, # only keep the audio
|
||||
'audioformat': 'mp3', # convert to mp3
|
||||
'outtmpl': '%(id)s', # save file as video ID
|
||||
'noplaylist': True, # only download single song, not playlist
|
||||
}
|
||||
|
||||
with youtube_dl.YoutubeDL(options) as ydl:
|
||||
info_dict = ydl.extract_info(video_url, download=False)
|
||||
audio_url_youtubedl = info_dict['formats'][0]['url']
|
||||
|
||||
#only audio
|
||||
audio_url_pytube = yt.streams.get_audio_only().url
|
||||
# Get captions
|
||||
captions = yt.captions['en'] if 'en' in yt.captions else None
|
||||
captions_srt = captions.generate_srt_captions() if captions else "No captions available"
|
||||
captions_vtt = captions.generate_vtt_captions() if captions else "No captions available"
|
||||
|
||||
# Return the audio stream URL and captions
|
||||
return jsonify({
|
||||
"status": True,
|
||||
"audio_url_youtubedl": audio_url_youtubedl,
|
||||
"audio_url_pytube": audio_url_pytube,
|
||||
"captions_srt": captions_srt,
|
||||
"captions_vtt": captions_vtt
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
return jsonify({"status": False, "error": str(e)})
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host="0.0.0.0", port=8888)
|
||||
Reference in New Issue
Block a user