import pandas as pd
import streamlit as st

def columns_list(data:pd.DataFrame, no_depth:bool=None, no_well:bool=None):
    columns_list = list(data.columns.unique())
    columns_list.sort()
    if "DEPTH" in columns_list and no_depth == True:
        columns_list.remove("DEPTH")
    if "WELL" in columns_list and no_well == True:
        columns_list.remove("WELL")
    return columns_list

def well_filter(df, well_name):
    return df[df.WELL == well_name]