get_youtube_title_from_gemini

app.py

@@ -37,6 +37,39 @@ def mock_summary():
     # 假資料模擬摘要
     return "這份文件主要討論人工智慧在工作效率提升方面的應用，並提供了實際案例來說明其價值。"
 
+def get_youtube_title_from_gemini(url):
+    """使用 Gemini 獲取 YouTube 標題"""
+    print(f"\n開始獲取 YouTube 標題: {url}")
+    try:
+        print("初始化 Gemini 模型設定...")
+        video = types.Part.from_uri(
+            file_uri=url,
+            mime_type="video/*",
+        )
+
+        print("開始生成標題...")
+        response = GENAI_CLIENT.models.generate_content(
+            model="gemini-2.0-flash-exp",
+            contents=[
+                types.Content(
+                    role="user",
+                    parts=[
+                        video,
+                        types.Part.from_text("請只回傳這個影片的標題，不要加入其他任何文字。")
+                    ]
+                )
+            ]
+        )
+        
+        if response and response.text:
+            title = response.text.strip()
+            print(f"成功獲取標題: {title}")
+            return title
+        return url
+    except Exception as e:
+        print(f"Gemini 獲取標題失敗: {str(e)}")
+        return url
+
 def get_youtube_title(url):
     """獲取 YouTube 影片標題"""
     try:
@@ -47,18 +80,32 @@ def get_youtube_title(url):
             else:
                 url = f'https://www.youtube.com/watch?v={url}'
         
-        ydl_opts = {
-            'quiet': True,
-            'no_warnings': True,
-            'extract_flat': True
-        }
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=False)
-            title = info.get('title', url)
-            print(f"YouTube title: {title}")
+        # 首先嘗試使用 yt-dlp
+        try:
+            ydl_opts = {
+                'quiet': True,
+                'no_warnings': True,
+                'extract_flat': True
+            }
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info = ydl.extract_info(url, download=False)
+                title = info.get('title', '')
+                if title:
+                    print(f"YouTube title from yt-dlp: {title}")
+                    return title
+        except Exception as e:
+            print(f"yt-dlp 獲取標題失敗: {str(e)}")
+        
+        # 如果 yt-dlp 失敗，嘗試使用 Gemini
+        print("嘗試使用 Gemini 獲取標題...")
+        title = get_youtube_title_from_gemini(url)
+        if title and title != url:
+            print(f"YouTube title from Gemini: {title}")
             return title
+        
+        return url
     except Exception as e:
-        print(f"Error fetching YouTube title: {str(e)}")
+        print(f"獲取標題失敗: {str(e)}")
         return url
 
 def add_to_file_list(file, file_list):