Switch to alternative google translate library

app.py

@@ -5,11 +5,13 @@ from pytube import YouTube
 from datasets import Dataset, Audio
 from moviepy.editor import AudioFileClip
 
-import googletrans
-from googletrans import Translator
+# import googletrans                  # googletrans api
+# from googletrans import Translator  # googletrans api
+from google_trans_new import google_translator  
 
 pipe = pipeline(model="Neprox/model")
-translator = Translator()
+translator = google_translator()
+# translator = Translator() # googletrans api
 
 # Get languages available for translation
 languages = []
@@ -75,16 +77,17 @@ def divide_into_30s_segments(audio_fpath, seconds_max):
 
     return segment_paths, segment_start_times
 
-def get_translation(text, dest="English (en)"):
+def get_translation(text, target_lang="English (en)"):
     """
     Translates the given Swedish text to the language specified.
     """
-    lang_code = dest.split(" ")[-1][1:-1]
-    result = translator.translate(text, lang_code, 'sv')
-    return result.text
+    lang_code = target_lang.split(" ")[-1][1:-1]
+    return translator.translate(text, lang_tgt=lang_code)
+    # result = translator.translate(text, lang_code, 'sv')  # googletrans api
+    # return result.text                                    # googletrans api
 
 
-def translate(audio, url, seconds_max, dest_language):
+def translate(audio, url, seconds_max, target_lang):
     """
     Translates a YouTube video if a url is specified and returns the transcription.
     If not url is specified, it translates the audio file as passed by Gradio.
@@ -103,8 +106,8 @@ def translate(audio, url, seconds_max, dest_language):
         for i, (seconds, output) in enumerate(zip(segment_start_times, pred)):
             text += f"[Segment {i+1}/{n_segments}, start time {get_timestamp(seconds)}]\n"
             text += f"{output['text']}\n"
-            text += f"[Translation ({dest_language})]\n"
-            text += f"{get_translation(output['text'], dest_language)}\n\n"
+            text += f"[Translation ({target_lang})]\n"
+            text += f"{get_translation(output['text'], target_lang)}\n\n"
         return text
 
     else:
@@ -117,7 +120,7 @@ iface = gr.Interface(
         gr.Audio(source="microphone", type="filepath", label="Translate from Microphone"),
         gr.Text(max_lines=1, placeholder="Enter YouTube Link with Swedish speech to be translated", label="Translate from YouTube URL"),
         gr.Slider(minimum=30, maximum=300, value=30, step=30, label="Number of seconds to translate from YouTube URL"),
-        gr.Dropdown(languages, label="Destination language")                    
+        gr.Dropdown(languages, label="Target language")
     ], 
     outputs="text",
     title="Whisper Small Swedish",

requirements.txt

@@ -7,6 +7,4 @@ torch
 torchaudio
 moviepy
 git+https://github.com/pytube/pytube
-httpx==0.23.1
-httpcore==0.16.2
-googletrans==4.0.0-rc1
+google-trans-new