Update app.py

app.py

@@ -11,9 +11,11 @@ from pixeltable.functions import openai
 if 'OPENAI_API_KEY' not in os.environ:
     os.environ['OPENAI_API_KEY'] = getpass.getpass('Enter your OpenAI API key:')
 
+# Create a Pixeltable directory to organize related tables
 pxt.drop_dir('directory', force=True)
 pxt.create_dir('directory')
 
+# Create a table to store video data
 t = pxt.create_table(
     'directory.video_table', {
     "video": pxt.VideoType(nullable=True),
@@ -21,6 +23,7 @@ t = pxt.create_table(
     }
 )
 
+# Create a view that automatically extracts frames from videos
 frames_view = pxt.create_view(
     "directory.frames",
     t,
@@ -33,6 +36,8 @@ t['metadata'] = get_metadata(t.audio)
 t['transcription'] = openai.transcriptions(audio=t.audio, model='whisper-1')
 t['transcription_text'] = t.transcription.text
 
+# Create a user-defined function (UDF) to construct the prompt
+# This shows how Pixeltable allows users to extend functionality with custom Python code
 @pxt.udf
 def prompt(A: str, B: str) -> list[dict]:
     return [
@@ -42,6 +47,7 @@ def prompt(A: str, B: str) -> list[dict]:
 
 t['message'] = prompt(t.sm_type, t.transcription_text)
 
+# Import a function from Pixeltable's built-in library for OpenAI
 t['response'] = openai.chat_completions(messages=t.message, model='gpt-4o-mini-2024-07-18', max_tokens=500)
 t['answer'] = t.response.choices[0].message.content
 
@@ -57,7 +63,7 @@ def process_and_generate_post(video_file, social_media_type):
         if video_size > MAX_VIDEO_SIZE_MB:
             return f"The video file is larger than {MAX_VIDEO_SIZE_MB} MB. Please upload a smaller file.", None
 
-        # Insert video in PixelTable
+        # # Insert a video into the table. Pixeltable supports referencing external data sources like URLs
         t.insert([{
             "video": video_file,
             "sm_type": social_media_type
@@ -67,8 +73,7 @@ def process_and_generate_post(video_file, social_media_type):
         social_media_post = t.select(t.answer).tail(1)['answer'][0]
 
         # Retrieve thumbnails
-        frames = frames_view.select(frames_view.frame).tail(4)
-        thumbnails = [frame['frame'] for frame in frames]
+        thumbnails = frames_view.select(frames_view.frame).tail(4)['frame']
 
         #Display content
         return social_media_post, thumbnails