Simplify YouTube functionality, remove YouTube API

app.py

@@ -14,7 +14,8 @@ import traceback # For printing full errors
 import platform
 import re
 import subprocess
-import youtube_api # 已安裝相關依賴，可以使用
+# 移除 YouTube API 相關導入，回到直接使用 yt-dlp
+# import youtube_api
 
 # --- 硬體檢查函數 ---
 def get_hardware_info():
@@ -140,13 +141,8 @@ def update_download_file(filepath):
     return None
 
 # --- YouTube 音訊處理 ---
-def process_youtube_url(youtube_url, user_api_key=None):
-    """處理 YouTube URL，下載音訊並返回播放器和下載按鈕的更新
-
-    Args:
-        youtube_url: YouTube 視頻 URL
-        user_api_key: 用戶輸入的 API 金鑰（可選）
-    """
+def process_youtube_url(youtube_url):
+    """處理 YouTube URL，下載音訊並返回播放器和下載按鈕的更新"""
     if not youtube_url or not youtube_url.strip():
         return gr.update(visible=False, value=None), gr.update(visible=False, value=None)
 
@@ -154,51 +150,80 @@ def process_youtube_url(youtube_url, user_api_key=None):
     import os
     is_spaces = os.environ.get("SPACE_ID") is not None
 
-    # 優先使用用戶輸入的 API 金鑰，如果沒有則使用環境變量中的 API 金鑰
-    youtube_api_key = user_api_key if user_api_key and user_api_key.strip() else os.environ.get("YOUTUBE_API_KEY")
-
     try:
-        # 如果有 API 金鑰，設置它
-        if youtube_api_key:
-            youtube_api.set_api_key(youtube_api_key)
-            print(f"Using YouTube Data API with {'user-provided' if user_api_key else 'environment'} API key")
-        else:
-            print("No YouTube API key found, falling back to direct download")
-
         # 處理 YouTube URL
         print(f"Processing YouTube URL: {youtube_url}")
 
-        # 檢查是否是在啟動時自動調用（緩存範例）
-        is_startup = not hasattr(youtube_api_key_input, 'value') if 'youtube_api_key_input' in globals() else False
-
-        if is_spaces and not youtube_api_key and not is_startup:
-            # 在 Spaces 環境中且沒有 API 金鑰，顯示警告（但不是在啟動時）
-            print("Warning: YouTube download is not supported in Hugging Face Spaces without an API key.")
-            raise gr.Error("YouTube 下載在 Hugging Face Spaces 中需要 API 金鑰。請在上方的 'YouTube API Key Settings' 中輸入您的 API 金鑰。\n\nYouTube download in Hugging Face Spaces requires an API key. Please enter your API key in the 'YouTube API Key Settings' section above.")
+        if is_spaces:
+            # 在 Spaces 環境中，顯示警告
+            print("Warning: YouTube download is not supported in Hugging Face Spaces.")
+            raise gr.Error("YouTube 下載在 Hugging Face Spaces 中不可用。由於安全限制，Spaces 環境無法通過 YouTube 的機器人驗證。請在本地環境中使用此功能。\n\nYouTube download is not available in Hugging Face Spaces. Due to security restrictions, Spaces environment cannot pass YouTube's bot verification. Please use this feature in a local environment.")
 
-        # 使用 API 模塊處理 URL
-        audio_path, video_info = youtube_api.process_youtube_url(youtube_url)
+        # 使用 yt-dlp 直接下載
+        audio_path, temp_dir, duration = download_youtube_audio(youtube_url)
 
         if audio_path and os.path.exists(audio_path):
-            # 如果有視頻信息，顯示它
-            if video_info:
-                title = video_info.get("title", "Unknown")
-                duration = video_info.get("duration", "Unknown")
-                print(f"Title: {title}, Duration: {duration}s")
-
             # 返回音訊播放器和下載按鈕的更新
             return gr.update(visible=True, value=audio_path), gr.update(visible=True, value=audio_path)
         else:
             return gr.update(visible=False, value=None), gr.update(visible=False, value=None)
     except Exception as e:
         print(f"Error processing YouTube URL: {e}")
