Upload app.py

app.py

@@ -45,18 +45,47 @@ TEMP_DIR = '/tmp'
 
 # --- バックエンド処理関数 ---
 
+TEMP_DIR = '/tmp' # Spacesで書き込み可能な一時ディレクトリ
+
 def download_and_extract_audio(youtube_url):
     """
     yt-dlpを使って動画をダウンロードし、音声をMP3形式で抽出する。
-    成功したら (音声ファイルパス, 動画情報) のタプルを、失敗したら (None, None) を返す。
+    環境変数から読み込んだCookieを使用する。
     """
-    # <<< 修正 >>> 一時ディレクトリを使用
     output_dir = os.path.join(TEMP_DIR, 'downloads')
     os.makedirs(output_dir, exist_ok=True)
     logging.debug(f"音声保存ディレクトリ: {os.path.abspath(output_dir)}")
 
     output_template = os.path.join(output_dir, '%(id)s.%(ext)s')
 
+    # --- Cookie処理 ---
+    cookie_secret_name = 'YOUTUBE_COOKIES' # 手順2で登録したSecret名
+    cookie_content = os.getenv(cookie_secret_name)
+    cookie_file_path = None
+    temp_cookie_file_handle = None # finallyで閉じるため
+
+    if cookie_content:
+        try:
+            # 一時ファイルとしてCookieを書き出す
+            # delete=False にして、yt-dlpが読み込めるようにファイルを保持する
+            temp_cookie_file_handle = tempfile.NamedTemporaryFile(
+                mode='w', encoding='utf-8', dir=TEMP_DIR, suffix='.txt', delete=False
+            )
+            temp_cookie_file_handle.write(cookie_content)
+            cookie_file_path = temp_cookie_file_handle.name
+            temp_cookie_file_handle.close() # 書き込み後に一度閉じる（yt-dlpがアクセスできるように）
+            logging.info(f"環境変数 '{cookie_secret_name}' からCookieを一時ファイル '{cookie_file_path}' に書き出しました。")
+        except Exception as e:
+            logging.error(f"一時Cookieファイルの書き出し中にエラー: {e}", exc_info=True)
+            cookie_file_path = None # エラー時はCookieを使用しない
+            if temp_cookie_file_handle:
+                try:
+                    temp_cookie_file_handle.close() # エラー時も閉じる試み
+                except: pass
+    else:
+        logging.warning(f"環境変数 '{cookie_secret_name}' が未設定です。Cookieなしで試行します。")
+    # ---------------
+
     ydl_opts = {
         'format': 'bestaudio/best',
         'postprocessors': [{
@@ -67,7 +96,11 @@ def download_and_extract_audio(youtube_url):
         'outtmpl': output_template,
         'noplaylist': True,
         'logger': logging.getLogger('yt_dlp'),
-        'verbose': False,
+        'verbose': False, # デバッグ時にTrueにすると詳細ログが出力される
+        # --- Cookieオプションを追加 ---
+        'cookiefile': cookie_file_path, # yt-dlpにCookieファイルのパスを渡す
+        # --------------------------
+        # '--cookies-from-browser' はサーバー環境では使えないので注意
     }
 
     audio_file_path = None
@@ -75,65 +108,71 @@ def download_and_extract_audio(youtube_url):
 
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            logging.info(f"yt-dlp: {youtube_url} のダウンロードと音声抽出を開始")
+            logging.info(f"yt-dlp: {youtube_url} のダウンロードと音声抽出を開始 (Cookie使用: {'あり' if cookie_file_path else 'なし'})")
             info_dict = ydl.extract_info(youtube_url, download=True)
+
+            # ... (動画情報取得、ファイルパス特定、ファイル待機処理は変更なし) ...
             video_id = info_dict.get('id', 'unknown_id')
-            # 動画情報も返すようにする
             info_dict_result = {
                 'id': video_id,
                 'title': info_dict.get('title', f'動画 {video_id}'),
-                'thumbnail': info_dict.get('thumbnail'), # サムネイルURL
+                'thumbnail': info_dict.get('thumbnail'),
                 'uploader': info_dict.get('uploader'),
                 'duration': info_dict.get('duration'),
             }
             logging.info(f"yt-dlp: 動画情報取得完了 (ID: {video_id}, Title: {info_dict_result['title']})")
 
             base_filename = ydl.prepare_filename(info_dict)
-            # yt-dlp の prepare_filename は絶対パスを返すことがあるので、
-            # 確実に /tmp 配下のパスにするため、ファイル名だけ取得して結合する
-            # expected_mp3_path = os.path.splitext(base_filename)[0] + '.mp3'
             expected_mp3_filename = os.path.splitext(os.path.basename(base_filename))[0] + '.mp3'
             expected_mp3_path = os.path.join(output_dir, expected_mp3_filename)
             logging.debug(f"yt-dlp: 期待されるMP3ファイルパス -> {expected_mp3_path}")
 
-
             wait_time = 0
-            max_wait = 15 # 少し長めに待つ (15秒) / 環境によっては時間がかかる場合あり
+            max_wait = 15
             while not os.path.exists(expected_mp3_path) and wait_time < max_wait:
                 logging.debug(f"MP3ファイル待機中: {expected_mp3_path} (待機時間: {wait_time}秒)")
                 time.sleep(1)
                 wait_time += 1
 
             if os.path.exists(expected_mp3_path):
-                 audio_file_path = expected_mp3_path
-                 logging.info(f"yt-dlp: 音声抽出完了 -> {audio_file_path}")
-                 return audio_file_path, info_dict_result
+                audio_file_path = expected_mp3_path
+                logging.info(f"yt-dlp: 音声抽出完了 -> {audio_file_path}")
+                return audio_file_path, info_dict_result
             else:
-                 logging.warning(f"期待されたMP3ファイルが見つかりません: {expected_mp3_path}")
-                 # output_dir 内のファイルをリストアップして確認
-                 potential_files = [f for f in os.listdir(output_dir) if f.startswith(video_id) and f.endswith('.mp3')]
-                 if potential_files:
-                     # output_dirを基準に絶対パスを生成
-                     audio_file_path = os.path.join(output_dir, potential_files[0])
-                     logging.info(f"yt-dlp: 代替検索で見つかった音声ファイル -> {audio_file_path}")
-                     return audio_file_path, info_dict_result
-                 else:
+                logging.warning(f"期待されたMP3ファイルが見つかりません: {expected_mp3_path}")
+                potential_files = [f for f in os.listdir(output_dir) if f.startswith(video_id) and f.endswith('.mp3')]
+                if potential_files:
+                    audio_file_path = os.path.join(output_dir, potential_files[0])
+                    logging.info(f"yt-dlp: 代替検索で見つかった音声ファイル -> {audio_file_path}")
+                    return audio_file_path, info_dict_result
+                else:
                     logging.error("yt-dlp: 音声抽出後のMP3ファイル特定に失敗しました。")
                     logging.error(f"Downloads directory ({output_dir}) contents: {os.listdir(output_dir)}")
-                    return None, info_dict_result # ファイルパスはNoneだが情報は返す
+                    return None, info_dict_result
 
     except yt_dlp.utils.DownloadError as e:
         logging.error(f"yt-dlp ダウンロードエラー: {e}")
-        if "confirm your age" in str(e).lower():
-             logging.error("年齢確認が必要な動画の可能性があります。クッキーファイルの使用を検討してください。")
-        elif "video unavailable" in str(e).lower():
-             logging.error("動画が利用不可能なようです。")
-        elif "Private video" in str(e):
-             logging.error("非公開動画のようです。")
+        err_str = str(e).lower()
+        # エラーメッセージに基づいて、より具体的なログを出力
+        if "sign in to confirm" in err_str or "confirm your age" in err_str:
+            logging.error("ボット確認/年齢確認エラー。提供されたCookieが無効か期限切れの可能性があります。再度Cookieを取得してSecretを更新してください。")
+        elif "video unavailable" in err_str:
+            logging.error("動画が利用不可です。URLが正しいか確認してください。")
+        elif "private video" in err_str:
+            logging.error("非公開動画です。Cookieがその動画へのアクセス権を持っているか確認してください。")
+        # 他にも 'livestream'、'members-only' などのエラーがありうる
         return None, None
     except Exception as e:
         logging.error("yt-dlp: 音声抽出中に予期せぬエラーが発生しました。", exc_info=True)
         return None, None
+    finally:
+        # --- 一時Cookieファイルを削除 ---
+        if cookie_file_path and os.path.exists(cookie_file_path):
+            try:
+                os.remove(cookie_file_path)
+                logging.info(f"一時Cookieファイル {cookie_file_path} を削除しました。")
+            except OSError as rm_err:
+                logging.error(f"一時Cookieファイル {cookie_file_path} の削除に失敗: {rm_err}")
 
 def transcribe_audio(audio_path):
     """