Update pages/02_📼_Upload_Video_File.py

pages/02_📼_Upload_Video_File.py

@@ -10,7 +10,7 @@ import numpy as np
 import pathlib
 import os
 
-st.set_page_config(page_title="Auto Subtitled Video Generator", page_icon=":movie_camera:", layout="wide")
+st.set_page_config(page_title="Semaitles", page_icon=":movie_camera:", layout="wide")
 
 # Define a function that we can use to load lottie files from a link.
 @st.cache(allow_output_mutation=True)
@@ -40,11 +40,11 @@ with col1:
 
 with col2:
     st.write("""
-    ## Auto Subtitled Video Generator 
+    ## Semaitles 
     ##### Upload a video file 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" 
-    ###### I recommend starting with the base model and then experimenting with the larger models, the small and medium models often work well. """)
+    ###### ➠ If you want to translate the subtitles to English, select the task as "Translate with Whisper" 
+    ###### ➠ If you want to translate the subtitles from English to any of the 200 supported languages, select the task as "Translate with Sema" """)
 
 
 @st.cache(allow_output_mutation=True)
@@ -114,7 +114,7 @@ def main():
         filename = input_file.name[:-4]
     else:
         filename = None
-    task = st.selectbox("Select Task", ["Transcribe", "Translate"], index=0)
+    task = st.selectbox("Select Task", ["Transcribe", "Translate with Whisper", "Translate with Sema"], index=0)
     if task == "Transcribe":
         if st.button("Transcribe"):
             results = inferecence(loaded_model, input_file, task)
@@ -168,7 +168,7 @@ def main():
                 st.download_button(label="Download Video with Subtitles",
                                 data=video_with_subs,
                                 file_name=f"{filename}_with_subs.mp4")
-    elif task == "Translate":
+    elif task == "Translate with Whisper":
         if st.button("Translate to English"):
             results = inferecence(loaded_model, input_file, task)
             col3, col4 = st.columns(2)
@@ -221,10 +221,17 @@ def main():
                 st.download_button(label="Download Video with Subtitles ",
                                 data=video_with_subs,
                                 file_name=f"{filename}_with_subs.mp4")
+                
+    elif task == "Translate with Sema":
+        default_language = "French"
+        target = st.selectbox("Select Language", list(flores_codes.keys()), index=list(flores_codes.keys()).index(default_language))
+        target_code = flores_codes[target]
+        
     else:
         st.error("Please select a task.")
 
 
 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("###### ")
+    st.markdown("###### Powered by [sema © 2024](https://www.sema.wiki)")