James McCool
Refactor import statements across multiple files to replace 'fuzzywuzzy' with 'rapidfuzz' for improved performance and consistency in string matching functionality. Additionally, clean up unused imports in app.py and related global functions to enhance code clarity and maintainability.
d9db89f
| import streamlit as st | |
| import numpy as np | |
| import pandas as pd | |
| import time | |
| from rapidfuzz import process | |
| import re | |
| ## import global functions | |
| from global_func.clean_player_name import clean_player_name | |
| def load_file(upload): | |
| if upload is not None: | |
| try: | |
| clean_name = re.sub(r' \(\d+\)', '', upload.name) | |
| print(clean_name) | |
| print(upload.name) | |
| if clean_name.endswith('.csv'): | |
| df = pd.read_csv(upload) | |
| elif clean_name.endswith(('.xls', '.xlsx')): | |
| df = pd.read_excel(upload) | |
| else: | |
| st.error('Please upload either a CSV or Excel file') | |
| return None, None | |
| for col in df.columns: | |
| if "Unnamed" in col: | |
| df = df.drop(columns=[col]) | |
| export_df = df.copy() | |
| for col in df.columns: | |
| if df[col].dtype == 'object': | |
| df[col] = df[col].apply(lambda x: clean_player_name(x) if isinstance(x, str) else x) | |
| return export_df, df | |
| except Exception as e: | |
| st.error(f'Error loading file: {str(e)}') | |
| return None | |
| return None |