Update app.py

app.py

@@ -17,11 +17,20 @@ st.title('Parental Guide Treemaps')
 # Split into two columns
 col1, col2 = st.columns(2)
 
+# Initialize variable for selection
+selection = None
+
 # Add buttons in each column
 with col1:
     if st.button('Movies'):
-        st.plotly_chart(create_treemap(df_movies, 'Parental Guide - Movies'))
+        selection = 'Movies'
 
 with col2:
     if st.button('TV Series'):
-        st.plotly_chart(create_treemap(df_tv_series, 'Parental Guide - TV Series'))
+        selection = 'TV Series'
+
+# Display the corresponding treemap in the center
+if selection == 'Movies':
+    st.plotly_chart(create_treemap(df_movies, 'Parental Guide - Movies'), use_container_width=True)
+elif selection == 'TV Series':
+    st.plotly_chart(create_treemap(df_tv_series, 'Parental Guide - TV Series'), use_container_width=True)