Maximofn commited on
Commit
9cfb673
1 Parent(s): 7eee649

Add transcription concatenated progress

Browse files
Files changed (1) hide show
  1. app.py +13 -7
app.py CHANGED
@@ -316,6 +316,7 @@ def change_visibility_texboxes(url):
316
  video_donwloaded = gr.Textbox(placeholder="Waiting", label="Video downloaded", elem_id="video_downloaded", interactive=False, visible=True)
317
  video_sliced = gr.Textbox(placeholder="Waiting", label="Video sliced", elem_id="video_sliced", interactive=False, visible=True)
318
  video_transcribed = gr.Textbox(placeholder="Waiting", label="Video transcribed", elem_id="video_transcribed", interactive=False, visible=True)
 
319
  video_translated = gr.Textbox(placeholder="Waiting", label="Transcription translated", elem_id="transcription_translated", interactive=False, visible=True)
320
  video_subtitled = gr.Textbox(placeholder="Waiting", label="Video subtitled", elem_id="video_subtitled", interactive=False, visible=True)
321
 
@@ -324,6 +325,7 @@ def change_visibility_texboxes(url):
324
  video_donwloaded,
325
  video_sliced,
326
  video_transcribed,
 
327
  video_translated,
328
  video_subtitled,
329
  )
@@ -421,7 +423,7 @@ def concatenate_transcriptions():
421
  result = f.read()
422
 
423
  return (
424
- gr.Textbox(value="Ok", label="Transcription translated", elem_id="transcription_translated", interactive=False, visible=True),
425
  gr.Textbox(value=audio_transcribed, label="Original audio transcribed", elem_id="original_audio_transcribed", visible=False),
426
  )
427
 
@@ -441,7 +443,7 @@ def translate_transcription(original_audio_transcribed_path, source_languaje, ta
441
  os.system(command)
442
 
443
  return (
444
- gr.Textbox(value="Ok", label="Video subtitled", elem_id="video_subtitled", interactive=False, visible=True),
445
  gr.Textbox(value=translated_transcription, label="Original audio translated", elem_id="original_audio_translated", visible=False)
446
  )
447
 
@@ -463,7 +465,10 @@ def add_translated_subtitles_to_video(original_video_path, original_audio_path,
463
 
464
  subtitled_video = "videos/download_video_with_subtitles.mp4"
465
 
466
- return gr.Video(value=subtitled_video, label="Subtitled video", elem_id="subtitled_video", visible=True, interactive=False)
 
 
 
467
 
468
  def subtify():
469
  with gr.Blocks() as demo:
@@ -491,6 +496,7 @@ def subtify():
491
  video_donwloaded = gr.Textbox(placeholder="Waiting", label="Video downloaded", elem_id="video_downloaded", interactive=False, visible=visible)
492
  video_sliced = gr.Textbox(placeholder="Waiting", label="Video sliced", elem_id="video_sliced", interactive=False, visible=visible)
493
  video_transcribed = gr.Textbox(placeholder="Waiting", label="Video transcribed", elem_id="video_transcribed", interactive=False, visible=visible)
 
494
  video_translated = gr.Textbox(placeholder="Waiting", label="Transcription translated", elem_id="transcription_translated", interactive=False, visible=visible)
495
  video_subtitled = gr.Textbox(placeholder="Waiting", label="Video subtitled", elem_id="video_subtitled", interactive=False, visible=visible)
496
 
@@ -526,13 +532,13 @@ def subtify():
526
  subtify_button,
527
  ]
528
  )
529
- subtify_button.click(fn=change_visibility_texboxes, inputs=[], outputs=[start_block, video_donwloaded, video_sliced, video_transcribed, video_translated, video_subtitled])
530
  start_block.change(fn=get_audio_and_video_from_video, inputs=[url_textbox], outputs=[video_donwloaded, original_audio_path, original_video_path])
531
  video_donwloaded.change(fn=slice_audio, inputs=[original_audio_path], outputs=[video_sliced])
532
  video_sliced.change(fn=trascribe_audio, inputs=[source_languaje], outputs=[video_transcribed])
533
- video_transcribed.change(fn=concatenate_transcriptions, inputs=[], outputs=[video_translated, original_audio_transcribed_path])
534
- video_translated.change(fn=translate_transcription, inputs=[original_audio_transcribed_path, source_languaje, target_languaje], outputs=[video_subtitled, original_audio_translated_path])
535
- video_subtitled.change(fn=add_translated_subtitles_to_video, inputs=[original_video_path, original_audio_path, original_audio_translated_path], outputs=subtitled_video)
536
 
