Update app.py

app.py

@@ -21,6 +21,13 @@ def create_genre_bar_chart(df, title):
     genre_counts = genre_counts.head(10)  # Get top 10 genres
     fig = px.bar(genre_counts, x='count', y='genre', orientation='h', title=title)
     return fig
+def create_country_map(df, title):
+    country_counts = df['country'].value_counts().reset_index()
+    country_counts.columns = ['country', 'count']
+    fig = px.choropleth(country_counts, locations="country", locationmode='country names',
+                        color="count", hover_name="country",
+                        color_continuous_scale=px.colors.sequential.Plasma, title=title)
+    return fig
 
 # Streamlit app
 st.title('Parental Guide Treemaps')
@@ -29,7 +36,7 @@ st.title('Parental Guide Treemaps')
 col1, col2 = st.columns(2)
 
 # Initialize variable for selection
-selection = None
+selection = 'Movies'
 
 # Add buttons in each column
 with col1:
@@ -44,7 +51,9 @@ with col2:
 if selection == 'Movies':
     st.plotly_chart(create_treemap(df_movies, 'Parental Guide - Movies'), use_container_width=True)
     st.plotly_chart(create_genre_bar_chart(df_movies, 'Top 10 Genres - Movies'), use_container_width=True)
+    st.plotly_chart(create_country_map(df_movies, 'Global Distribution of 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)
     st.plotly_chart(create_genre_bar_chart(df_tv_series, 'Top 10 Genres - TV Series'), use_container_width=True)
+    st.plotly_chart(create_country_map(df_tv_series, 'Global Distribution of TV Series'), use_container_width=True)