screenshot_paths

app.py

@@ -75,25 +75,24 @@ def process_youtube_link(link):
     df_summarise = generate_df_summarise(transcript)
 
     formatted_transcript = []
+    screenshot_paths = []
     for entry in transcript:
         start_time = format_seconds_to_time(entry['start'])
         end_time = format_seconds_to_time(entry['start'] + entry['duration'])
         embed_url = get_embedded_youtube_link(video_id, entry['start'])
         # 截圖
         screenshot_path = screenshot_youtube_video(video_id, entry['start'])
-        image_path = os.path.join(os.path.dirname(__file__), screenshot_path),
         line = {
             "start_time": start_time,
             "end_time": end_time,
             "text": entry['text'],
             "embed_url": embed_url,
             "time_sec": entry['start'],
-            "image_path": image_path
+            "screenshot_path": screenshot_path
         }
         formatted_transcript.append(line)
+        screenshot_paths.append(screenshot_path)
 
-        
-    
     html_content = format_transcript_to_html(formatted_transcript)
     print("=====html_content=====")
     print(html_content)
@@ -104,7 +103,8 @@ def process_youtube_link(link):
             questions[1] if len(questions) > 1 else "", \
             questions[2] if len(questions) > 2 else "", \
             df_summarise, \
-            html_content
+            html_content,  \
+            screenshot_paths, 
 
 def format_transcript_to_html(formatted_transcript):
     html_content = ""
@@ -290,6 +290,9 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Tab("YouTube Transcript and Video"):
                 transcript_html = gr.HTML(label="YouTube Transcript and Video")
+            with gr.Tab("images"):
+                gallery = gr.Gallery(label="截图")
+            
             with gr.Tab("資料本文"):
                 df_string_output = gr.Textbox()
             with gr.Tab("資料摘要"):
@@ -317,7 +320,7 @@ with gr.Blocks() as demo:
     file_upload.change(process_file, inputs=file_upload, outputs=[btn_1, btn_2, btn_3, df_summarise, df_string_output])
 
     # 当输入 YouTube 链接时触发
-    youtube_link.change(process_youtube_link, inputs=youtube_link, outputs=[btn_1, btn_2, btn_3, df_summarise, transcript_html])
+    youtube_link.change(process_youtube_link, inputs=youtube_link, outputs=[btn_1, btn_2, btn_3, df_summarise, transcript_html, gallery])
 
     # 当输入网页链接时触发
     web_link.change(process_web_link, inputs=web_link, outputs=[btn_1, btn_2, btn_3, df_summarise, df_string_output])