Update app.py

app.py

@@ -0,0 +1,63 @@
+from flask import Flask, request, send_file, jsonify
+from pytube import YouTube
+import os
+
+app = Flask(__name__)
+
+@app.route('/watch', methods=['GET'])
+def download_video():
+    # URLパラメータを取得
+    video_id = request.args.get('id')
+    file_type = request.args.get('file', 'mp4')  # デフォルトは mp4
+    quality = request.args.get('quality', '720p')  # デフォルトは 720p
+
+    if not video_id:
+        return jsonify({'error': 'Missing video ID'}), 400
+
+    try:
+        # YouTube URLを作成
+        youtube_url = f'https://www.youtube.com/watch?v={video_id}'
+        yt = YouTube(youtube_url)
+
+        # ファイル保存用ディレクトリ
+        download_dir = "downloads"
+        if not os.path.exists(download_dir):
+            os.makedirs(download_dir)
+
+        # 動画をダウンロード (MP4)
+        if file_type == 'mp4':
+            stream = yt.streams.filter(res=quality, file_extension='mp4').first()
+            if not stream:
+                return jsonify({'error': f'Video with quality {quality} not found'}), 404
+
+            file_path = stream.download(output_path=download_dir)
+        
+        # オーディオをダウンロード (MP3)
+        elif file_type == 'mp3':
+            stream = yt.streams.filter(only_audio=True).first()
+            if not stream:
+                return jsonify({'error': 'Audio stream not found'}), 404
+
+            temp_file_path = stream.download(output_path=download_dir)
+            file_path = os.path.splitext(temp_file_path)[0] + '.mp3'
+
+            # MP3に変換
+            os.rename(temp_file_path, file_path)
+        
+        else:
+            return jsonify({'error': 'Invalid file type. Use "mp4" or "mp3"'}), 400
+
+        # ファイルをクライアントに送信
+        return send_file(file_path, as_attachment=True)
+
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+    finally:
+        # クリーンアップ: 使用後にファイルを削除
+        if os.path.exists(file_path):
+            os.remove(file_path)
+
+
+if __name__ == '__main__':
+    app.run(debug=True)