src/utils/visualization.py

import seaborn as sns
import matplotlib.pyplot as plt
import streamlit as st

def plot_distribution(df, selected_parameter):
    """Plots the distribution of a selected parameter."""
    
    try:
        fig, ax = plt.subplots()
        sns.histplot(df[selected_parameter], kde=True, ax=ax)
        ax.set_title(f'Distribution of {selected_parameter}')
        ax.set_xlabel(selected_parameter)
        ax.set_ylabel('Frequency')
        st.pyplot(fig)
    except Exception as e:
        st.write(f"An error occurred while plotting: {e}")
        
def plot_cumulative_frequency(df):
    try:
        fig, ax = plt.subplots()
        sns.ecdfplot(df['Longest Feret Diameter'], ax=ax)
        ax.set_title(f'Cumulative Frequency Plot')
        ax.set_xlabel('Grains diameter')
        ax.set_ylabel('Cumulative Frequency')
        st.pyplot(fig)
    except Exception as e:
        st.write(f"An error occurred while plotting: {e}")