Enhance download functionality for contest data in app.py

- Added a browser selection option for downloading contest data, allowing users to choose between Chrome, Firefox, and Edge for cookie retrieval.
- Updated the cookie retrieval logic to accommodate the selected browser, improving the reliability of the download process.
- Enhanced error handling to provide clearer feedback when cookie access fails or when the downloaded file format is incorrect, ensuring a better user experience.

app.py

@@ -132,36 +132,50 @@ with tab1:
                 download_col, upload_col = st.columns(2)
                 with download_col:
                     st.markdown("### Download Option")
+                    browser_select = st.selectbox("Select your browser", ['Chrome', 'Firefox', 'Edge'])
                     if st.button("Download from DraftKings"):
                         try:
+                            
                             # Get the download URL
                             download_url = f"https://www.draftkings.com/contest/exportfullstandingscsv/{contest_id_map[contest_name_var]}"
                             
-                            # Get cookies from the user's browser
-                            cookies = browser_cookie3.load(domain_name='.draftkings.com')
-                            cookie_dict = {cookie.name: cookie.value for cookie in cookies}
-                            
-                            # Attempt to download using requests with browser cookies
-                            headers = {
-                                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
-                            }
-                            response = requests.get(download_url, cookies=cookie_dict, headers=headers)
-                            
-                            if response.status_code == 200:
-                                # Check if it's a zip file
-                                if 'application/zip' in response.headers.get('content-type', ''):
-                                    # Extract CSV from zip
-                                    with zipfile.ZipFile(BytesIO(response.content)) as zip_ref:
-                                        # Get the first CSV file in the zip
-                                        csv_filename = [f for f in zip_ref.namelist() if f.endswith('.csv')][0]
-                                        with zip_ref.open(csv_filename) as csv_file:
-                                            st.session_state['Contest_file'] = pd.read_csv(csv_file)
-                                    st.success("Successfully downloaded and loaded contest data!")
+                            # Get cookies from the user's selected browser
+                            try:
+                                if browser_select == 'Chrome':
+                                    cookies = browser_cookie3.chrome(domain_name='.draftkings.com')
+                                elif browser_select == 'Firefox':
+                                    cookies = browser_cookie3.firefox(domain_name='.draftkings.com')
+                                else:  # Edge
+                                    cookies = browser_cookie3.edge(domain_name='.draftkings.com')
+                                    
+                                cookie_dict = {cookie.name: cookie.value for cookie in cookies}
+                                
+                                # Attempt to download using requests with browser cookies
+                                headers = {
+                                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
+                                }
+                                response = requests.get(download_url, cookies=cookie_dict, headers=headers)
+                                
+                                if response.status_code == 200:
+                                    # Check if it's a zip file
+                                    if 'application/zip' in response.headers.get('content-type', ''):
+                                        # Extract CSV from zip
+                                        with zipfile.ZipFile(BytesIO(response.content)) as zip_ref:
+                                            # Get the first CSV file in the zip
+                                            csv_filename = [f for f in zip_ref.namelist() if f.endswith('.csv')][0]
+                                            with zip_ref.open(csv_filename) as csv_file:
+                                                st.session_state['Contest_file'] = pd.read_csv(csv_file)
+                                        st.success("Successfully downloaded and loaded contest data!")
+                                    else:
+                                        st.error("Downloaded file is not in the expected format. Please use manual upload instead.")
                                 else:
-                                    st.error("Downloaded file is not in the expected format. Please use manual upload instead.")
-                            else:
-                                st.error(f"Failed to download (Status code: {response.status_code}). You may need to log into DraftKings in your browser first.")
+                                    st.error(f"Failed to download (Status code: {response.status_code}). You may need to log into DraftKings in your browser first.")
+                                    st.info("Please use the manual upload option instead.")
+                                    
+                            except browser_cookie3.BrowserCookieError as e:
+                                st.error(f"Could not access cookies from {browser_select}. Make sure you're logged into DraftKings in {browser_select}.")
                                 st.info("Please use the manual upload option instead.")
+                                
                         except Exception as e:
                             st.error(f"Error during download: {str(e)}")
                             st.info("Please use the manual upload option instead.")