Update app.py

app.py

@@ -6,7 +6,7 @@ import linktransformer as lt
 def convert_df_to_csv(df):
     return df.to_csv().encode('utf-8')
 
-st.title('DataFrame Merger using LinkTransformer')
+st.title('Merge Dataframes using LinkTransformer')
 
 # Function to load DataFrame
 def load_dataframe(upload):
@@ -39,31 +39,37 @@ if df2 is not None:
 
 # Model selection
 model_path = st.text_input("Model path (HuggingFace or local)", value="all-MiniLM-L6-v2")
+##More on model selection available on https://linktransformer.github.io/
 
-# Checkbox for columns to match on
-if not df1.empty and not df2.empty:
-    columns_df1 = df1.columns.tolist()
-    columns_df2 = df2.columns.tolist()
-    
-    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)
-        st.write("Merged DataFrame Preview:")
-        st.dataframe(df_lm_matched.head())
+if not df1==None and not df2==None:
+    # Checkbox for columns to match on
+    if not df1.empty and not df2.empty:
+        columns_df1 = df1.columns.tolist()
+        columns_df2 = df2.columns.tolist()
+        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)
+            st.write("Merged DataFrame Preview:")
+            st.dataframe(df_lm_matched.head())
 
-        # Download button for merged DataFrame
-        csv = convert_df_to_csv(df_lm_matched)
-        st.download_button(
-            label="Download merged DataFrame as CSV",
-            data=csv,
-            file_name='merged_dataframe.csv',
-            mime='text/csv',
-        )
+            # Download button for merged DataFrame
+            csv = convert_df_to_csv(df_lm_matched)
+            st.download_button(
+                label="Download merged DataFrame as CSV",
+                data=csv,
+                file_name='merged_dataframe.csv',
+                mime='text/csv',
+            )
+    else:
+        st.write("It appears that your dataframes are empty. Please upload valid dataframes.")
+
+       
 else:
     st.write("Please upload or enter paths for both DataFrames.")
-        
-    
+            
+##Add website and citation
+st.write("For more information, please visit the [LinkTransformer website](https://linktransformer.github.io/).")
+st.write("If you use LinkTransformer in your research, please cite the following paper: [LinkTransformer: A Unified Package for Record Linkage with Transformer Language Models](https://arxiv.org/abs/2309.00789)")