Update app.py

app.py

@@ -77,7 +77,7 @@ def fetch_and_plot_average_rating_by_genre(conn):
     return fig
 
 # Function to create word cloud of genres
-def create_genre_wordcloud(conn):
+def create_genre_wordcloud(conn, mask_path):
     query = r'''
     SELECT genres
     FROM title_basics
@@ -89,14 +89,17 @@ def create_genre_wordcloud(conn):
     genres = df['genres'].str.split(',', expand=True).stack().replace('\\N', pd.NA).dropna().reset_index(drop=True)
     genre_counts = Counter(genres)
 
-    # Generate the word cloud
-    wordcloud = WordCloud(width=800, height=400, colormap='inferno').generate_from_frequencies(genre_counts)
+    # Load the mask image
+    mask = imageio.imread(mask_path)
+
+    # Generate the word cloud with the mask
+    wordcloud = WordCloud(width=800, height=400, background_color='white', mask=mask).generate_from_frequencies(genre_counts)
 
     # Display the word cloud
     plt.figure(figsize=(10, 5))
     plt.imshow(wordcloud, interpolation='bilinear')
     plt.axis('off')
-    st.pyplot(plt.gcf()) 
+    st.pyplot(plt.gcf())
 
 # Function to find best movie of each genre by numVotes * averageRating
 def find_best_movies_by_genre(conn):
@@ -299,7 +302,7 @@ def run_app():
         st.dataframe(best_movies_by_genre)
 
     with col2:
-        create_genre_wordcloud(conn)
+        create_genre_wordcloud(conn, 'mask.png')
 
     with col3:
         st.subheader('Average Rating by Genre')