updating load_csv to handle dataframes

app.py

@@ -173,11 +173,6 @@ except:
     nfl_slate_name_lookup_fd = {}
 
 nfl_reg_salaries = grab_nfl_reg_salaries()
-print(nfl_reg_salaries.head(10))
-
-print(nfl_slate_names_dk)
-print(nfl_slate_name_lookup_dk)
-print(dk_db_nfl_showdown_selections)
 
 # Memory optimization helper functions
 def chunk_name_matching(portfolio_names, csv_names, chunk_size=1000):
@@ -605,7 +600,7 @@ if selected_tab == 'Data Load':
         upload_csv_col, csv_template_col = st.columns([3, 1])
         if pricing_source == 'Paydirt DB':
             if st.button("Load from Paydirt DB"):
-                csv_file = nfl_reg_salaries.to_csv(index=False)
+                csv_file = nfl_reg_salaries
                 if 'csv_file' in st.session_state:
                     del st.session_state['csv_file']
         else:

global_func/load_csv.py

@@ -7,17 +7,20 @@ from rapidfuzz import process
 def load_csv(upload):
     if upload is not None:
         try:
-            if upload.name.endswith('.csv'):
-                df = pd.read_csv(upload)
-                try:
-                    df['Name + ID'] = df['Name'] + ' (' + df['ID'].astype(str) + ')'
-                except:
-                    pass
+            if isinstance(upload, pd.DataFrame):
+                df = upload.copy()
             else:
-                st.error('Please upload either a CSV or Excel file')
-                return None
-            
-            return df
+                if upload.name.endswith('.csv'):
+                    df = pd.read_csv(upload)
+                    try:
+                        df['Name + ID'] = df['Name'] + ' (' + df['ID'].astype(str) + ')'
+                    except:
+                        pass
+                else:
+                    st.error('Please upload either a CSV or Excel file')
+                    return None
+                
+                return df
         except Exception as e:
             st.error(f'Error loading file: {str(e)}')
             return None