Update app.py

app.py

@@ -327,7 +327,8 @@ class App:
                     with gr.Row():
                         with gr.Column(scale=4):
                             #tb_indicator = gr.Textbox(label="Output preview (Always review output generated by AI models)", show_copy_button=True, show_label=True)
-                            tb_indicator = gr.Dataframe(label="Output preview (Always review output generated by AI models)", headers=["Time","Speaker","Text"], show_search="search", show_label=True, show_copy_button=True, show_fullscreen_button=True, interactive=False)
+                            tb_indicator = gr.Dataframe(label="Output preview (Always review output generated by AI models)", show_search="search", show_label=True, show_copy_button=True, show_fullscreen_button=True, interactive=False)
+                            tb_indicator = self.update_dataframe(cb_timestamp_preview,cb_diarize) #Show needed columns
                         with gr.Column(scale=1):
                             tb_info = gr.Textbox(label="Output info", interactive=False, show_copy_button=True)
                             files_subtitles = gr.Files(label="Output data", interactive=False, file_count="multiple")
@@ -395,9 +396,9 @@ class App:
         elif value_cb_timestamp_preview==False and value_cb_diarize==True:
             return gr.Dataframe(headers=["Speaker","Text"], column_widths=["15%","85%"])
         elif value_cb_timestamp_preview==False and value_cb_diarize==False:
-            return gr.Dataframe(headers=[Text"], column_widths=["100%"])
+            return gr.Dataframe(headers=["Text"], column_widths=["100%"])
         else:
-            return gr.Dataframe(headers=[Text"], column_widths=["100%"])
+            return gr.Dataframe(headers=["Text"], column_widths=["100%"])
 
 # Create the parser for command-line arguments
 parser = argparse.ArgumentParser()