global_func/load_csv.py

import streamlit as st
import numpy as np
import pandas as pd
import time
from rapidfuzz import process

def load_csv(upload):
    if upload is not None:
        try:
            if isinstance(upload, pd.DataFrame):
                df = upload.copy()
                df['Name + ID'] = df['Name'] + ' (' + df['ID'].astype(str) + ')'

                return df
            else:
                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