Update pages/03_📝_Upload_Video_File_and_Transcript.py

pages/03_📝_Upload_Video_File_and_Transcript.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from streamlit_lottie import st_lottie
 from utils import write_vtt, write_srt
+from flores200_codes import flores_codes
 import ffmpeg
 import requests
 from typing import Iterator
@@ -10,7 +11,10 @@ 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="Captions", page_icon=":movie_camera:", layout="wide")
+
+# Sema Translator
+Public_Url = 'https://lewiskimaru-helloworld.hf.space' #endpoint
 
 # Define a function that we can use to load lottie files from a link.
 @st.cache(allow_output_mutation=True)
@@ -36,7 +40,7 @@ with col1:
 
 with col2:
     st.write("""
-    ## Auto Subtitled Video Generator 
+    ## Sema Captions
     ##### ➠ Upload a video file and a transcript as .srt or .vtt file and get a video with subtitles.
     ##### ➠ Processing time will increase as the video length increases. """)
 
@@ -62,6 +66,32 @@ def split_video_audio(uploaded_file):
     audio = ffmpeg.output(audio, f"{save_dir}/output.wav", acodec="pcm_s16le", ac=1, ar="16k")
     ffmpeg.run(audio, overwrite_output=True)
 
+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 main():
     uploaded_video = st.file_uploader("Upload Video File", type=["mp4", "avi", "mov", "mkv"])