Fixed missing local variable

app.py

@@ -65,7 +65,7 @@ def translate_data(data, valid_languages_col, valid_languages_col_trg, col_for_t
     return new_df
 
 
-def select_column(data, is_excel=False):
+def select_column(data, valid_source, valid_target, is_excel=False):
 
     if is_excel:
         columns = (col for col in data[0].columns)
@@ -80,11 +80,11 @@ def select_column(data, is_excel=False):
     if src_col:
         col_src_lang = st.selectbox(
             'Source language:',
-            valid_languages_col,
+            valid_source,
         )
         col_trg_lang = st.selectbox(
             'Target language:',
-            valid_languages_col_trg,
+            valid_target,
         )
         submitted_cols = st.button("Translate column")
     
@@ -149,7 +149,7 @@ if uploaded_file is not None:
 
     if len(sheets) > 0:
         translated_sheets = []
-        submitted_cols, src_col, valid_col, valid_col_trg = select_column(data, is_excel=True)
+        submitted_cols, src_col, valid_col, valid_col_trg = select_column(data, valid_col, valid_col_trg, is_excel=True)
 
         if submitted_cols:
             for sheet in data:
@@ -158,7 +158,7 @@ if uploaded_file is not None:
         done = True
         
     else:
-        submitted_cols, src_col, valid_col, valid_col_trg = select_column(data)
+        submitted_cols, src_col, valid_col, valid_col_trg = select_column(data, valid_col, valid_col_trg)
         
         st.subheader("DataFrame")
         st.write(data)