Update app.py

app.py

@@ -9,25 +9,24 @@ def convert_df_to_csv(df):
 st.title('DataFrame Merger using LinkTransformer')
 
 # Function to load DataFrame
-def load_dataframe(upload, path):
+def load_dataframe(upload):
+    ##if csv is uploaded use read_csv to load the data , otherwise use read_excel
     if upload is not None:
-        return pd.read_csv(upload)
-    elif path != "":
-        return pd.read_csv(path)
+        if upload.name.endswith('csv'):
+            return pd.read_csv(upload)
+        else:
+            return pd.read_excel(upload)
     else:
-        return None
-
+        return pd.DataFrame()
 # Options for DataFrame 1
 df1_upload = st.file_uploader("Upload DataFrame 1 (CSV)", type=['csv'], key='df1_upload')
-df1_path = st.text_input("...or enter path for DataFrame 1 (CSV)", key='df1_path')
 
 # Options for DataFrame 2
 df2_upload = st.file_uploader("Upload DataFrame 2 (CSV)", type=['csv'], key='df2_upload')
-df2_path = st.text_input("...or enter path for DataFrame 2 (CSV)", key='df2_path')
 
 # Load and display the DataFrames
-df1 = load_dataframe(df1_upload, df1_path)
-df2 = load_dataframe(df2_upload, df2_path)
+df1 = load_dataframe(df1_upload)
+df2 = load_dataframe(df2_upload)
 
 if df1 is not None:
     st.write("DataFrame 1 Preview:")
@@ -63,4 +62,8 @@ if not df1.empty and not df2.empty:
             data=csv,
             file_name='merged_dataframe.csv',
             mime='text/csv',
-        )
+        )
+else:
+    st.write("Please upload or enter paths for both DataFrames.")
+        
+