LAP-DEV commited on
Commit
8995181
·
verified ·
1 Parent(s): d9a2963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -22
app.py CHANGED
@@ -339,8 +339,8 @@ class App:
339
  btn_run.click(fn=self.whisper_inf.transcribe_file,
340
  inputs=params + whisper_params.as_list(),
341
  outputs=[tb_indicator, files_subtitles, tb_info])
342
-
343
  btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,cb_diarize,input_multi],outputs=tb_indicator)
 
344
  # btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
345
 
346
  input_multi.change(fn=self.update_viewer,inputs=input_multi,outputs=[input_file_audio,input_file_video,input_file_multi])
@@ -391,28 +391,16 @@ class App:
391
 
392
  @staticmethod
393
  def update_dataframe(value_cb_timestamp_preview,value_cb_diarize,value_input_multi):
394
- if value_input_multi!="Multiple":
395
- if value_cb_timestamp_preview==True and value_cb_diarize==True:
396
- return gr.Dataframe(headers=["Time","Speaker","Text"],column_widths=["10%","10%","79%"])
397
- elif value_cb_timestamp_preview==True and value_cb_diarize==False:
398
- return gr.Dataframe(headers=["Time","Text"],column_widths=["10%","89%"])
399
- elif value_cb_timestamp_preview==False and value_cb_diarize==True:
400
- return gr.Dataframe(headers=["Speaker","Text"],column_widths=["10%","89%"])
401
- elif value_cb_timestamp_preview==False and value_cb_diarize==False:
402
- return gr.Dataframe(headers=["Text"],column_widths=["99%"])
403
- else:
404
- return gr.Dataframe(headers=["Text"],column_widths=["99%"])
405
  else:
406
- if value_cb_timestamp_preview==True and value_cb_diarize==True:
407
- return gr.Dataframe(headers=["File","Time","Speaker","Text"],column_widths=["10%","10%","10%","69%"])
408
- elif value_cb_timestamp_preview==True and value_cb_diarize==False:
409
- return gr.Dataframe(headers=["File","Time","Text"],column_widths=["10%","10%","79%"])
410
- elif value_cb_timestamp_preview==False and value_cb_diarize==True:
411
- return gr.Dataframe(headers=["File","Speaker","Text"],column_widths=["10%","10%","79%"])
412
- elif value_cb_timestamp_preview==False and value_cb_diarize==False:
413
- return gr.Dataframe(headers=["File","Text"],column_widths=["10%","89%"])
414
- else:
415
- return gr.Dataframe(headers=["File","Text"],column_widths=["10%","89%"])
416
 
417
  # Create the parser for command-line arguments
418
  parser = argparse.ArgumentParser()
 
339
  btn_run.click(fn=self.whisper_inf.transcribe_file,
340
  inputs=params + whisper_params.as_list(),
341
  outputs=[tb_indicator, files_subtitles, tb_info])
 
342
  btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,cb_diarize,input_multi],outputs=tb_indicator)
343
+
344
  # btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
345
 
346
  input_multi.change(fn=self.update_viewer,inputs=input_multi,outputs=[input_file_audio,input_file_video,input_file_multi])
 
391
 
392
  @staticmethod
393
  def update_dataframe(value_cb_timestamp_preview,value_cb_diarize,value_input_multi):
394
+ if value_cb_timestamp_preview==True and value_cb_diarize==True:
395
+ return gr.Dataframe(headers=["File","Time","Speaker","Text"],column_widths=["10%","10%","10%","69%"])
396
+ elif value_cb_timestamp_preview==True and value_cb_diarize==False:
397
+ return gr.Dataframe(headers=["File","Time","Text"],column_widths=["10%","10%","79%"])
398
+ elif value_cb_timestamp_preview==False and value_cb_diarize==True:
399
+ return gr.Dataframe(headers=["File","Speaker","Text"],column_widths=["10%","10%","79%"])
400
+ elif value_cb_timestamp_preview==False and value_cb_diarize==False:
401
+ return gr.Dataframe(headers=["File","Text"],column_widths=["10%","89%"])
 
 
 
402
  else:
403
+ return gr.Dataframe(headers=["File","Text"],column_widths=["10%","89%"])
 
 
 
 
 
 
 
 
 
404
 
405
  # Create the parser for command-line arguments
406
  parser = argparse.ArgumentParser()