-        error_message = str(e)
-        if "quota" in error_message.lower():
-            raise gr.Error(f"YouTube API 配額已用盡。請嘗試使用其他 API 金鑰。\n\nYouTube API quota exceeded. Please try using a different API key.\n\nError: {error_message}")
-        elif "invalid" in error_message.lower() and "api key" in error_message.lower():
-            raise gr.Error(f"無效的 API 金鑰。請檢查您的 API 金鑰是否正確。\n\nInvalid API key. Please check if your API key is correct.\n\nError: {error_message}")
-        else:
-            raise gr.Error(f"處理 YouTube URL 時發生錯誤。\n\nError processing YouTube URL.\n\nError: {error_message}")
+        return gr.update(visible=False, value=None), gr.update(visible=False, value=None)
+
+# --- YouTube Audio Download Function ---
+def download_youtube_audio(url):
+    # 使用固定的目錄來存儲下載的音訊文件，這樣它們就不會被刪除
+    download_dir = os.path.join(tempfile.gettempdir(), "youtube_downloads")
+    os.makedirs(download_dir, exist_ok=True)
+
+    # 從 URL 中提取視頻 ID 作��文件名的一部分
+    video_id = url.split("v=")[-1].split("&")[0] if "v=" in url else str(int(time.time()))
+    filename = f"youtube_{video_id}_{int(time.time())}"
+
+    temp_dir = tempfile.mkdtemp()
+    downloaded_path = None
+    try:
+        temp_filepath_tmpl = os.path.join(download_dir, f"{filename}.%(ext)s")
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'outtmpl': temp_filepath_tmpl,
+            'noplaylist': True,
+            'quiet': True,
+            'postprocessors': [{'key': 'FFmpegExtractAudio','preferredcodec': 'mp3','preferredquality': '192',}],
+            'ffmpeg_location': shutil.which("ffmpeg"),
+        }
+        if not ydl_opts['ffmpeg_location']: print("Warning: ffmpeg not found... / 警告：找不到 ffmpeg...")
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=True)
+            duration = info_dict.get('duration')
+            title = info_dict.get('title', 'unknown')
+
+            final_filepath = ydl.prepare_filename(info_dict)
+            if not final_filepath.endswith('.mp3'):
+                base_name = final_filepath.rsplit('.', 1)[0]
+                final_filepath = base_name + '.mp3'
+
+            if os.path.exists(final_filepath):
+                downloaded_path = final_filepath
+                print(f"YouTube audio downloaded: {downloaded_path}")
+                print(f"Title: {title}, Duration: {duration}s")
+            else:
+                potential_files = [os.path.join(download_dir, f) for f in os.listdir(download_dir) if f.startswith(filename) and f.endswith(".mp3")]
+                if potential_files:
+                    downloaded_path = potential_files[0]
+                    print(f"Warning: Could not find expected MP3, using fallback: {downloaded_path}")
+                    duration = None
+                else:
+                    raise FileNotFoundError(f"Audio file not found after download in {download_dir}")
+
+        return downloaded_path, temp_dir, duration
+    except Exception as e:
+        print(f"Error processing YouTube URL: {e}")
+        if temp_dir and os.path.exists(temp_dir):
+            try: shutil.rmtree(temp_dir)
+            except Exception as cleanup_e: print(f"Error cleaning temp directory {temp_dir}: {cleanup_e}")
+        return None, None, None
 
 # --- Load ASR Pipeline ---
 def load_asr_pipeline(model_id):
@@ -337,7 +362,7 @@ def load_phi4_model(model_id):
 def transcribe_audio(mic_input, file_input, youtube_url, selected_model_identifier,
                      task, language, return_timestamps,
                      phi4_prompt_text, device_choice,
-                     previous_output_text, active_tab, youtube_api_key_input=None):
+                     previous_output_text, active_tab):
     global pipe, phi4_model, phi4_processor, current_model_name, current_device
     audio_source = None
     source_type_en = ""
