vitaliy-sharandin
commited on
Commit
•
86aa0c2
1
Parent(s):
157733b
Update app.py
Browse files
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 |
-
|
309 |
except Exception as e:
|
310 |
print(f"An error occurred: {e}")
|
311 |
raise e
|
312 |
-
return gr.Markdown(
|
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 |
-
|
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=[
|
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(
|