Update app.py

app.py

@@ -3,8 +3,8 @@ import pandas as pd
 import plotly.express as px
 
 # Load your dataframes
-df1= pd.read_csv('series_after_cleaning.csv')
-df2= pd.read_csv('movie_after_cleaning.csv')
+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):
@@ -14,11 +14,14 @@ def create_treemap(df, title):
 # Streamlit app
 st.title('Parental Guide Treemaps')
 
-# Add buttons to switch between dataframes
-option = st.selectbox('Select DataFrame', ['DataFrame 1', 'DataFrame 2'])
+# Split into two columns
+col1, col2 = st.columns(2)
 
-# Display the corresponding treemap based on the selected dataframe
-if option == 'DataFrame 1':
-    st.plotly_chart(create_treemap(df1, 'Parental Guide - DataFrame 1'))
-else:
-    st.plotly_chart(create_treemap(df2, 'Parental Guide - DataFrame 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'))