Update app.py

app.py

@@ -48,15 +48,14 @@ if df1_upload is not None and df2_upload is not None:
     if not df1.empty and not df2.empty:
         columns_df1 = df1.columns.tolist()
         columns_df2 = df2.columns.tolist()
-        selected_columns_df1 = list(st.multiselect("Select columns from DataFrame 1 to match on:", columns_df1, default=columns_df1[0]))
-        selected_columns_df2 = list(st.multiselect("Select columns from DataFrame 2 to match on:", columns_df2, default=columns_df2[0]))
-        st.write(selected_columns_df1)
-        st.write(selected_columns_df2)
+        selected_columns_df1 = st.multiselect("Select columns from DataFrame 1 to match on:", columns_df1, default=columns_df1[0])
+        selected_columns_df2 = st.multiselect("Select columns from DataFrame 2 to match on:", columns_df2, default=columns_df2[0])
+
 
          # Perform merge
         if st.button("Merge DataFrames"):
             model=lt.LinkTransformer(model_path)
-            df_lm_matched = lt.merge(df2, df1, merge_type='1:m', on=None, model=model, left_on=selected_columns_df1, right_on=selected_columns_df2)
+            df_lm_matched = lt.merge(df1, df2, merge_type='1:m', on=None, model=model, left_on=selected_columns_df1, right_on=selected_columns_df2)
             st.write("Merged DataFrame Preview:")
             st.dataframe(df_lm_matched.head())