vitaliy-sharandin commited on
Commit
86aa0c2
1 Parent(s): 157733b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -297,6 +297,9 @@ def translation_limit():
297
  else:
298
  return "<div style='color: red; text-align: center;'>Translation limit is reached</div>"
299
 
 
 
 
300
  def translate_video(video_path, youtube_link, target_language, speaker_model):
301
  try:
302
  if not video_path and not youtube_link:
@@ -305,14 +308,11 @@ def translate_video(video_path, youtube_link, target_language, speaker_model):
305
  if youtube_link:
306
  video_path = download_youtube_video(youtube_link)
307
  dubbed_video = video_translation(video_path, target_language, speaker_model, HF_TOKEN, DEEPL_TOKEN)
308
- translation_limit_info = translation_limit()
309
  except Exception as e:
310
  print(f"An error occurred: {e}")
311
  raise e
312
- return gr.Markdown(translation_limit_info), gr.components.Video(dubbed_video)
313
-
314
- def clear_inputs():
315
- return None, "", None, None
316
 
317
 
318
 
@@ -335,7 +335,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
335
  with gr.Row():
336
  with gr.Column(elem_classes=["column-frame"]):
337
  gr.Markdown("<h2 style='text-align: center;'>Inputs</h3>")
338
- limit_info = gr.Markdown(initial_usage_info)
339
  video = gr.Video(label="Upload a video file")
340
  gr.Markdown("<h3 style='text-align: center;'>OR</h3>")
341
  youtube_link = gr.Textbox(label="Paste YouTube link")
@@ -355,7 +355,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
355
  translate_btn.click(
356
  fn=translate_video,
357
  inputs=[video, youtube_link, target_language, speaker_model],
358
- outputs=[limit_info, output_video]
359
  )
360
 
361
  clear_btn.click(
 
297
  else:
298
  return "<div style='color: red; text-align: center;'>Translation limit is reached</div>"
299
 
300
+ def clear_inputs():
301
+ return None, "", None, None
302
+
303
  def translate_video(video_path, youtube_link, target_language, speaker_model):
304
  try:
305
  if not video_path and not youtube_link:
 
308
  if youtube_link:
309
  video_path = download_youtube_video(youtube_link)
310
  dubbed_video = video_translation(video_path, target_language, speaker_model, HF_TOKEN, DEEPL_TOKEN)
311
+ limit_info = translation_limit()
312
  except Exception as e:
313
  print(f"An error occurred: {e}")
314
  raise e
315
+ return gr.Markdown(limit_info), gr.components.Video(dubbed_video)
 
 
 
316
 
317
 
318
 
 
335
  with gr.Row():
336
  with gr.Column(elem_classes=["column-frame"]):
337
  gr.Markdown("<h2 style='text-align: center;'>Inputs</h3>")
338
+ translation_limit_info = gr.Markdown(initial_usage_info)
339
  video = gr.Video(label="Upload a video file")
340
  gr.Markdown("<h3 style='text-align: center;'>OR</h3>")
341
  youtube_link = gr.Textbox(label="Paste YouTube link")
 
355
  translate_btn.click(
356
  fn=translate_video,
357
  inputs=[video, youtube_link, target_language, speaker_model],
358
+ outputs=[translation_limit_info, output_video]
359
  )
360
 
361
  clear_btn.click(