Files
json-server-vercel/api/main.py
T
2024-01-15 09:13:25 +07:00

41 lines
1.3 KiB
Python

from flask import Flask, request, jsonify
from pytube import YouTube
from flask_cors import CORS
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)
# Get the audio stream URL
audio_stream_url = yt.streams.filter(only_audio=True).first().url
# Get all audio stream URLs
all_audio_stream_urls = [stream.url for stream in yt.streams.filter(only_audio=True)]
# Get video stream URL
video_stream_url = yt.streams.filter(file_extension='mp4').first().url
# Return the audio stream URL and captions
return jsonify({
"status": True,
"audio_stream_url": audio_stream_url,
"all_audio_stream_urls": all_audio_stream_urls,
"video_stream_url": video_stream_url
})
except Exception as e:
return jsonify({"status": False, "error": str(e)})
if __name__ == '__main__':
app.run(host="0.0.0.0", port=8888)