example_app.py

import streamlit as st
import fancylit
import pandas as pd
from sklearn.linear_model import Lasso

###
# Streamlit Main Functionality
###


def stats_examples(df: pd.DataFrame) -> None:
    """
    Purpose:
        Shows examples for modeling
    Args:
        N/A
    Returns:
        N/A
    """

    # Stats Example
    st.write("Describe Example")
    with st.echo():
        fancylit.stats.df_describe(df)
        fancylit.stats.show_metrics(df)
        fancylit.stats.gen_wordcloud(df)


def modeling_viz_examples(df: pd.DataFrame) -> None:
    """
    Purpose:
        Shows examples for modeling
    Args:
        N/A
    Returns:
        N/A
    """

    # classification_report example
    st.write("Classification Report Example")
    with st.echo():
        fancylit.yellowbrick_funcs.show_classification_report(df)
    
    st.write("Feature Correlation")
    with st.echo():
        fancylit.yellowbrick_funcs.feature_correlation(df) 

    st.write("Class balance")
    with st.echo():
        fancylit.yellowbrick_funcs.class_balance(df)
    
    # st.write("UMAP Corpus Visualization Example")
    # with st.echo():
    #     fancylit.yellowbrick_funcs.umap_viz(df)
    
    st.write("Prediction Error Plot")
    with st.echo():
        fancylit.yellowbrick_funcs.prediction_error(df, Lasso()) 


def viz_examples(df: pd.DataFrame) -> None:
    """
    Purpose:
        Shows examples for viz
    Args:
        N/A
    Returns:
        N/A
    """

    # Start bar chart example
    st.write("Bar Chart Example")
    with st.echo():
        fancylit.viz.charts.bar_chart(df)

    st.write("3D Chart Example")
    with st.echo():
        fancylit.viz.charts.chart_3d(df)

    # Start pair plot example
    st.write("Pair Plot Example")
    with st.echo():
        fancylit.viz.charts.pair_plot(df)

    # Start Scatter Plot example
    st.write("Scatter Plot Example")
    with st.echo():
        fancylit.viz.charts.scatter_plot(df)

    # Start Line Chart example
    st.write("Line Chart Example")
    with st.echo():
        fancylit.viz.charts.line_chart(df)
    
    # Start heatmap example
    st.write("Heatmap Example")
    with st.echo():
        fancylit.viz.charts.heatmap(df) 

def sidebar() -> None:
    """
    Purpose:
        Shows the side bar
    Args:
        N/A
    Returns:
        N/A
    """

    st.sidebar.header(f"Fancylit Example")


def app() -> None:
    """
    Purpose:
        Controls the app flow
    Args:
        N/A
    Returns:
        N/A
    """

    # Spin up the sidebar
    # sidebar()

    # load example csv
    df = pd.read_csv("datasets/iris.csv")

    st.header("Fancylit Examples")
    st.write(
        "The following examples highlight what fancylit can do using the following dataset")

    st.write("Iris CSV data")
    st.write(df)

    # Start examples

    st.subheader("Stats Examples")
    stats_examples(df)

    st.subheader("Viz Examples")
    viz_examples(df)

    st.subheader("Modeling Examples")
    modeling_viz_examples(df)


def main() -> None:
    """
    Purpose:
        Controls the flow of the streamlit app
    Args:
        N/A
    Returns:
        N/A
    """

    # Start the streamlit app
    app()


if __name__ == "__main__":
    main()