parse_time

app.py

@@ -22,6 +22,7 @@ import os
 import io
 import time
 import json
+from datetime import timedelta
 from urllib.parse import urlparse, parse_qs
 
 from google.cloud import storage
@@ -347,6 +348,18 @@ def docx_to_text(file):
 
 
 # ---- YouTube link ----
+def parse_time(time_str):
+    """將時間字符串 'HH:MM:SS' 或 'MM:SS' 轉換為 timedelta 物件。"""
+    parts = list(map(int, time_str.split(':')))
+    if len(parts) == 3:
+        hours, minutes, seconds = parts
+    elif len(parts) == 2:
+        hours = 0  # 沒有小時部分時，將小時設為0
+        minutes, seconds = parts
+    else:
+        raise ValueError("時間格式不正確，應為 'HH:MM:SS' 或 'MM:SS'")
+    return timedelta(hours=hours, minutes=minutes, seconds=seconds)
+
 def format_seconds_to_time(seconds):
     """将秒数格式化为 时:分:秒 的形式"""
     hours = int(seconds // 3600)
@@ -1188,10 +1201,13 @@ def generate_key_moments(formatted_simple_transcript, formatted_transcript):
     print(key_moments)
     print("=====key_moments=====")
     image_links = {entry['start_time']: entry['screenshot_path'] for entry in formatted_transcript}
+
     for moment in key_moments:
-        start_time = moment['start']
-        end_time = moment['end']
-        moment_images = [image_links[time] for time in image_links if start_time <= time <= end_time]
+        start_time = parse_time(moment['start'])
+        end_time = parse_time(moment['end'])
+        # 使用轉換後的 timedelta 物件進行時間比較
+        moment_images = [image_links[time] for time in image_links 
+                        if start_time <= parse_time(time) <= end_time]
         moment['images'] = moment_images
 
     return key_moments