Update app.py

app.py

@@ -5,6 +5,7 @@ import plotly.express as px
 # Load your dataframes
 df_tv_series= pd.read_csv('series_after_cleaning.csv')
 df_movies= pd.read_csv('movie_after_cleaning.csv')
+
 df_movies['genre'] = df_movies['genre'].str.split(',')
 df_tv_series['genre'] = df_tv_series['genre'].str.split(',')
 
@@ -13,7 +14,9 @@ def create_treemap(df, title):
     fig = px.treemap(df, path=['parentalguide'], title=title)
     return fig
 def create_genre_bar_chart(df, title):
-    genre_counts = df['genre'].value_counts().reset_index()
+    # Explode the genre column to count each genre separately
+    df_exploded = df.explode('genre')
+    genre_counts = df_exploded['genre'].value_counts().reset_index()
     genre_counts.columns = ['genre', 'count']
     fig = px.bar(genre_counts, x='genre', y='count', title=title)
     return fig
@@ -41,6 +44,7 @@ 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 Genres - 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_treemap(df, 'Parental Guide - TV Series'), use_container_width=True)
+
     st.plotly_chart(create_genre_bar_chart(df_tv_series, 'Top Genres - TV Series'), use_container_width=True)