Update app.py

app.py

@@ -2,21 +2,6 @@ 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'
-    elif count <= 50:
-        return '10-50'
-    elif count <= 100:
-        return '50-100'
-    elif count <= 500:
-        return '100-500'
-    elif count <= 1000:
-        return '500-1000'
-    else:
-        return '>1000'
-
 # Country mapping dictionary
 country_mapping = {
     'United States': 'USA',
@@ -124,7 +109,7 @@ country_mapping = {
     'Maldives': 'MDV'
 }
 
-# Load your dataframes
+# Load your dataframes (replace with actual CSV filenames)
 df_movies = pd.read_csv('movies.csv')
 df_tv_series = pd.read_csv('tv_series.csv')
 
@@ -171,6 +156,11 @@ def create_country_map(df, title):
     
     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)
+    return fig
+
 # Streamlit app
 st.title('Parental Guide Analysis')
 
@@ -189,12 +179,14 @@ with col2:
     if st.button('TV Series'):
         selection = 'TV Series'
 
-# Display treemap, genre bar chart, and choropleth map based on selection
+# Display treemap, genre bar chart, rating distribution, and choropleth map based on selection
 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_rating_box_chart(df_movies, 'Rating Distribution - 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)
+    st.plotly_chart(create_rating_box_chart(df_tv_series, 'Rating Distribution - TV Series'), use_container_width=True)
+    st.plotly_chart(create_country_map(df_tv_series, 'Global Distribution of TV Series'), use_container_width=True)