app.py

from flask import Flask, request, send_file, jsonify
import os
import subprocess

app = Flask(__name__)

@app.route('/download', methods=['GET'])
def download_video():
    try:
        # URLの取得
        video_url = request.args.get('url')
        if not video_url:
            return jsonify({"error": "URL is required"}), 400

        # ダウンロード先ディレクトリ
        output_path = "downloads"
        os.makedirs(output_path, exist_ok=True)  # ダウンロード先ディレクトリが存在しない場合、作成する

        # 出力ファイル名テンプレート
        output_file = os.path.join(output_path, '%(title)s.%(ext)s')

        # yt-dlpコマンドの設定
        cookies_file = 'cookies.txt'  # エクスポートしたクッキーを指定
        cmd = [
            'yt-dlp',
            '--cookies', cookies_file,
            '--output', output_file,  # 出力ファイルのパスを指定
            video_url
        ]

        # yt-dlpコマンドを実行
        result = subprocess.run(cmd, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

        # yt-dlpが実行された結果を表示（デバッグ用）
        print(result.stdout.decode())
        print(result.stderr.decode())

        # ダウンロードされたファイルのパスを取得
        downloaded_file = output_file % {'title': video_url.split('=')[-1], 'ext': 'mp4'}

        # ファイルをクライアントに送信
        return send_file(downloaded_file, as_attachment=True)

    except Exception as e:
        return jsonify({"error": str(e)}), 500

if __name__ == '__main__':
    app.run(host="0.0.0.0", port=7860, debug=True)