Update app.py

app.py

@@ -267,10 +267,47 @@ def translate_video(video_path, youtube_link, target_language, speaker_model):
  if youtube_link:
  video_path = download_youtube_video(youtube_link)
  dubbed_video = video_translation(video_path, target_language, speaker_model, HF_TOKEN, DEEPL_TOKEN)
+ translation_limit_info = translation_limit()
  except Exception as e:
  print(f"An error occurred: {e}")
  raise e
- return gr.components.Video(dubbed_video)
+ return translation_limit_info, gr.components.Video(dubbed_video)
+
+ 
+def translation_limit():
+ translator = deepl.Translator(DEEPL_TOKEN)
+ usage = translator.get_usage()
+ if usage.character.valid:
+ characters_used = usage.character.count
+ minutes_used = int(characters_used / 150) # Assuming 150 characters per minute, converted to int
+ max_minutes = int(usage.character.limit / 150) # Converted to int
+ percent_used = (minutes_used / max_minutes) * 100
+
+ # Formatting the output for hours and minutes
+ used_time_str = f"{minutes_used} min used"
+ max_time_str = f"{max_minutes} min total"
+ if minutes_used >= 60:
+ hours_used = minutes_used // 60
+ minutes_used = minutes_used % 60
+ used_time_str = f"{hours_used} hrs, {minutes_used} min used"
+ if max_minutes >= 60:
+ hours_max = max_minutes // 60
+ remaining_minutes_max = max_minutes % 60
+ max_time_str = f"{hours_max} hrs, {remaining_minutes_max} min total"
+
+ progress_bar_html = (
+ "<div style='width: 100%; background-color: #ddd; position: relative; text-align: center; "
+ "line-height: 2em; font-size: 0.9em;'>"
+ "<div style='position: absolute; width: 100%; left: 0; top: 0;'>"
+ f"{used_time_str} / {max_time_str}"
+ "</div>"
+ f"<div style='height: 2em; background-color: #4caf50; width: {percent_used}%;'>"
+ "</div>"
+ "</div>"
+ )
+ return progress_bar_html
+ else:
+ return "<div style='color: red; text-align: center;'>Translation limit is reached</div>"
 
 def clear_inputs():
  return None, "", None, None
@@ -287,12 +324,16 @@ css = """
 """
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+ 
+ initial_usage_info = translation_limit()
+
  gr.Markdown("<h1 style='text-align: center;'>🌐AI Video Translation</h2>")
  gr.Markdown("<h3 style='text-align: center;'>Currently supported languages are: English, Polish, Ukrainian, and Russian</h3>")
  
  with gr.Row():
  with gr.Column(elem_classes=["column-frame"]):
  gr.Markdown("<h2 style='text-align: center;'>Inputs</h3>")
+ translation_limit_info = gr.Markdown(initial_usage_info)
  video = gr.Video(label="Upload a video file")
  gr.Markdown("<h3 style='text-align: center;'>OR</h3>")
  youtube_link = gr.Textbox(label="Paste YouTube link")
@@ -312,7 +353,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
  translate_btn.click(
  fn=translate_video,
  inputs=[video, youtube_link, target_language, speaker_model],
- outputs=output_video
+ outputs=[translation_limit_info, output_video]
  )
 
  clear_btn.click(