data_loader.py

import pandas as pd

def load_data(file):
    if file.name.endswith('.csv'):
        return pd.read_csv(file.name)
    elif file.name.endswith('.xlsx'):
        return pd.read_excel(file.name)
    elif file.name.endswith('.json'):
        return pd.read_json(file.name)
    else:
        raise ValueError("Unsupported file type")