dabbu2000's picture
Update app.py
64d5363
import streamlit as st
from easynmt import EasyNMT
languageTranslationModelValue = EasyNMT('opus-mt')
languagesSourceDestination = [('en', 'es')]
def languageTranslationValue(languageSentenceOriginated, languageOriginalContent, languageTranslatedContent):
return languageTranslationModelValue.translate(languageSentenceOriginated, source_lang=languageOriginalContent, target_lang=languageTranslatedContent)
def languageShowMarkdown():
st.markdown("Video Translation For Subtitles")
st.sidebar.markdown('Video Translation Language From English to Spanish')
st.sidebar.text("\n")
st.sidebar.markdown("English to Spanish Translation Language")
st.sidebar.text("\n")
return
languageShowMarkdown()
languageSourceValue = st.file_uploader("",type=['srt'])
if languageSourceValue is not None:
translationValueEntireData = [translationValueLineOrigin for translationValueLineOrigin in languageSourceValue]
translationLanguageAmountLines = sum(1 for translationValueLineOrigin in translationValueEntireData)
translationValueNextStep = int(translationLanguageAmountLines/100)
for translationValueFirstIndex, translationValueLanguagePairGeneration in enumerate(languagesSourceDestination):
if translationValueFirstIndex==0:
st.markdown('Processing __English__ to __Spanish__'.format(translationValueLanguagePairGeneration[0], translationValueLanguagePairGeneration[1]))
translationValueOutputFile=open('English_Spanish.srt'.format(translationValueLanguagePairGeneration[0], translationValueLanguagePairGeneration[1]), 'w')
dropdownLanguageGeneration = st.progress(0)
languageGenerationSRTText = ""
q=0
for translationValueSecondIndex, translationValueLineOrigin in enumerate(translationValueEntireData):
print (translationValueSecondIndex, q, translationLanguageAmountLines, translationValueNextStep)
translationValueLineOrigin = str(translationValueLineOrigin)
if q!=100:
if translationValueSecondIndex%translationValueNextStep==0:
q=q+1
dropdownLanguageGeneration.progress(q)
if translationValueLineOrigin[0].isnumeric():
languageGenerationSRTText += translationValueLineOrigin
else:
translationValueConvertedOutput = languageTranslationValue(translationValueLineOrigin, translationValueLanguagePairGeneration[0], translationValueLanguagePairGeneration[1])
languageGenerationSRTText += translationValueConvertedOutput
translationValueOutputFile.write(languageGenerationSRTText)
if translationValueFirstIndex==len(languagesSourceDestination)-1:
st.markdown('The respective translations for the video from English to Spanish has been done!')
st.write('The following files with the translated language content are saved locally on the disk!')