Update 01_🎥_Input_YouTube_Link.py

01_🎥_Input_YouTube_Link.py

@@ -11,8 +11,9 @@ from io import StringIO
 from utils import write_vtt, write_srt
 import ffmpeg
 from languages import LANGUAGES
+from flores200_codes import flores_codes
 
-st.set_page_config(page_title="Auto Subtitled Video Generator ", page_icon=":movie_camera:", layout="wide")
+st.set_page_config(page_title="Sema Titles", page_icon="🎦", layout="wide")
 
 # Define a function that we can use to load lottie files from a link.
 @st.cache()
@@ -29,10 +30,10 @@ with col1:
 
 with col2:
     st.write("""
-    ## Auto Subtitled Video Generator 
+    ## Sema-Titles
     ##### Input a YouTube video link and get a video with subtitles.
     ###### ➠ If you want to transcribe the video in its original language, select the task as "Transcribe"
-    ###### ➠ If you want to translate the subtitles to English, select the task as "Translate" 
+    ###### ➠ If you want to translate the subtitles from English to any of the 200 supported languages, select the task as "Translate" 
     ###### I recommend starting with the base model and then experimenting with the larger models, the small and medium models often work well. """)
     
 
@@ -116,6 +117,32 @@ def get_language_code(language):
     else:
         raise ValueError("Language not supported")
 
+def translate(userinput, target_lang, source_lang=None):
+    if source_lang:
+       url = f"{Public_Url}/translate_enter/"
+       data = {
+           "userinput": userinput,
+           "source_lang": source_lang,
+           "target_lang": target_lang,
+        }
+       response = requests.post(url, json=data)
+       result = response.json()
+       print(type(result))
+       source_lange = source_lang
+       translation = result['translated_text']
+       
+    else:
+      url = f"{Public_Url}/translate_detect/"
+      data = {
+        "userinput": userinput,
+        "target_lang": target_lang,
+      }
+
+      response = requests.post(url, json=data)
+      result = response.json()
+      source_lange = result['source_language']
+      translation = result['translated_text']
+    return source_lange, translation
 
 def generate_subtitled_video(video, audio, transcript):
     video_file = ffmpeg.input(video)
@@ -192,7 +219,10 @@ def main():
                                     data=video_with_subs,
                                     file_name=f"{title} with subtitles.mp4")
     elif task == "Translate":
-        if st.button("Translate to English"):
+        default_language = "French"
+        target = st.selectbox("Select Language", list(languages.keys()), index=list(languages.keys()).index(default_language))
+        target_code = languages[selected_language]
+        if st.button("Translate"):
             author, title, description, thumbnail, length, views = populate_metadata(link)
             results = inference(link, loaded_model, task)
             video = download_video(link)
@@ -255,4 +285,4 @@ def main():
 
 if __name__ == "__main__":
     main()
-    st.markdown("###### Made with :heart: by [@BatuhanYılmaz](https://github.com/BatuhanYilmaz26) [![this is an image link](https://i.imgur.com/thJhzOO.png)](https://www.buymeacoffee.com/batuhanylmz)")
+    #st.markdown("###### Made with :heart: by [@BatuhanYılmaz](https://github.com/BatuhanYilmaz26) [![this is an image link](https://i.imgur.com/thJhzOO.png)](https://www.buymeacoffee.com/batuhanylmz)")