Spaces:
Running
Running
Update app.py
Browse files
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']
|
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
|
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 |
-
|
71 |
-
|
|
|
|
|
|
|
|
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})")
|