updating the thumbnail gallery

app.py

@@ -5,6 +5,7 @@ import os
 import getpass
 from pixeltable.functions.video import extract_audio
 from pixeltable.functions import openai as pxop
+from pixeltable.iterators import FrameIterator
 import openai
 
 
@@ -164,7 +165,15 @@ def process_and_generate_post(video_file, social_media_type):
                 if sm_gen_flag:
                     cur_video_df.update({"sm_post": social_media_post})
 
-            return cur_video_df["sm_post"]
+
+            print("######## processing thumbnail ##########")
+
+            images = []
+            for frame in FrameIterator(video=video_file, num_frames=10):
+                images.append(frame['frame'])  # frame['frame'] is a PIL image
+            
+
+            return cur_video_df["sm_post"], images
 
         except Exception as e:
             return f"An error occurred: {e}"
@@ -203,6 +212,8 @@ def gradio_interface():
 
         output = gr.Textbox(label="Generated Social Media Post", show_copy_button=True)
 
+        thumbnail = gr.Gallery(label = "Pick your favorite Post Thumbnail", show_download_button= True, show_fullscreen_button= True)
+
         examples = gr.Examples(
             [["example1.mp4"], ["example2.mp4"],["example3.mp4"]], inputs=[video_input]
         )
@@ -211,7 +222,7 @@ def gradio_interface():
         generate_btn.click(
             fn=process_and_generate_post,
             inputs=[video_input, social_media_type],
-            outputs=[output],
+            outputs=[output, thumbnail],
         )
 
     return demo