Update app.py

app.py

@@ -2,6 +2,41 @@ import streamlit as st
 import pandas as pd
 import plotly.express as px
 
+country_mapping = {
+    'United States': 'USA',
+    'United Kingdom': 'GBR',
+    'France': 'FRA',
+    'Canada': 'CAN',
+    'Germany': 'DEU',
+    'Japan': 'JPN',
+    'India': 'IND',
+    'Australia': 'AUS',
+    'China': 'CHN',
+    'Italy': 'ITA',
+    'Spain': 'ESP',
+    'Mexico': 'MEX',
+    'Hong Kong': 'HKG',
+    'Sweden': 'SWE',
+    'Denmark': 'DNK',
+    'New Zealand': 'NZL',
+    'Belgium': 'BEL',
+    'South Korea': 'KOR',
+    'Ireland': 'IRL',
+    'Czech Republic': 'CZE',
+    'Switzerland': 'CHE',
+    'Hungary': 'HUN',
+    'Norway': 'NOR',
+    'United Arab Emirates': 'ARE',
+    'Netherlands': 'NLD',
+    'South Africa': 'ZAF',
+    'Poland': 'POL',
+    'Austria': 'AUT',
+    'Turkey': 'TUR'
+}
+
+
+
+
 # Load your dataframes
 df_tv_series= pd.read_csv('series_after_cleaning.csv')
 df_movies= pd.read_csv('movie_after_cleaning.csv')
@@ -15,6 +50,7 @@ df_tv_series['country'] = df_tv_series['country'].str.split(',')
 def create_treemap(df, title):
     fig = px.treemap(df, path=['parentalguide'], title=title)
     return fig
+    
 def create_genre_bar_chart(df, title):
     # Explode the genre column to count each genre separately
     df_exploded = df.explode('genre')
@@ -23,12 +59,25 @@ 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()
+    # Explode the country column to count each country separately
+    df_exploded = df.explode('country')
+    country_counts = df_exploded['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)
+    
+    # Map country names to ISO codes
+    country_counts['country'] = country_counts['country'].map(country_mapping)
+    
+    fig = px.choropleth(country_counts, 
+                        locations="country",
+                        color="count",
+                        hover_name="country",
+                        title=title,
+                        projection="natural earth",
+                        color_continuous_scale='Viridis')
+    fig.update_layout(template='plotly_dark', font=dict(color='yellow'))
+    
     return fig
 
 # Streamlit app