screenshot_path = screenshot_youtube_video(video_id, entry['start'])

app.py

@@ -51,7 +51,7 @@ OUTPUT_PATH = 'videos'
 OPEN_AI_KEY = os.getenv("OPEN_AI_KEY")
 client = OpenAI(api_key=OPEN_AI_KEY)
 
-# 初始化Google Drive服务
+# # ====drive====初始化Google Drive服务
 def init_drive_service():
     credentials_json_string = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")
     credentials_dict = json.loads(credentials_json_string)
@@ -158,6 +158,9 @@ def update_file_on_drive(service, file_id, file_content):
     
     print(f"文件已更新，文件ID: {updated_file['id']}")
 
+
+# ====drive====
+
 def process_file(file):
     # 读取文件
     if file.name.endswith('.csv'):
@@ -239,7 +242,7 @@ def process_transcript_and_screenshots(video_id):
     # 处理逐字稿中的每个条目，检查并上传截图
     for entry in transcript:
         if 'img_src' not in entry:
-            screenshot_youtube_video(video_id, entry['start'])
+            screenshot_path = screenshot_youtube_video(video_id, entry['start'])
             img_file_id = upload_img_directly(service, f"{video_id}_{entry['start']}.jpg", folder_id, screenshot_path)
             set_public_permission(service, img_file_id)
             img_src = f"https://drive.google.com/uc?export=view&id={img_file_id}"
@@ -324,9 +327,12 @@ 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'
+    file_name = f"{youtube_id}_{snapshot_sec}.jpg"
     with VideoFileClip(video_path) as video:
         screenshot_path = f'{OUTPUT_PATH}/{file_name}'
         video.save_frame(screenshot_path, snapshot_sec)
+    
+    return screenshot_path
 
 def get_screenshot_from_video(video_link, start_time):
     # 实现从视频中提取帧的逻辑