537
  demo.launch()
538
 
 
316
  video_donwloaded = gr.Textbox(placeholder="Waiting", label="Video downloaded", elem_id="video_downloaded", interactive=False, visible=True)
317
  video_sliced = gr.Textbox(placeholder="Waiting", label="Video sliced", elem_id="video_sliced", interactive=False, visible=True)
318
  video_transcribed = gr.Textbox(placeholder="Waiting", label="Video transcribed", elem_id="video_transcribed", interactive=False, visible=True)
319
+ transcriptions_concatenated = gr.Textbox(placeholder="Waiting", label="Transcriptions concatenated", elem_id="transcriptions_concatenated", interactive=False, visible=True)
320
  video_translated = gr.Textbox(placeholder="Waiting", label="Transcription translated", elem_id="transcription_translated", interactive=False, visible=True)
321
  video_subtitled = gr.Textbox(placeholder="Waiting", label="Video subtitled", elem_id="video_subtitled", interactive=False, visible=True)
322
 
 
325
  video_donwloaded,
326
  video_sliced,
327
  video_transcribed,
328
+ transcriptions_concatenated,
329
  video_translated,
330
  video_subtitled,
331
  )
 
423
  result = f.read()
424
 
425
  return (
426
+ gr.Textbox(value="Ok", label="Transcriptions concatenated", elem_id="transcriptions_concatenated", interactive=False, visible=True),
427
  gr.Textbox(value=audio_transcribed, label="Original audio transcribed", elem_id="original_audio_transcribed", visible=False),
428
  )
429
 
 
443
  os.system(command)
444
 
445
  return (
446
+ gr.Textbox(value="Ok", label="Transcription translated", elem_id="transcription_translated", interactive=False, visible=True),
447
  gr.Textbox(value=translated_transcription, label="Original audio translated", elem_id="original_audio_translated", visible=False)
448
  )
449
 
 
465
 
466
  subtitled_video = "videos/download_video_with_subtitles.mp4"
467
 
468
+ return (
469
+ gr.Textbox(value="Ok", label="Video subtitled", elem_id="video_subtitled", interactive=False, visible=True),
470
+ gr.Video(value=subtitled_video, label="Subtitled video", elem_id="subtitled_video", visible=True, interactive=False),
471
+ )
472
 
473
  def subtify():
474
  with gr.Blocks() as demo:
 
496
  video_donwloaded = gr.Textbox(placeholder="Waiting", label="Video downloaded", elem_id="video_downloaded", interactive=False, visible=visible)
497
  video_sliced = gr.Textbox(placeholder="Waiting", label="Video sliced", elem_id="video_sliced", interactive=False, visible=visible)
498
  video_transcribed = gr.Textbox(placeholder="Waiting", label="Video transcribed", elem_id="video_transcribed", interactive=False, visible=visible)
499
+ transcriptions_concatenated = gr.Textbox(placeholder="Waiting", label="Transcriptions concatenated", elem_id="transcriptions_concatenated", interactive=False, visible=visible)
500
  video_translated = gr.Textbox(placeholder="Waiting", label="Transcription translated", elem_id="transcription_translated", interactive=False, visible=visible)
501
  video_subtitled = gr.Textbox(placeholder="Waiting", label="Video subtitled", elem_id="video_subtitled", interactive=False, visible=visible)
502
 
 
532
  subtify_button,
533
  ]
534
  )
535
+ subtify_button.click(fn=change_visibility_texboxes, inputs=[], outputs=[start_block, video_donwloaded, video_sliced, video_transcribed, transcriptions_concatenated, video_translated, video_subtitled])
536
  start_block.change(fn=get_audio_and_video_from_video, inputs=[url_textbox], outputs=[video_donwloaded, original_audio_path, original_video_path])
537
  video_donwloaded.change(fn=slice_audio, inputs=[original_audio_path], outputs=[video_sliced])
538
  video_sliced.change(fn=trascribe_audio, inputs=[source_languaje], outputs=[video_transcribed])
539
+ video_transcribed.change(fn=concatenate_transcriptions, inputs=[], outputs=[transcriptions_concatenated, original_audio_transcribed_path])
540
+ transcriptions_concatenated.change(fn=translate_transcription, inputs=[original_audio_transcribed_path, source_languaje, target_languaje], outputs=[video_translated, original_audio_translated_path])
541
+ video_translated.change(fn=add_translated_subtitles_to_video, inputs=[original_video_path, original_audio_path, original_audio_translated_path], outputs=[video_subtitled, subtitled_video])
542
 
543
  demo.launch()
544