Update app.py

app.py

@@ -340,7 +340,7 @@ class App:
                     btn_run.click(fn=self.whisper_inf.transcribe_file,
                                   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,input_multi,tb_indicator,files_subtitles],outputs=tb_indicator)
+                    btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,cb_diarize],outputs=tb_indicator)
                     
                     # btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
 
@@ -391,7 +391,21 @@ 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,value_input_multi,dataframe_data,list_files):
+    def update_dataframe(value_cb_timestamp_preview,value_cb_diarize,value_input_multi):
+        
+        if value_cb_timestamp_preview==True and value_cb_diarize==True:
+            return gr.Dataframe(headers=["Time","Speaker","Text"],column_widths=["10%","15%","70%"],datatype=["str","str","markdown"])
+        elif value_cb_timestamp_preview==True and value_cb_diarize==False:
+            return gr.Dataframe(headers=["Time","Text"],column_widths=["10%","85%"],datatype=["str","markdown")
+        elif value_cb_timestamp_preview==False and value_cb_diarize==True:
+            return gr.Dataframe(headers=["Speaker","Text"],column_widths=["15%","80%"],datatype=["str","markdown"])
+        elif value_cb_timestamp_preview==False and value_cb_diarize==False:
+            return gr.Dataframe(headers=["Text"],column_widths=["95%"],datatype=["markdown"])
+        else:
+            return gr.Dataframe(headers=["Text"],column_widths=["95%"],datatype=["markdown"])
+
+    @staticmethod
+    def update_dataframe_column(dataframe_data,list_files):
 
         temp = dataframe_data
         print("test df data: ")
@@ -407,17 +421,6 @@ class App:
         new_data_rows = [[1,2],[3,4]]
         updated_data = {"headers": new_headers, "data": new_data_rows}
         return gr.update(value=updated_data)
-        
-        if value_cb_timestamp_preview==True and value_cb_diarize==True:
-            return gr.Dataframe(headers=["Time","Speaker","Text","Download"],column_widths=["10%","15%","50%","20%"],datatype=["str","str","markdown","html"])
-        elif value_cb_timestamp_preview==True and value_cb_diarize==False:
-            return gr.Dataframe(headers=["Time","Text","Download"],column_widths=["10%","65%","20%"],datatype=["str","markdown","html"])
-        elif value_cb_timestamp_preview==False and value_cb_diarize==True:
-            return gr.Dataframe(headers=["Speaker","Text","Download"],column_widths=["15%","60%","20%"],datatype=["str","markdown","html"])
-        elif value_cb_timestamp_preview==False and value_cb_diarize==False:
-            return gr.Dataframe(headers=["Text","Download"],column_widths=["75%","20%"],datatype=["markdown","html"])
-        else:
-            return gr.Dataframe(headers=["Text","Download"],column_widths=["75%","20%"],datatype=["markdown","html"])
 
 # Create the parser for command-line arguments
 parser = argparse.ArgumentParser()