Update directory

api/main.py

@@ -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)