Update app.py

app.py

@@ -184,7 +184,7 @@ def main(file, num_clusters_to_display):
         filtered_df = filtered_df.sort_values(by='Category')
         
         wordcloud_img = generate_wordcloud(filtered_df)
-        bar_chart_img = generate_bar_chart(filtered_df, num_clusters_to_display)
+        bar_chart_img = generate_bar_chart(df, num_clusters_to_display)
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
             filtered_df.to_csv(tmpfile.name, index=False)
@@ -211,72 +211,3 @@ interface = gr.Interface(
 )
 
 interface.launch(share=True)
-
-
-
-
-
-—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------—--------
-
-
-
-
-
-def main(file, bot_name, num_clusters_to_display):
-    try:
-        global categories_keywords
-        if bot_name == "Teach For India":
-            categories_keywords = categories_keywords_tfi
-        else:
-            categories_keywords = categories_keywords_firki
-            
-        df = pd.read_csv(file.name)
-        
-        df = df[df['Answer'] == 'Fallback Message shown']
-        
-        df = preprocess_data(df, categories_keywords)
-        
-        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()
-        
-        largest_categories = sorted_categories[:num_clusters_to_display]
-        
-        filtered_df = df[df['Category'].isin(largest_categories)]
-        
-        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)
-
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmpfile:
-            filtered_df.to_csv(tmpfile.name, index=False)
-            csv_file_path = tmpfile.name
-
-        return csv_file_path, wordcloud_img, bar_chart_img
-    except Exception as e:
-        print(f"Error: {e}")
-        return str(e), None, None
-
-def categorize_unanswered_queries(bot_name, file, num_clusters_to_display):
-    return main(file, bot_name, num_clusters_to_display)
-
-interface = gr.Interface(
-    fn=categorize_unanswered_queries,
-    inputs=[
-        gr.Radio(["Teach For India", "Firki"], label="Select ChatBot"),
-        gr.File(label="Upload CSV File (.csv)"),
-        gr.Slider(label="Number of Categories to Display", minimum=1, maximum=10, step=1, value=5)
-    ],
-    outputs=[
-        gr.File(label="Categorized Data CSV"),
-        gr.Image(label="Word Cloud"),
-        gr.Image(label="Bar Chart")
-    ],
-    title="Unanswered User Queries Categorization",
-    description="Select the bot, upload the CSV file, and specify the number of categories to display to categorize unanswered user queries."
-)
-
-interface.launch()
-
-