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_categories_to_display):
+def main(file, num_clusters_to_display):
     try:
         df = pd.read_csv(file)
         
@@ -169,13 +169,13 @@ def main(file, num_categories_to_display):
         
         df = preprocess_data(df)
         
-        # Get category sizes and sort by size in descending order
+        # Get category sizes and sort by size in ascending order
         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 = category_sizes.sort_values(by='Count', ascending=True)['Category'].tolist()
         
-        # Get the largest x categories as specified by num_categories_to_display
-        largest_categories = sorted_categories[:num_categories_to_display]
+        # Get the largest x categories as specified by num_clusters_to_display
+        largest_categories = sorted_categories[:num_clusters_to_display]
         
         # Filter the dataframe to include only the largest categories
         filtered_df = df[df['Category'].isin(largest_categories)]
@@ -183,11 +183,11 @@ def main(file, num_categories_to_display):
         # Sort the dataframe by Category
         filtered_df = filtered_df.sort_values(by='Category')
         
-        wordcloud_img = generate_wordcloud(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
+        wordcloud_img = generate_wordcloud(filtered_df)
+        bar_chart_img = generate_bar_chart(filtered_df, num_clusters_to_display)
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
-            filtered_df.to_csv(tmpfile.name, index=False)
+            filtered_df.to_csv(tmpfile.name, index=False)x
             csv_file_path = tmpfile.name
 
         return csv_file_path, wordcloud_img, bar_chart_img
@@ -199,7 +199,7 @@ interface = gr.Interface(
     fn=main,
     inputs=[
         gr.File(label="Upload CSV File (.csv)"),
-        gr.Slider(label="Number of Categories to Display", minimum=1, maximum=15, step=1, value=5)
+        gr.Slider(label="Number of Categories to Display", minimum=1, maximum=10, step=1, value=5)
     ],
     outputs=[
         gr.File(label="Categorized Data CSV"),