import streamlit as st
import pandas as pd
import plotly.express as px

# Load your dataframes
df_tv_series= pd.read_csv('series_after_cleaning.csv')
df_movies= pd.read_csv('movie_after_cleaning.csv')

# Function to generate treemap
def create_treemap(df, title):
    fig = px.treemap(df, path=['parentalguide'], title=title)
    return fig

# Streamlit app
st.title('Parental Guide Treemaps')

# Split into two columns
col1, col2 = st.columns(2)

# Add buttons in each column
with col1:
    if st.button('Movies'):
        st.plotly_chart(create_treemap(df_movies, 'Parental Guide - Movies'))

with col2:
    if st.button('TV Series'):
        st.plotly_chart(create_treemap(df_tv_series, 'Parental Guide - TV Series'))