@@ -444,29 +469,20 @@ def transcribe_audio(mic_input, file_input, youtube_url, selected_model_identifi
         status_update_str = f"Downloading YouTube Audio / 正在下載 YouTube 音訊..."
         output_text_accumulated = status_update_prefix + status_update_str
 
-        # 使用傳入的 API 金鑰
-        user_api_key = youtube_api_key_input
+        # 檢查是否在 Hugging Face Spaces 環境中
+        is_spaces = os.environ.get("SPACE_ID") is not None
+        if is_spaces:
+            output_text_accumulated = status_update_prefix + "Error: YouTube download is not supported in Hugging Face Spaces. / 錯誤：YouTube 下載在 Hugging Face Spaces 中不可用。"
+            return (output_text_accumulated, gr.update(), gr.update(), gr.update())
 
-        try:
-            # 使用 API 模塊處理 URL，傳入用戶輸入的 API 金鑰
-            audio_path, video_info = youtube_api.process_youtube_url(youtube_url, user_api_key)
-
-            if audio_path and os.path.exists(audio_path):
-                audio_source = audio_path
-                # 從視頻信息中獲取時長
-                if video_info and "duration" in video_info:
-                    audio_duration = video_info["duration"]
-                    if video_info.get("title"):
-                        print(f"Processing YouTube video: {video_info['title']}")
-                else:
-                    # 如果沒有時長信息，稍後會嘗試從音頻文件獲取
-                    audio_duration = None
-            else:
-                output_text_accumulated = status_update_prefix + "Error: Failed to download YouTube audio. / 錯誤：無法下載 YouTube 音訊。"
-                return (output_text_accumulated, gr.update(), gr.update(), gr.update())
-        except Exception as e:
-            error_message = str(e)
-            output_text_accumulated = status_update_prefix + f"Error: {error_message}"
+        # 使用 yt-dlp 直接下載
+        audio_path, temp_dir_to_clean, duration_yt = download_youtube_audio(youtube_url)
+
+        if audio_path and os.path.exists(audio_path):
+            audio_source = audio_path
+            audio_duration = duration_yt
+        else:
+            output_text_accumulated = status_update_prefix + "Error: Failed to download YouTube audio. / 錯誤：無法下載 YouTube 音訊。"
             return (output_text_accumulated, gr.update(), gr.update(), gr.update())
     else:
         # 如果沒有選擇任何輸入源或當前標籤沒有有效輸入
@@ -810,52 +826,15 @@ with gr.Blocks(css=compact_css, theme=gr.themes.Default(spacing_size=gr.themes.s
 
                     # 如果在 Spaces 環境中，顯示警告訊息
                     if is_spaces:
-                        # 檢查是否有 API 金鑰
-                        youtube_api_key = os.environ.get("YOUTUBE_API_KEY")
-                        if youtube_api_key:
-                            gr.Markdown("""
-                            ℹ️ **注意：YouTube 下載使用 YouTube Data API**
-
-                            在 Hugging Face Spaces 中，YouTube 下載功能使用 YouTube Data API。這提供了更穩定的體驗。
-
-                            ℹ️ **Note: YouTube download uses YouTube Data API**
-
-                            In Hugging Face Spaces, YouTube download functionality uses the YouTube Data API. This provides a more stable experience.
-                            """, elem_classes="info-box")
-                        else:
-                            gr.Markdown("""
-                            ⚠️ **注意：YouTube 下載在 Hugging Face Spaces 中需要 API 金鑰**
-
-                            由於安全限制，Spaces 環境需要 YouTube Data API 金鑰才能下載 YouTube 視頻。請在環境變量中設置 YOUTUBE_API_KEY，或在本地環境中使用此功能。
-
-                            ⚠️ **Note: YouTube download in Hugging Face Spaces requires an API key**
-
-                            Due to security restrictions, Spaces environment requires a YouTube Data API key to download YouTube videos. Please set YOUTUBE_API_KEY in environment variables, or use this feature in a local environment.
-                            """, elem_classes="warning-box")
-
-                    # API 金鑰輸入框
-                    with gr.Accordion("YouTube API Key Settings / YouTube API 金鑰設置", open=False):
                         gr.Markdown("""
-                        ### YouTube API 金鑰設置 / YouTube API Key Settings
+                        ⚠️ **注意：YouTube 下載在 Hugging Face Spaces 中不可用**
 
-                        您可以在此輸入您自己的 YouTube API 金鑰。如果您沒有 API 金鑰，您可以在 [Google Cloud Console](https://console.cloud.google.com/) 中創建一個。
+                        由於安全限制，Spaces 環境無法通過 YouTube 的機器人驗證。請在本地環境中使用此功能。
 
-                        You can enter your own YouTube API key here. If you don't have an API key, you can create one in the [Google Cloud Console](https://console.cloud.google.com/).
+                        ⚠️ **Note: YouTube download is not available in Hugging Face Spaces**
 
-                        **步驟 / Steps:**
-                        1. 前往 [Google Cloud Console](https://console.cloud.google.com/)
-                        2. 創建一個新項目（或選擇現有項目）/ Create a new project (or select an existing one)
-                        3. 啟用 YouTube Data API v3 / Enable YouTube Data API v3
-                        4. 創建 API 金鑰 / Create an API key
-                        5. 將 API 金鑰複製到下方輸入框 / Copy the API key to the input box below
-
-                        **注意 / Note:** API 金鑰僅在當前會話中有效，頁面刷新後需要重新輸入。/ The API key is only valid for the current session and needs to be re-entered after page refresh.
-                        """)
-                        youtube_api_key_input = gr.Textbox(
-                            label="YouTube API Key / YouTube API 金鑰",
-                            placeholder="Enter your YouTube API key here / 在此輸入您的 YouTube API 金鑰",
-                            type="password"
-                        )
+                        Due to security restrictions, Spaces environment cannot pass YouTube's bot verification. Please use this feature in a local environment.
+                        """, elem_classes="warning-box")
 
                     # YouTube URL 輸入框
                     youtube_input = gr.Textbox(label="YouTube URL / 網址", placeholder="Paste YouTube link here / 在此貼上 YouTube 連結")
@@ -872,17 +851,15 @@ with gr.Blocks(css=compact_css, theme=gr.themes.Default(spacing_size=gr.themes.s
                         is_spaces = os.environ.get("SPACE_ID") is not None
 
                         # 檢查是否是在啟動時自動調用（緩存範例）
-                        is_startup = not hasattr(youtube_api_key_input, 'value')
+                        is_startup = True  # 簡化邏輯，假設啟動時調用
 
                         # ��果是在 Spaces 環境中啟動時調用，則不處理 URL
-                        if is_spaces and is_startup:
-                            print("Skipping example URL processing during startup in Spaces environment")
+                        if is_spaces:
+                            print("Skipping example URL processing in Spaces environment")
                             return gr.update(visible=False, value=None), gr.update(visible=False, value=None)
 
-                        # 獲取 API 金鑰
-                        api_key = youtube_api_key_input.value if hasattr(youtube_api_key_input, 'value') else None
                         # 處理 URL
-                        return process_youtube_url(url, api_key)
+                        return process_youtube_url(url)
 
                     gr.Examples(
                         examples=[["https://www.youtube.com/watch?v=5D7l0tqQJ7k"]],
@@ -948,7 +925,7 @@ with gr.Blocks(css=compact_css, theme=gr.themes.Default(spacing_size=gr.themes.s
     # 連接 YouTube 處理功能
     youtube_input.change(
         fn=process_youtube_url,
-        inputs=[youtube_input, youtube_api_key_input],
+        inputs=youtube_input,
         outputs=[youtube_audio_player, youtube_download],
         show_progress=True
     )
@@ -1037,7 +1014,7 @@ with gr.Blocks(css=compact_css, theme=gr.themes.Default(spacing_size=gr.themes.s
     # Main submit action - Corrected outputs list
     submit_button.click(
         fn=transcribe_audio_with_error_handling,
-        inputs=[mic_input, file_input, youtube_input, model_select, task_input, language_input, timestamp_input, phi4_prompt_input, device_input, output_text, active_tab, youtube_api_key_input],
+        inputs=[mic_input, file_input, youtube_input, model_select, task_input, language_input, timestamp_input, phi4_prompt_input, device_input, output_text, active_tab],
         outputs=[output_text, mic_input, file_input, youtube_input], # 保持原始輸出
         show_progress="full" # 顯示完整進度條
     )

requirements.txt

@@ -14,9 +14,6 @@ safetensors>=0.3.0
 yt-dlp>=2023.0.0
 soundfile>=0.12.0
 pydub>=0.25.0
-google-api-python-client>=2.0.0
-google-auth-httplib2>=0.1.0
-google-auth-oauthlib>=0.5.0
 
 # Data processing
 numpy>=2.0.0

requirements_local.txt

@@ -17,9 +17,10 @@ safetensors>=0.3.0
 yt-dlp>=2023.0.0
 soundfile>=0.12.0
 pydub>=0.25.0
-google-api-python-client>=2.0.0
-google-auth-httplib2>=0.1.0
-google-auth-oauthlib>=0.5.0
+# 移除 Google API 相關依賴，因為 YouTube API 不提供下載功能
+# google-api-python-client>=2.0.0
+# google-auth-httplib2>=0.1.0
+# google-auth-oauthlib>=0.5.0
 
 # Data processing
 numpy>=2.0.0

youtube_api.py

@@ -1,265 +0,0 @@
-"""
-YouTube API 處理模塊
-使用 YouTube Data API 獲取視頻信息，並使用 yt-dlp 下載音頻
-"""
-
-import os
-import time
-import tempfile
-import shutil
-import yt_dlp
-from googleapiclient.discovery import build
-from googleapiclient.errors import HttpError
-
-# YouTube API 配置
-YOUTUBE_API_SERVICE_NAME = "youtube"
-YOUTUBE_API_VERSION = "v3"
-YOUTUBE_API_KEY = None  # 將在運行時設置
-
-def set_api_key(api_key):
-    """設置 YouTube API 金鑰"""
-    global YOUTUBE_API_KEY
-    YOUTUBE_API_KEY = api_key
-    return YOUTUBE_API_KEY is not None
-
-def extract_video_id(youtube_url):
-    """從 YouTube URL 中提取視頻 ID"""
-    if "youtube.com/watch" in youtube_url:
-        # 標準 YouTube URL
-        video_id = youtube_url.split("v=")[1].split("&")[0]
-    elif "youtu.be/" in youtube_url:
-        # 短 URL
-        video_id = youtube_url.split("youtu.be/")[1].split("?")[0]
-    else:
-        # 不支持的 URL 格式
-        return None
-    return video_id
-
-def get_video_info(video_id):
-    """使用 YouTube Data API 獲取視頻信息"""
-    if not YOUTUBE_API_KEY:
-        raise ValueError("YouTube API 金鑰未設置。請先調用 set_api_key() 函數。")
-
-    try:
-        youtube = build(YOUTUBE_API_SERVICE_NAME, YOUTUBE_API_VERSION, developerKey=YOUTUBE_API_KEY)
-
-        # 獲取視頻詳細信息
-        video_response = youtube.videos().list(
-            part="snippet,contentDetails,statistics",
-            id=video_id
-        ).execute()
-
-        # 檢查是否找到視頻
-        if not video_response.get("items"):
-            return None
-
-        video_info = video_response["items"][0]
-        snippet = video_info["snippet"]
-        content_details = video_info["contentDetails"]
-
-        # 解析時長
-        duration_str = content_details["duration"]  # 格式: PT#H#M#S
-        duration_seconds = parse_duration(duration_str)
-
-        # 返回視頻信息
-        return {
-            "title": snippet["title"],
-            "description": snippet["description"],
-            "channel": snippet["channelTitle"],
-            "published_at": snippet["publishedAt"],
-            "duration": duration_seconds,
-            "thumbnail": snippet["thumbnails"]["high"]["url"] if "high" in snippet["thumbnails"] else snippet["thumbnails"]["default"]["url"]
-        }
-
-    except HttpError as e:
-        print(f"YouTube API 錯誤: {e}")
-        return None
-    except Exception as e:
-        print(f"獲取視頻信息時發生錯誤: {e}")
-        return None
-
-def parse_duration(duration_str):
-    """解析 ISO 8601 時長格式 (PT#H#M#S)"""
-    duration_str = duration_str[2:]  # 移除 "PT"
-    hours, minutes, seconds = 0, 0, 0
-
-    # 解析小時
-    if "H" in duration_str:
-        hours_part = duration_str.split("H")[0]
-        hours = int(hours_part)
-        duration_str = duration_str.split("H")[1]
-
-    # 解析分鐘
-    if "M" in duration_str:
-        minutes_part = duration_str.split("M")[0]
-        minutes = int(minutes_part)
-        duration_str = duration_str.split("M")[1]
-
-    # 解析秒
-    if "S" in duration_str:
-        seconds_part = duration_str.split("S")[0]
-        seconds = int(seconds_part)
-
-    # 計算總秒數
-    total_seconds = hours * 3600 + minutes * 60 + seconds
-    return total_seconds
-
-def download_audio(video_id, api_info=None):
-    """下載 YouTube 視頻的音頻
-
-    Args:
-        video_id: YouTube 視頻 ID
-        api_info: 從 API 獲取的視頻信息 (可選)
-
-    Returns:
-        tuple: (音頻文件路徑, 臨時目錄, 視頻時長)
-    """
-    # 使用固定的目錄來存儲下載的音訊文件
-    download_dir = os.path.join(tempfile.gettempdir(), "youtube_downloads")
-    os.makedirs(download_dir, exist_ok=True)
-
-    # 使用視頻 ID 和時間戳作為文件名
-    filename = f"youtube_{video_id}_{int(time.time())}"
-    temp_dir = tempfile.mkdtemp()
-
-    try:
-        # 準備下載路徑
-        temp_filepath_tmpl = os.path.join(download_dir, f"{filename}.%(ext)s")
-
-        # 設置 yt-dlp 選項
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'outtmpl': temp_filepath_tmpl,
-            'noplaylist': True,
-            'quiet': True,
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'ffmpeg_location': shutil.which("ffmpeg"),
-        }
-
-        # 檢查 ffmpeg
-        if not ydl_opts['ffmpeg_location']:
-            print("Warning: ffmpeg not found... / 警告：找不到 ffmpeg...")
-
-        # 如果已經有 API 信息，使用它
-        duration = api_info["duration"] if api_info else None
-        title = api_info["title"] if api_info else "Unknown"
-
-        # 下載音頻
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            # 如果沒有 API 信息，從 yt-dlp 獲取
-            if not api_info:
-                info_dict = ydl.extract_info(f"https://www.youtube.com/watch?v={video_id}", download=True)
-                duration = info_dict.get('duration')
-                title = info_dict.get('title', 'unknown')
-            else:
-                # 有 API 信息，直接下載
-                ydl.download([f"https://www.youtube.com/watch?v={video_id}"])
-
-            # 確定最終文件路徑
-            final_filepath = os.path.join(download_dir, f"{filename}.mp3")
-
-            # 檢查文件是否存在
-            if os.path.exists(final_filepath):
-                print(f"YouTube audio downloaded: {final_filepath}")
-                print(f"Title: {title}, Duration: {duration}s")
-                return final_filepath, temp_dir, duration
-            else:
-                # 嘗試查找可能的文件
-                potential_files = [
-                    os.path.join(download_dir, f)
-                    for f in os.listdir(download_dir)
-                    if f.startswith(filename) and f.endswith(".mp3")
-                ]
-                if potential_files:
-                    downloaded_path = potential_files[0]
-                    print(f"Warning: Could not find expected MP3, using fallback: {downloaded_path}")
-                    return downloaded_path, temp_dir, duration
-                else:
-                    raise FileNotFoundError(f"Audio file not found after download in {download_dir}")
-
-    except Exception as e:
-        print(f"Error downloading YouTube audio: {e}")
-        if temp_dir and os.path.exists(temp_dir):
-            try:
-                shutil.rmtree(temp_dir)
-            except Exception as cleanup_e:
-                print(f"Error cleaning temp directory {temp_dir}: {cleanup_e}")
-        return None, None, None
-
-def process_youtube_url(youtube_url, user_api_key=None):
-    """處理 YouTube URL，獲取信息並下載音頻
-
-    Args:
-        youtube_url: YouTube 視頻 URL
-        user_api_key: 用戶提供的 API 金鑰（可選）
-
-    Returns:
-        tuple: (音頻文件路徑, 視頻信息)
-    """
-    # 檢查 URL 是否有效
-    if not youtube_url or not youtube_url.strip():
-        return None, None
-
-    # 檢查是否在 Hugging Face Spaces 環境中
-    is_spaces = os.environ.get("SPACE_ID") is not None
-
-    # 如果提供了用戶 API 金鑰，設置它
-    if user_api_key:
-        set_api_key(user_api_key)
-
-    # 提取視頻 ID
-    video_id = extract_video_id(youtube_url)
-    if not video_id:
-        print(f"Invalid YouTube URL: {youtube_url}")
-        return None, None
-
-    # 檢查是否設置了 API 金鑰
-    if YOUTUBE_API_KEY:
-        # 使用 API 獲取視頻信息
-        video_info = get_video_info(video_id)
-        if not video_info:
-            print(f"Could not get video info from API for: {video_id}")
-
-            # 如果在 Spaces 環境中且沒有 API 信息，則不嘗試下載
-            if is_spaces:
-                raise ValueError("YouTube 下載在 Hugging Face Spaces 中需要有效的 API 金鑰。")
-
-            # 如果 API 失敗，嘗試直接下載
-            audio_path, temp_dir, duration = download_audio(video_id)
-            return audio_path, {"title": "Unknown", "duration": duration}
-
-        # 使用 API 信息下載音頻
-        audio_path, temp_dir, _ = download_audio(video_id, video_info)
-        return audio_path, video_info
-    else:
-        # 沒有 API 金鑰
-        if is_spaces:
-            # 在 Spaces 環境中需要 API 金鑰
-            raise ValueError("YouTube 下載在 Hugging Face Spaces 中需要 API 金鑰。請在上方的 'YouTube API Key Settings' 中輸入您的 API 金鑰。\n\nYouTube download in Hugging Face Spaces requires an API key. Please enter your API key in the 'YouTube API Key Settings' section above.")
-
-        # 本地環境，直接使用 yt-dlp
-        print("No YouTube API key set, using yt-dlp directly")
-        audio_path, temp_dir, duration = download_audio(video_id)
-        return audio_path, {"title": "Unknown", "duration": duration}
-
-# 測試代碼
-if __name__ == "__main__":
-    # 設置 API 金鑰（實際使用時應從環境變量或配置文件獲取）
-    api_key = "YOUR_API_KEY"
-    set_api_key(api_key)
-
-    # 測試 URL
-    test_url = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
-
-    # 處理 URL
-    audio_path, video_info = process_youtube_url(test_url)
-
-    if audio_path and video_info:
-        print(f"Downloaded: {audio_path}")
-        print(f"Video info: {video_info}")
-    else:
-        print("Failed to process YouTube URL")