Update app.py

app.py

@@ -160,7 +160,7 @@ def generate_bar_chart(df, num_clusters_to_display):
     img = Image.open(buf)
     return img
 
-def main(file, num_clusters_to_display):
+def main(file, num_categories_to_display):
     try:
         df = pd.read_csv(file)
         
@@ -169,26 +169,22 @@ def main(file, num_clusters_to_display):
         
         df = preprocess_data(df)
         
-        # Get category sizes and sort by size in ascending order
+        # Get category sizes and sort by size in descending order
         category_sizes = df['Category'].value_counts().reset_index()
         category_sizes.columns = ['Category', 'Count']
-        sorted_categories = category_sizes.sort_values(by='Count', ascending=True)['Category'].tolist()
+        sorted_categories = category_sizes.sort_values(by='Count', ascending=False)['Category'].tolist()
         
-        # Get the largest x categories as specified by num_clusters_to_display
-        largest_categories = sorted_categories[:num_clusters_to_display]
-
-        print(largest_categories)
+        # Get the largest x categories as specified by num_categories_to_display
+        largest_categories = sorted_categories[:num_categories_to_display]
         
         # Filter the dataframe to include only the largest categories
         filtered_df = df[df['Category'].isin(largest_categories)]
-
-        
         
         # Sort the dataframe by Category
         filtered_df = filtered_df.sort_values(by='Category')
         
-        wordcloud_img = generate_wordcloud(filtered_df)
-        bar_chart_img = generate_bar_chart(df, num_clusters_to_display)
+        wordcloud_img = generate_wordcloud(filtered_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:
             filtered_df.to_csv(tmpfile.name, index=False)
@@ -199,6 +195,8 @@ def main(file, num_clusters_to_display):
         print(f"Error: {e}")
         return str(e), None, None
 
+
+
 interface = gr.Interface(
     fn=main,
     inputs=[