Update app.py

app.py

@@ -173,17 +173,22 @@ def main(file, num_categories_to_display):
         category_sizes = df['Category'].value_counts().reset_index()
         category_sizes.columns = ['Category', 'Count']
         sorted_categories = category_sizes.sort_values(by='Count', ascending=False)['Category'].tolist()
+        sorted_categories_sm = category_sizes.sort_values(by='Count', ascending=False)['Category'].tolist()
+
         
         # Get the largest x categories as specified by num_categories_to_display
         largest_categories = sorted_categories[:num_categories_to_display]
+        smallest_categories = sorted_categories_sm[:num_categories_to_display]
         
         # Filter the dataframe to include only the largest categories
         filtered_df = df[df['Category'].isin(largest_categories)]
+        filtered_cloud_df = df[df['Category'].isin(smallest_categories)]
         
         # Sort the dataframe by Category
         filtered_df = filtered_df.sort_values(by='Category')
+        filtered_cloud_df = filtered_cloud_df.sort_values(by='Category')
         
-        wordcloud_img = generate_wordcloud(filtered_df)  # Word cloud based on the entire dataframe
+        wordcloud_img = generate_wordcloud(filtered_cloud_df)  # Word cloud based on the entire dataframe
         bar_chart_img = generate_bar_chart(df, num_categories_to_display)  # Bar chart based on the filtered dataframe
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile: