Create app.py

app.py

@@ -0,0 +1,37 @@
+from flask import Flask, request, jsonify
+from pytube import YouTube
+import os
+
+app = Flask(__name__)
+
+@app.route('/', methods=['POST'])
+def transcribe_video():
+    # Get YouTube video link from request data
+    youtube_link = request.json['youtube_link']
+    
+    # Download the video
+    yt = YouTube(youtube_link)
+    video = yt.streams.filter().first()
+    destination = '.'
+    out_file = video.download(output_path=destination)
+
+    # Transcribe the video
+    base, ext = os.path.splitext(out_file)
+    aac_file_out = base + ".aac"
+    os.system(f"ffmpeg -i {out_file} -vn -acodec copy {aac_file_out}")
+    os.system(f"whisper {aac_file_out} --model medium --language Spanish --output_dir ./ --verbose False")
+
+    # Read the transcription
+    transcription_file = base + "_transcription.txt"
+    with open(transcription_file, 'r', encoding='utf-8') as file:
+        transcription = file.read()
+
+    # Clean up temporary files
+    os.remove(out_file)
+    os.remove(aac_file_out)
+    os.remove(transcription_file)
+
+    return jsonify({'transcription': transcription})
+
+if __name__ == '__main__':
+    app.run(debug=True)