Update app.py

app.py

@@ -21,6 +21,11 @@ from logging.handlers import RotatingFileHandler
 from werkzeug.utils import secure_filename
 from collections import deque
 
+from dotenv import load_dotenv
+# 嘗試加載 .env 文件
+if os.path.exists('.env'):
+    load_dotenv()
+    
 # 設置基本日誌配置
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -81,35 +86,15 @@ def send_sse_message(q, data):
 def clean_filename(filename):
     return ''.join(c for c in filename if c.isalnum() or c in (' ', '.', '_')).rstrip()
 
-def download_audio(youtube_url, save_directory, q):
+def download_audio(youtube_url, save_directory, q, ydl_opts):
     send_sse_message(q, {"status": "開始下載 YouTube 音頻..."})
-    unique_id = str(uuid.uuid4())[:8]  # 生成一個唯一的識別碼
+    unique_id = str(uuid.uuid4())[:8]
     output_filename = f"audio_{unique_id}"
     output_path = os.path.join(save_directory, output_filename)
 
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'mp3',
-            'preferredquality': '192',
-        }],
-        'ffmpeg_location': ffmpeg_path,
+    ydl_opts.update({
         'outtmpl': output_path + ".%(ext)s",
-        'quiet': True,
-        'no_warnings': True,
-        'no_check_certificate': True,
-        'ignoreerrors': True,
-        'nocheckcertificate': True,
-        'logtostderr': False,
-        'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36',
-        'socket_timeout': 30,  # 增加超時時間
-        'retries': 5,  # 失敗時的重試次數
-        'verbose': True,  # 啟用詳細日誌輸出，有助於調試
-        'extract_flat': 'in_playlist',
-        'youtube_include_dash_manifest': False,
-        'source_address': '0.0.0.0',  # 綁定到所有接口
-    }
+    })
 
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
@@ -317,6 +302,17 @@ def process_video(url_or_path, q, local_video_description=''):
         logger.info(f"開始處理視頻: {url_or_path}")
         save_directory = config['save_directory']
 
+        # 處理 YouTube cookies
+        cookies_content = os.environ.get('YOUTUBE_COOKIES')
+        cookies_file = os.path.join(os.path.dirname(__file__), 'youtube.com_cookies.txt')
+        if cookies_content:
+            cookies_content = cookies_content.strip('"').replace('\\n', '\n').replace('\\t', '\t')
+            with open(cookies_file, 'w') as f:
+                f.write(cookies_content)
+            logger.info("已創建 YouTube cookies 文件")
+        else:
+            logger.warning("未找到 YouTube cookies 環境變量")
+
         processed_description = ""
         if url_or_path.startswith('http'):
             logger.info("檢測到 YouTube URL，開始獲取視頻信息")
@@ -331,9 +327,7 @@ def process_video(url_or_path, q, local_video_description=''):
                 'outtmpl': os.path.join(save_directory, 'audio_%(id)s.%(ext)s'),
                 'quiet': True,
                 'no_warnings': True,
-                'no_check_certificate': True,
                 'ignoreerrors': True,
-                'nocheckcertificate': True,
                 'logtostderr': False,
                 'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36',
                 'socket_timeout': 30,
@@ -342,6 +336,7 @@ def process_video(url_or_path, q, local_video_description=''):
                 'extract_flat': 'in_playlist',
                 'youtube_include_dash_manifest': False,
                 'source_address': '0.0.0.0',
+                'cookiefile': cookies_file,  # 添加 cookies 文件
             }
 
             try:
@@ -365,7 +360,7 @@ def process_video(url_or_path, q, local_video_description=''):
                 processed_description = process_youtube_description(raw_description)
 
                 logger.info("開始下載 YouTube 音頻")
-                audio_path, video_title = download_audio(url_or_path, save_directory, q)
+                audio_path, video_title = download_audio(url_or_path, save_directory, q, ydl_opts)
             except yt_dlp.utils.DownloadError as e:
                 error_message = str(e)
                 if "Sign in to confirm you're not a bot" in error_message:
@@ -378,11 +373,14 @@ def process_video(url_or_path, q, local_video_description=''):
                 send_sse_message(q, {"status": f"錯誤：{str(e)}"})
                 logger.error(f"值錯誤: {str(e)}")
                 raise
+
         else:
+            # 本地文件處理邏輯保持不變
             logger.info("檢測到本地文件路徑，開始處理本地視頻")
             audio_path, video_title = process_local_video(url_or_path, save_directory, q)
             processed_description = local_video_description if local_video_description else "這是一個本地視頻文件，用戶沒有提供視頻描述。"
 
+        # 剩餘的處理邏輯保持不變
         if not audio_path or not os.path.exists(audio_path):
             raise FileNotFoundError(f"音頻文件不存在: {audio_path}")
 
@@ -452,6 +450,11 @@ def process_video(url_or_path, q, local_video_description=''):
     except Exception as e:
         logger.exception("處理視頻時發生錯誤")
         send_sse_message(q, {"status": f"錯誤: {str(e)}"})
+    finally:
+        # 處理完成後刪除 cookies 文件
+        if os.path.exists(cookies_file):
+            os.remove(cookies_file)
+            logger.info("已刪除 YouTube cookies 文件")
 
 
 # 在全局變量部分添加：