Update pages/02_📼_Upload_Video_File.py

pages/02_📼_Upload_Video_File.py

@@ -2,6 +2,7 @@ import whisper
 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
@@ -12,6 +13,9 @@ import os
 
 st.set_page_config(page_title="Semaitles", 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)
 def load_lottieurl(url: str):
@@ -102,7 +106,33 @@ def generate_subtitled_video(video, audio, transcript):
     video_with_subs = open("final.mp4", "rb")
     return video_with_subs
 
+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():
     size = st.selectbox("Select Model Size (The larger the model, the more accurate the transcription will be, but it will take longer)", ["tiny", "base", "small", "medium", "large"], index=1)
     loaded_model = change_model(current_size, size)