Add support for DraftKings/Fanduel file uploads in app.py: implement a new upload option for processing lineups, integrating a dedicated loading function for enhanced compatibility with different data sources.

app.py

@@ -10,6 +10,7 @@ import random
 from global_func.clean_player_name import clean_player_name
 from global_func.load_file import load_file
 from global_func.load_ss_file import load_ss_file
+from global_func.load_dk_fd_file import load_dk_fd_file
 from global_func.find_name_mismatches import find_name_mismatches
 from global_func.predict_dupes import predict_dupes
 from global_func.highlight_rows import highlight_changes, highlight_changes_winners, highlight_changes_losers
@@ -62,15 +63,14 @@ with tab1:
     with col2:
         st.subheader("Portfolio File")
         st.info("Go ahead and upload a portfolio file here. Only include player columns and an optional 'Stack' column if you are playing MLB.")
-        saber_toggle = st.radio("Are you uploading from SaberSim?", options=['No', 'Yes'])
-        st.info("If you are uploading from SaberSim, you will need to upload a CSV file for the slate for name matching.")
-        if saber_toggle == 'Yes':
-            if csv_file is not None:
-                portfolio_file = st.file_uploader("Upload Portfolio File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
-                if 'portfolio' in st.session_state:
-                    del st.session_state['portfolio']
-                if 'export_portfolio' in st.session_state:
-                    del st.session_state['export_portfolio']
+
+        upload_toggle = st.selectbox("What source are you uploading from?", options=['SaberSim (Just IDs)', 'Draftkings/Fanduel (Names + IDs)', 'Other (Just Names)'])
+        if upload_toggle == 'SaberSim (Just IDs)' or upload_toggle == 'Draftkings/Fanduel (Names + IDs)':
+            portfolio_file = st.file_uploader("Upload Portfolio File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
+            if 'portfolio' in st.session_state:
+                del st.session_state['portfolio']
+            if 'export_portfolio' in st.session_state:
+                del st.session_state['export_portfolio']
 
         else:
             portfolio_file = st.file_uploader("Upload Portfolio File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
@@ -80,12 +80,18 @@ with tab1:
                 del st.session_state['export_portfolio']
         if 'portfolio' not in st.session_state:
             if portfolio_file:
-                if saber_toggle == 'Yes':
+                if upload_toggle == 'SaberSim (Just IDs)':
                     st.session_state['export_portfolio'], st.session_state['portfolio'] = load_ss_file(portfolio_file, st.session_state['csv_file'])
                     st.session_state['export_portfolio'] = st.session_state['export_portfolio'].dropna(how='all')
                     st.session_state['export_portfolio'] = st.session_state['export_portfolio'].reset_index(drop=True)
                     st.session_state['portfolio'] = st.session_state['portfolio'].dropna(how='all')
                     st.session_state['portfolio'] = st.session_state['portfolio'].reset_index(drop=True)
+                elif upload_toggle == 'Draftkings/Fanduel (Names + IDs)':
+                    st.session_state['export_portfolio'], st.session_state['portfolio'] = load_dk_fd_file(portfolio_file, st.session_state['csv_file'])
+                    st.session_state['export_portfolio'] = st.session_state['export_portfolio'].dropna(how='all')
+                    st.session_state['export_portfolio'] = st.session_state['export_portfolio'].reset_index(drop=True)
+                    st.session_state['portfolio'] = st.session_state['portfolio'].dropna(how='all')
+                    st.session_state['portfolio'] = st.session_state['portfolio'].reset_index(drop=True)
                 else:
                     st.session_state['export_portfolio'], st.session_state['portfolio'] = load_file(portfolio_file)
                     st.session_state['export_portfolio'] = st.session_state['export_portfolio'].dropna(how='all')

global_func/load_dk_fd_file.py

@@ -0,0 +1,34 @@
+import streamlit as st
+import numpy as np
+import pandas as pd
+import time
+from fuzzywuzzy import process
+
+def load_dk_fd_file(lineups, csv_file):
+    df = csv_file.copy()
+    try:
+        name_dict = dict(zip(df['Name + ID'], df['Name']))
+    except:
+        name_dict = dict(zip(df['Id'], df['Nickname']))
+
+    # Now load and process the lineups file
+    try:
+        if lineups.name.endswith('.csv'):
+            lineups_df = pd.read_csv(lineups)
+        elif lineups.name.endswith(('.xls', '.xlsx')):
+            lineups_df = pd.read_excel(lineups)
+        else:
+            st.error('Please upload either a CSV or Excel file for lineups')
+            return None, None
+        
+        export_df = lineups_df.copy()
+        
+        # Map the IDs to names
+        for col in lineups_df.columns:
+            lineups_df[col] = lineups_df[col].map(name_dict)
+        
+        return export_df, lineups_df
+        
+    except Exception as e:
+        st.error(f'Error loading lineups file: {str(e)}')
+        return None, None