Update app.py

app.py

@@ -8,12 +8,12 @@ from pixeltable.functions.video import extract_audio
 from pixeltable.functions.audio import get_metadata
 from pixeltable.functions import openai
 
-"""## Store OpenAI API Key"""
+# Store OpenAI API Key
 
 if 'OPENAI_API_KEY' not in os.environ:
     os.environ['OPENAI_API_KEY'] = getpass.getpass('Enter your OpenAI API key:')
 
-"""## Create a Table, a View, and Computed Columns"""
+# Create a Table, a View, and Computed Columns
 
 pxt.drop_dir('directory', force=True)
 pxt.create_dir('directory')
@@ -37,7 +37,7 @@ t['metadata'] = get_metadata(t.audio)
 t['transcription'] = openai.transcriptions(audio=t.audio, model='whisper-1')
 t['transcription_text'] = t.transcription.text
 
-"""## Custom UDF for Generating Social Media Prompts"""
+# Custom UDF for Generating Social Media Prompts
 
 #Custom User-Defined Function (UDF) for Generating Social Media Prompts
 @pxt.udf
@@ -100,7 +100,7 @@ def process_and_generate_post(video_file, social_media_type, progress=gr.Progres
         progress(0.8, desc="Preparing results...")
 
         # Retrieve Pixeltable Table containing all videos and stored data
-        df_output = t.select(t.transcription_text).collect().to_pandas()
+        df_output = t.select(t.transcription_text).tail(1)['transcription_text'][0]
 
         #Display content
         return social_media_post, thumbnails, df_output, audio
@@ -173,8 +173,8 @@ def gradio_interface():
                 )
                 audio = gr.Audio(label="Extracted audio", show_download_button=True)
 
-        df_output = gr.DataFrame(label="Transcription")
-
+        df_output = gr.Textbox(label="Transcription", show_copy_button=True)
+        
         generate_btn.click(
             fn=process_and_generate_post,
             trigger_mode='once',