Refactored two variables in translate_data() to properly load src/trg language

app.py

@@ -45,14 +45,14 @@ def open_input(the_file):
     return parsed, sheets
 
 
-def translate_data(df, valid_languages_col, valid_languages_col_trg, col_for_translation, languages):
+def translate_data(df, s_lang, t_lang, col_for_translation, languages):
     translated_data = []
     new_df = df
     for text in df[col_for_translation]:
-        if len(text) > 0 and col_src_lang in languages and col_trg_lang in languages:
+        if len(text) > 0 and s_lang in languages and t_lang in languages:
             with st.spinner("Translating..."):
                 try:
-                    target_text = get_translation(valid_languages_col, valid_languages_col_trg, text)[0]
+                    target_text = get_translation(s_lang, t_lang, text)[0]
                     translated_data.append(target_text)
                 except:
                     st.subheader("Translation failed :sad:")
@@ -149,11 +149,11 @@ if uploaded_file is not None:
 
     if len(sheets) > 0:
         translated_sheets = []
-        submitted_cols, src_col, valid_col, valid_col_trg = select_column(data, valid_col, valid_col_trg, is_excel=True)
+        submitted_cols, src_col, src_code, trg_code = select_column(data, valid_col, valid_col_trg, is_excel=True)
 
         if submitted_cols:
             for sheet in data:
-                translated_sheets.append(translate_data(data, valid_col, valid_col_trg, src_col, valid_languages))
+                translated_sheets.append(translate_data(sheet, src_code, trg_code, src_col, valid_languages))
                 
             done = True
         
@@ -189,4 +189,4 @@ if done:
 
     
 else:
-    st.info("☝️ Upload a TSV file")
+    st.info("☝️ Upload a TSV file")