image_src

app.py

@@ -127,7 +127,14 @@ def upload_img_directly(service, file_name, folder_id, file_path):
     file_metadata = {'name': file_name, 'parents': [folder_id]}
     media = MediaFileUpload(file_path, mimetype='image/jpeg')
     file = service.files().create(body=file_metadata, media_body=media, fields='id').execute()
-    print(f"File ID: {file.get('id')}")
+    return file.get('id')  # 返回文件ID
+
+def set_public_permission(service, file_id):
+    service.permissions().create(
+        fileId=file_id,
+        body={"type": "anyone", "role": "reader"},
+        fields='id',
+    ).execute()
 
 def process_file(file):
     # 读取文件
@@ -275,22 +282,29 @@ def download_youtube_video(youtube_id, output_path=OUTPUT_PATH):
 
 
 def screenshot_youtube_video(youtube_id, snapshot_sec):    
-    # 这里假设视频已经在适当的位置
-    video_path = f'{OUTPUT_PATH}/{youtube_id}.mp4'
-
-    # Load the video and take a screenshot
-    with VideoFileClip(video_path) as video:
-        screenshot_path = f'{OUTPUT_PATH}/{youtube_id}_{snapshot_sec}.jpg'
-        video.save_frame(screenshot_path, snapshot_sec)
-    
-    # 上传截图到Google Drive
     service = init_drive_service()
-    parent_folder_id = '1GgI4YVs0KckwStVQkLa1NZ8IpaEMurkL'  # youtube逐字稿圖檔的ID
+    parent_folder_id = '1GgI4YVs0KckwStVQkLa1NZ8IpaEMurkL'  # 替换为实际的Google Drive文件夹ID
     folder_id = create_folder_if_not_exists(service, youtube_id, parent_folder_id)
     file_name = f'{youtube_id}_{snapshot_sec}.jpg'
-    upload_img_directly(service, file_name, folder_id, screenshot_path)
+
+    # 检查Google Drive是否已有截图
+    exists, file_id = check_file_exists(service, folder_id, file_name)
+    if not exists:
+        # 如果没有找到截图，处理视频并上传
+        video_path = f'{OUTPUT_PATH}/{youtube_id}.mp4'
+        with VideoFileClip(video_path) as video:
+            screenshot_path = f'{OUTPUT_PATH}/{file_name}'
+            video.save_frame(screenshot_path, snapshot_sec)
+        # 上传并获取文件ID
+        file_id = upload_img_directly(service, file_name, folder_id, screenshot_path)
+        print(f"截图已上传到Google Drive, 文件ID: {file_id}")
+        set_public_permission(service, file_id)
+
+        # 上传后删除本地临时截图文件
+        os.remove(screenshot_path)
     
-    return screenshot_path
+    image_src = f"https://drive.google.com/uc?export=view&id={file_id}"
+    return image_src
 
 def get_screenshot_from_video(video_link, start_time):
     # 实现从视频中提取帧的逻辑