Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import pandas as pd
 import plotly.express as px
+
+# Function to assign color based on count
 def assign_color(count):
     if count <= 10:
         return '1-10'
@@ -14,6 +16,7 @@ def assign_color(count):
         return '500-1000'
     else:
         return '>1000'
+
 # Country mapping dictionary
 country_mapping = {
     'United States': 'USA',
@@ -121,6 +124,7 @@ country_mapping = {
     'Maldives': 'MDV'
 }
 
+# Load your dataframes (replace with actual CSV filenames)
 df_movies = pd.read_csv('movie_after_cleaning.csv')
 df_tv_series = pd.read_csv('series_after_cleaning.csv')
 
@@ -141,7 +145,11 @@ def create_genre_bar_chart(df, title):
     genre_counts = df_exploded['genre'].value_counts().reset_index()
     genre_counts.columns = ['genre', 'count']
     genre_counts = genre_counts.head(10)  # Top 10 genres
-    fig = px.bar(genre_counts, x='count', y='genre', orientation='h', title=title)
+    fig = px.bar(genre_counts, x='count', y='genre', orientation='h', title=title, 
+                 labels={'count': 'Count', 'genre': 'Genre'},
+                 color_discrete_sequence=['#FFA07A'])
+    fig.update_traces(marker_line_color='rgb(8,48,107)', marker_line_width=1.5, opacity=0.6)
+    fig.update_layout(title_font_size=20, title_font_family='Arial', title_font_color='#00308F')
     return fig
 
 # Function to create choropleth map
@@ -165,11 +173,22 @@ def create_country_map(df, title):
                         color_discrete_sequence=['#7FFF00', '#FFD700', '#FFA500', '#FF4500', '#DC143C', '#8B0000'],
                         category_orders={"color": ['1-10', '10-50', '50-100', '100-500', '500-1000', '>1000']})
     
+    fig.update_geos(showcoastlines=True, coastlinecolor="LightBlue", showland=True, landcolor="LightGreen",
+                    showocean=True, oceancolor="LightBlue", showlakes=True, lakecolor="LightBlue", 
+                    showrivers=True, rivercolor="LightBlue")
+    
+    fig.update_layout(title_font_size=20, title_font_family='Arial', title_font_color='#00308F')
     return fig
 
 # Function to create rating distribution box chart
 def create_rating_box_chart(df, title):
-    fig = px.box(df, x="rating", points="all", title=title)
+    fig = px.box(df, x="rating", points="all", title=title, 
+                 labels={'rating': 'Rating'},
+                 boxmean=True, 
+                 orientation='h',
+                 color_discrete_sequence=['#FF6347'])
+    fig.update_traces(marker_line_color='rgb(8,48,107)', marker_line_width=1.5, opacity=0.6)
+    fig.update_layout(title_font_size=20, title_font_family='Arial', title_font_color='#00308F')
     return fig
 
 # Streamlit app