Update app.py

app.py

@@ -10,8 +10,9 @@ import matplotlib.pyplot as plt
 import plotly.express as px
 from PIL import Image
 
-# Define categories for Bot 1 and Bot 2
-categories_keywords_bot1 = {
+# Define categories for Teach For India Bot and Firki
+categories_keywords_tfi = {
+    'Start of Conversation': ['hi', 'hello', 'hi I have a query', 'query', 'good morning', 'good afternoon', 'good evening'],
     'Application Status': ['application', 'applied', 'update on my application', 'result of my application', 'selected', 'selection process', 'apply', 'fellow', 'lesson plan', 'status of my application', 'application update', 'application status', 'applied for'],
     'Volunteering': ['volunteering', 'volunteer', 'volunteering certificate', 'resume my volunteering', 'volunteering journey', 'volunteering with TFI', 'volunteering opportunities', 'volunteer work', 'volunteer program'],
     'Certificates': ['certificate', 'certificates', 'certificate of completion', 'volunteer certificate', 'issue certificate'],
@@ -26,11 +27,12 @@ categories_keywords_bot1 = {
     'Salary and Benefits': ['salary', 'increment', 'accommodation', 'training period', 'reside', 'stipend', 'pay', 'wage', 'salary details', 'benefits information'],
     'Technical Issues': ['network issues', 'zoom meeting', 'passcode', 'technical', 'issue','technical problem', 'system issue', 'technical support'],
     'Complaint Handling': ['help', 'i need help', 'Help me', 'complaint', 'issue is unresolved', 'unsatisfied', 'bad experience'],
+    'User Feedback': ['feedback', 'loved', 'dissapointed', 'hated', 'it was good', 'it was bad', 'helpful',],
     'End of Conversation': ['thanks', 'thankss', 'thank u', 'thank you', 'ok', 'okay', 'done', 'joining', 'sounds good', 'goodbye', 'end chat', 'end'],
     'Miscellaneous': []
 }
 
-categories_keywords_bot2 = {
+categories_keywords_firki = {
     'Service Issues': ['service', 'support', 'help', 'assistance'],
     'Billing': ['bill', 'billing', 'invoice', 'payment', 'charge'],
     'Technical Problems': ['technical', 'error', 'problem', 'issue'],
@@ -40,7 +42,7 @@ categories_keywords_bot2 = {
 }
 
 # Initialize with Bot 1's categories
-categories_keywords = categories_keywords_bot1
+categories_keywords = categories_keywords_tfi
 
 def categorize_question(question, categories_keywords):
     for category, keywords in categories_keywords.items():
@@ -169,10 +171,10 @@ def generate_bar_chart(df, num_clusters_to_display):
 def main(file, bot_name, num_clusters_to_display):
     try:
         global categories_keywords
-        if bot_name == "Bot 1":
-            categories_keywords = categories_keywords_bot1
+        if bot_name == "Teach For India":
+            categories_keywords = categories_keywords_tfi
         else:
-            categories_keywords = categories_keywords_bot2
+            categories_keywords = categories_keywords_firki
             
         df = pd.read_csv(file.name)
         
@@ -208,7 +210,7 @@ def categorize_unanswered_queries(bot_name, file, num_clusters_to_display):
 interface = gr.Interface(
     fn=categorize_unanswered_queries,
     inputs=[
-        gr.Radio(["Bot 1", "Bot 2"], label="Select Bot"),
+        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)
     ],