Update app.py

app.py

@@ -109,7 +109,10 @@ def fetch_and_plot_average_rating_by_genre(conn):
     return fig
 
 # Function to create word cloud of genres
-def create_genre_wordcloud(conn, mask_path):
+def genre_color_func(word, font_size, position, orientation, random_state=None, **kwargs):
+    return genre_color_map.get(word, '#FFFFFF')  # Default to white if genre is not in the map
+
+def create_genre_wordcloud(conn):
     query = r'''
     SELECT genres
     FROM title_basics
@@ -121,11 +124,19 @@ def create_genre_wordcloud(conn, mask_path):
     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 with the mask
-    wordcloud = WordCloud(width=800, height=400, background_color='white').generate_from_frequencies(genre_counts)
+    # Create a circular mask
+    mask = np.zeros((800, 800), dtype=np.uint8)
+    center_x, center_y = 400, 400
+    radius = 400
+    y, x = np.ogrid[:800, :800]
+    mask_area = (x - center_x)**2 + (y - center_y)**2 <= radius**2
+    mask[mask_area] = 255
+
+    # Generate the word cloud with the circular mask and custom colors
+    wordcloud = WordCloud(width=800, height=800, background_color='white', mask=mask, color_func=genre_color_func).generate_from_frequencies(genre_counts)
 
     # Display the word cloud
-    plt.figure(figsize=(10, 5))
+    plt.figure(figsize=(10, 10))
     plt.imshow(wordcloud, interpolation='bilinear')
     plt.axis('off')
     st.pyplot(plt.gcf())