Update app.py

app.py

@@ -21,14 +21,13 @@ def process_and_generate_post(video_file, social_media_type, progress=gr.Progres
     progress(0, desc="Initializing...")
     
     # Create a Table, a View, and Computed Columns
-
     pxt.drop_dir('directory', force=True)
     pxt.create_dir('directory')
     
     t = pxt.create_table(
         'directory.video_table', {
-        "video": pxt.VideoType(nullable=True),
-        "sm_type": pxt.StringType(nullable=True),
+        "video": pxt.Video,
+        "sm_type": pxt.String
         }
     )
     
@@ -39,10 +38,10 @@ def process_and_generate_post(video_file, social_media_type, progress=gr.Progres
     )
     
     # Create computed columns to store transformations and persist outputs
-    t['audio'] = extract_audio(t.video, format='mp3')
-    t['metadata'] = get_metadata(t.audio)
-    t['transcription'] = openai.transcriptions(audio=t.audio, model='whisper-1')
-    t['transcription_text'] = t.transcription.text
+    t.add_computed_column(audio=extract_audio(t.video, format='mp3'))
+    t.add_computed_column(metadata=get_metadata(t.audio))
+    t.add_computed_column(transcription=openai.transcriptions(audio=t.audio, model='whisper-1'))
+    t.add_computed_column(transcription_text=t.transcription.text)
 
     progress(0.1, desc="Creating UDFs...")
     
@@ -64,10 +63,10 @@ def process_and_generate_post(video_file, social_media_type, progress=gr.Progres
     progress(0.2, desc="Calling LLMs")
     
     # # Generate responses using OpenAI's chat completion API
-    t['response'] = openai.chat_completions(messages=t.message, model='gpt-4o-mini-2024-07-18', max_tokens=500)
+    t.add_computed_column(response=openai.chat_completions(messages=t.message, model='gpt-4o-mini-2024-07-18', max_tokens=500))
     
     ## Extract the content of the response
-    t['answer'] = t.response.choices[0].message.content
+    t.add_computed_column(answer=t.response.choices[0].message.content)
     
     if not video_file:
         return "Please upload a video file.", None