global_func/load_csv.py

import streamlit as st
import numpy as np
import pandas as pd
import time
from fuzzywuzzy 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
            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
    return None