upadte app.py +youtube api input

app.py

@@ -14,6 +14,7 @@ import traceback # For printing full errors
 import platform
 import re
 import subprocess
+import youtube_api # 已安裝相關依賴，可以使用
 
 # --- 硬體檢查函數 ---
 def get_hardware_info():
@@ -117,59 +118,7 @@ MIC_PROMPT = """**Try Reading / 試著朗讀:**
 "Success is stumbling from failure to failure with no loss of enthusiasm." - Winston Churchill
 「成功是在一次又一次失敗中，依然熱情不減地前行。」 - 溫斯頓・邱吉爾"""
 
-# --- 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
+# YouTube 音訊處理現在由 youtube_api.py 模塊處理
 
 # --- Timestamp Formatting ---
 def format_timestamp(seconds):
@@ -191,24 +140,62 @@ def update_download_file(filepath):
     return None
 
 # --- YouTube 音訊處理 ---
-def process_youtube_url(youtube_url):
-    """處理 YouTube URL，下載音訊並返回播放器和下載按鈕的更新"""
+def process_youtube_url(youtube_url, user_api_key=None):
+    """處理 YouTube URL，下載音訊並返回播放器和下載按鈕的更新
+
+    Args:
+        youtube_url: YouTube 視頻 URL
+        user_api_key: 用戶輸入的 API 金鑰（可選）
+    """
     if not youtube_url or not youtube_url.strip():
         return gr.update(visible=False, value=None), gr.update(visible=False, value=None)
 
+    # 檢查是否在 Hugging Face Spaces 環境中
+    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}")
-        # 只使用我們需要的返回值
-        audio_path, _, _ = download_youtube_audio(youtube_url)
+
+        if is_spaces and not youtube_api_key:
+            # 在 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.")
+
+        # 使用 API 模塊處理 URL
+        audio_path, video_info = youtube_api.process_youtube_url(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}")
-        return gr.update(visible=False, value=None), gr.update(visible=False, value=None)
+        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}")
 
 # --- Load ASR Pipeline ---
 def load_asr_pipeline(model_id):
@@ -347,7 +334,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):
+                     previous_output_text, active_tab, youtube_api_key_input=None):
     global pipe, phi4_model, phi4_processor, current_model_name, current_device
     audio_source = None
     source_type_en = ""
@@ -453,13 +440,30 @@ def transcribe_audio(mic_input, file_input, youtube_url, selected_model_identifi
         source_type_zh = "YouTube"
         status_update_str = f"Downloading YouTube Audio / 正在下載 YouTube 音訊..."
         output_text_accumulated = status_update_prefix + status_update_str
-        audio_path, temp_dir, duration_yt = download_youtube_audio(youtube_url)
-        if audio_path:
-            audio_source = audio_path
-            temp_dir_to_clean = temp_dir
-            audio_duration = duration_yt
-        else:
-            output_text_accumulated = status_update_prefix + "Error: Failed to download YouTube audio. / 錯誤：無法下載 YouTube 音訊。"
+
+        # 使用傳入的 API 金鑰
+        user_api_key = youtube_api_key_input
+
+        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}"
             return (output_text_accumulated, gr.update(), gr.update(), gr.update())
     else:
         # 如果沒有選擇任何輸入源或當前標籤沒有有效輸入
@@ -750,6 +754,22 @@ def update_language_ui(model_id, task):
 compact_css = """
 .tabitem { margin: 0rem !important; padding: 0rem !important;}
 .compact-file > div { min-height: unset !important; }
+.warning-box {
+    background-color: #fff3cd;
+    color: #856404;
+    padding: 10px;
+    border-radius: 5px;
+    border-left: 5px solid #ffc107;
+    margin-bottom: 15px;
+}
+.info-box {
+    background-color: #d1ecf1;
+    color: #0c5460;
+    padding: 10px;
+    border-radius: 5px;
+    border-left: 5px solid #17a2b8;
+    margin-bottom: 15px;
+}
 """
 
 # 移除 JavaScript 代碼，改用純 CSS 解決方案
@@ -782,14 +802,82 @@ with gr.Blocks(css=compact_css, theme=gr.themes.Default(spacing_size=gr.themes.s
                     file_audio_player = gr.Audio(label="Audio Preview / 音訊預覽", interactive=False, visible=False)
 
                 with gr.TabItem("▶️ YouTube") as youtube_tab:
+                    # 檢查是否在 Hugging Face Spaces 環境中
+                    is_spaces = os.environ.get("SPACE_ID") is not None
+
+                    # 如果在 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 API 金鑰。如果您沒有 API 金鑰，您可以在 [Google Cloud Console](https://console.cloud.google.com/) 中創建一個。
+
+                        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/).
+
+                        **步驟 / 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"
+                        )
+
+                    # YouTube URL 輸入框
                     youtube_input = gr.Textbox(label="YouTube URL / 網址", placeholder="Paste YouTube link here / 在此貼上 YouTube 連結")
-                    gr.Examples(examples=[["https://www.youtube.com/watch?v=5D7l0tqQJ7k"]], inputs=[youtube_input], label="Example YouTube URL / 範例 YouTube 網址")
 
                     # 添加 YouTube 音訊播放器和下載按鈕
                     with gr.Row():
                         youtube_audio_player = gr.Audio(label="YouTube Audio / YouTube 音訊", interactive=False, visible=False)
                     youtube_download = gr.File(label="Download YouTube Audio / 下載 YouTube 音訊", interactive=False, visible=False, elem_classes="compact-file")
 
+                    # 添加範例，點擊時自動處理
+                    def process_example_url(url):
+                        """處理範例 URL 的函數"""
+                        # 獲取 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)
+
+                    gr.Examples(
+                        examples=[["https://www.youtube.com/watch?v=5D7l0tqQJ7k"]],
+                        inputs=[youtube_input],
+                        label="Example YouTube URL / 範例 YouTube 網址",
+                        fn=process_example_url,  # 點擊範例時自動處理 URL
+                        outputs=[youtube_audio_player, youtube_download]
+                    )
+
                 # 添加標籤切換事件
                 mic_tab.select(fn=lambda: set_active_tab("mic"), inputs=[], outputs=[active_tab])
                 file_tab.select(fn=lambda: set_active_tab("file"), inputs=[], outputs=[active_tab])
@@ -846,7 +934,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,
+        inputs=[youtube_input, youtube_api_key_input],
         outputs=[youtube_audio_player, youtube_download],
         show_progress=True
     )
@@ -935,7 +1023,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],
+        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],
         outputs=[output_text, mic_input, file_input, youtube_input], # 保持原始輸出
         show_progress="full" # 顯示完整進度條
     )

requirements.txt

@@ -14,6 +14,9 @@ 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,6 +17,9 @@ 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

youtube_api.py

@@ -0,0 +1,247 @@
+"""
+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):
+    """處理 YouTube URL，獲取信息並下載音頻
+    
+    Args:
+        youtube_url: YouTube 視頻 URL
+    
+    Returns:
+        tuple: (音頻文件路徑, 視頻信息)
+    """
+    # 檢查 URL 是否有效
+    if not youtube_url or not youtube_url.strip():
+        return None, None
+    
+    # 提取視頻 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}")
+            # 如果 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 金鑰，直接使用 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")