File size: 1,126 Bytes
a35b524 9d66c9c a35b524 9d66c9c a35b524 9d66c9c a35b524 9d66c9c dbc3f23 a35b524 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import streamlit as st
import numpy as np
import pandas as pd
import time
from fuzzywuzzy 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
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 |