Update app.py

app.py

@@ -166,8 +166,14 @@ def generate_bar_chart(df, num_clusters_to_display):
     img = Image.open(buf)
     return img
 
-def main(file, num_clusters_to_display, categories_keywords):
+def main(file, bot_name, num_clusters_to_display):
     try:
+        global categories_keywords
+        if bot_name == "Bot 1":
+            categories_keywords = categories_keywords_bot1
+        else:
+            categories_keywords = categories_keywords_bot2
+            
         df = pd.read_csv(file)
         
         df = df[df['Answer'] == 'Fallback Message shown']
@@ -196,32 +202,13 @@ def main(file, num_clusters_to_display, categories_keywords):
         print(f"Error: {e}")
         return str(e), None, None
 
-def select_bot(bot_name):
-    global categories_keywords
-    if bot_name == "Bot 1":
-        categories_keywords = categories_keywords_bot1
-    else:
-        categories_keywords = categories_keywords_bot2
-    return f"Selected {bot_name}"
+def categorize_unanswered_queries(file, bot_name, num_clusters_to_display):
+    return main(file, bot_name, num_clusters_to_display)
 
-def categorize_unanswered_queries(file, num_clusters_to_display):
-    return main(file, num_clusters_to_display, categories_keywords)
-
-bot_selection_interface = gr.Interface(
-    fn=select_bot,
-    inputs=[
-        gr.Radio(["Bot 1", "Bot 2"], label="Select Bot")
-    ],
-    outputs=[
-        gr.Textbox(label="Selected Bot")
-    ],
-    title="Select Bot",
-    description="Select the bot for categorizing unanswered queries."
-)
-
-categorize_interface = gr.Interface(
+interface = gr.Interface(
     fn=categorize_unanswered_queries,
     inputs=[
+        gr.Radio(["Bot 1", "Bot 2"], label="Select Bot"),
         gr.File(label="Upload CSV File (.csv)"),
         gr.Slider(label="Number of Categories to Display", minimum=1, maximum=10, step=1, value=5)
     ],
@@ -231,9 +218,7 @@ categorize_interface = gr.Interface(
         gr.Image(label="Bar Chart")
     ],
     title="Unanswered User Queries Categorization",
-    description="Categorize unanswered user queries into predefined categories"
+    description="Select the bot, upload the CSV file, and specify the number of categories to display to categorize unanswered user queries."
 )
 
-main_interface = gr.TabbedInterface([bot_selection_interface, categorize_interface], ["Select Bot", "Categorize Queries"])
-
-main_interface.launch(share=True)
+interface.launch(share=True)