utils/helpers.py

import pandas as pd

def clean_data(data):
    """Cleans and preprocesses data before visualization."""
    return data.dropna().reset_index(drop=True) if isinstance(data, pd.DataFrame) else data