Update app.py

app.py

@@ -340,7 +340,7 @@ class App:
                                   inputs=params + whisper_params.as_list(),
                                   outputs=[tb_indicator, files_subtitles, tb_info])
                     
-                    btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,cb_diarize],outputs=tb_indicator)
+                    btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,cb_diarize,input_multi],outputs=tb_indicator)
                     # btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
 
                     input_multi.change(fn=self.update_viewer,inputs=input_multi,outputs=[input_file_audio,input_file_video,input_file_multi])
@@ -390,17 +390,29 @@ class App:
             return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
 
     @staticmethod
-    def update_dataframe(value_cb_timestamp_preview,value_cb_diarize):
-        if value_cb_timestamp_preview==True and value_cb_diarize==True:
-            return gr.Dataframe(headers=["Time","Speaker","Text"],column_widths=["10%","10%","79%"])
-        elif value_cb_timestamp_preview==True and value_cb_diarize==False:
-            return gr.Dataframe(headers=["Time","Text"],column_widths=["10%","89%"])
-        elif value_cb_timestamp_preview==False and value_cb_diarize==True:
-            return gr.Dataframe(headers=["Speaker","Text"],column_widths=["10%","89%"])
-        elif value_cb_timestamp_preview==False and value_cb_diarize==False:
-            return gr.Dataframe(headers=["Text"],column_widths=["99%"])
+    def update_dataframe(value_cb_timestamp_preview,value_cb_diarize,value_input_multi):
+        if value_input_multi!="Multiple":
+            if value_cb_timestamp_preview==True and value_cb_diarize==True:
+                return gr.Dataframe(headers=["Time","Speaker","Text"],column_widths=["10%","10%","79%"])
+            elif value_cb_timestamp_preview==True and value_cb_diarize==False:
+                return gr.Dataframe(headers=["Time","Text"],column_widths=["10%","89%"])
+            elif value_cb_timestamp_preview==False and value_cb_diarize==True:
+                return gr.Dataframe(headers=["Speaker","Text"],column_widths=["10%","89%"])
+            elif value_cb_timestamp_preview==False and value_cb_diarize==False:
+                return gr.Dataframe(headers=["Text"],column_widths=["99%"])
+            else:
+                return gr.Dataframe(headers=["Text"],column_widths=["99%"])
         else:
-            return gr.Dataframe(headers=["Text"],column_widths=["99%"])
+            if value_cb_timestamp_preview==True and value_cb_diarize==True:
+                return gr.Dataframe(headers=["File","Time","Speaker","Text"],column_widths=["10%","10%","10%","69%"])
+            elif value_cb_timestamp_preview==True and value_cb_diarize==False:
+                return gr.Dataframe(headers=["File","Time","Text"],column_widths=["10%","10%","79%"])
+            elif value_cb_timestamp_preview==False and value_cb_diarize==True:
+                return gr.Dataframe(headers=["File","Speaker","Text"],column_widths=["10%","10%","79%"])
+            elif value_cb_timestamp_preview==False and value_cb_diarize==False:
+                return gr.Dataframe(headers=["File","Text"],column_widths=["10%","89%"])
+            else:
+                return gr.Dataframe(headers=["File","Text"],column_widths=["10%","89%"])
 
 # Create the parser for command-line arguments
 parser = argparse.ArgumentParser()