Lenylvt commited on
Commit
929fbce
·
verified ·
1 Parent(s): a5524a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -14,10 +14,9 @@ def fetch_languages(url):
14
  df = pd.read_csv(io.StringIO(csv_content), delimiter="|", skiprows=2, header=None).dropna(axis=1, how='all')
15
  df.columns = ['ISO 639-1', 'ISO 639-2', 'Language Name', 'Native Name']
16
  df['ISO 639-1'] = df['ISO 639-1'].str.strip()
17
- language_options = [(row['ISO 639-1'], f"{row['ISO 639-1']} - {row['Language Name'].strip()}") for index, row in df.iterrows()]
18
  return language_options
19
  else:
20
- st.error("Failed to fetch language options. Please try again later.")
21
  return []
22
 
23
  def translate_text(text, source_language_code, target_language_code):
@@ -38,7 +37,7 @@ def translate_text(text, source_language_code, target_language_code):
38
  return "\n".join(translated_texts)
39
 
40
  def translate_srt(input_file, source_language_code, target_language_code):
41
- subs = pysrt.open(input_file.name)
42
  translated_subs = []
43
  progress_bar = st.progress(0)
44
  for idx, sub in enumerate(subs):
@@ -67,5 +66,8 @@ target_language_code = st.selectbox("Select Target Language", options=language_o
67
  file_input = st.file_uploader("Upload SRT File", type=["srt"])
68
 
69
  if file_input is not None:
70
- translated_srt_path = translate_srt(file_input, source_language_code, target_language_code)
71
- st.success(f"Translation complete! You can download the translated SRT file from [here]({translated_srt_path})")
 
 
 
 
14
  df = pd.read_csv(io.StringIO(csv_content), delimiter="|", skiprows=2, header=None).dropna(axis=1, how='all')
15
  df.columns = ['ISO 639-1', 'ISO 639-2', 'Language Name', 'Native Name']
16
  df['ISO 639-1'] = df['ISO 639-1'].str.strip()
17
+ language_options = [(row['ISO 639-1'], f"{row['ISO 639-1']} - {row['Language Name']}") for index, row in df.iterrows()]
18
  return language_options
19
  else:
 
20
  return []
21
 
22
  def translate_text(text, source_language_code, target_language_code):
 
37
  return "\n".join(translated_texts)
38
 
39
  def translate_srt(input_file, source_language_code, target_language_code):
40
+ subs = pysrt.open(input_file)
41
  translated_subs = []
42
  progress_bar = st.progress(0)
43
  for idx, sub in enumerate(subs):
 
66
  file_input = st.file_uploader("Upload SRT File", type=["srt"])
67
 
68
  if file_input is not None:
69
+ with tempfile.NamedTemporaryFile(suffix=".srt", delete=False) as temp_file:
70
+ temp_file.write(file_input.read())
71
+ temp_file.seek(0)
72
+ translated_srt_path = translate_srt(temp_file.name, source_language_code, target_language_code)
73
+ st.success(f"Translation complete! You can download the translated SRT file from [here]({translated_srt_path})")