Update app.py

app.py

@@ -42,12 +42,14 @@ categories_keywords = {
 
 def categorize_question(question):
     words = question.split()
-
+    
+    # List of words to exclude from 'End of Conversation'
     exclusion_words = {'is', 'please', 'not', 'resolved', 'problem', 'help', 'issue', 'webinar', 'office', 'leave', 'approved', 'notice', 'period'}
     
-     if len(words) == 1:
+    # Check if the question has only one word
+    if len(words) == 1:
         single_word = words[0].lower()
-        # Check if the single word is in the Start of Conversation category
+        
         if any(single_word in keyword for keyword in categories_keywords["Start of Conversation"]):
             return "Start of Conversation"
         else:
@@ -59,7 +61,7 @@ def categorize_question(question):
             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):
+    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"
     
     return "Miscellaneous"