upload_img_directly(service, file_name, folder_id, screenshot_path)

app.py

@@ -123,6 +123,12 @@ def download_file_as_string(service, file_id):
     content = fh.read().decode('utf-8')
     return content
 
+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')}")
+
 def process_file(file):
     # 读取文件
     if file.name.endswith('.csv'):
@@ -195,10 +201,8 @@ def process_youtube_link(link):
     transcript = None
     exists, file_id = check_file_exists(service, folder_id, file_name)
     if not exists:
-        # 获取逐字稿
         transcript = YouTubeTranscriptApi.get_transcript(video_id, languages=['zh-TW'])
         transcript_text = json.dumps(transcript, ensure_ascii=False, indent=2)
-        # 上传到Google Drive
         upload_content_directly(service, file_name, folder_id, transcript_text)
         print("逐字稿已上传到Google Drive")
     else:
@@ -280,6 +284,13 @@ def screenshot_youtube_video(youtube_id, snapshot_sec):
         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
+    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)
+    
     return screenshot_path
 
 def get_screenshot_from_video(video_link, start_time):