Update app.py

app.py

@@ -45,21 +45,25 @@ def categorize_question(question):
     
     # List of words to exclude from 'End of Conversation'
     exclusion_words = {'is', 'please', 'not', 'resolved', 'problem', 'help', 'issue', 'webinar', 'office', 'leave', 'approved', 'notice', 'period'}
-    
+
     # Check if the question has only one word
     if len(words) == 1:
         single_word = words[0].lower()
-        
-        if any(single_word in keyword for keyword in categories_keywords["Start of Conversation"]):
-            return "Start of Conversation"
-        else:
+        # Check if the single word fits into any other category
+        for category, keywords in categories_keywords.items():
+            if any(single_word in keyword for keyword in keywords):
+                return category
+        # If it doesn't fit into any other category, check if it should be 'End of Conversation'
+        if any(single_word in keyword for keyword in categories_keywords["End of Conversation"]):
             return "End of Conversation"
-    
+        else:
+            return "Miscellaneous"
+
     # Categorization of other queries
     for category, keywords in categories_keywords.items():
         if any(keyword.lower() in question.lower() for keyword in keywords):
             return category
-    
+
     # Secondary check for 'End of Conversation' category
     if "end of conversation" in question.lower() and not any(exclusion_word in question.lower() for exclusion_word in exclusion_words):
         return "End of Conversation"
@@ -67,7 +71,6 @@ def categorize_question(question):
     return "Miscellaneous"
 
 
-
 def preprocess_data(df):
     df.rename(columns={'Question Asked': 'texts'}, inplace=True)
     df['texts'] = df['texts'].astype(str).str.lower()