Update app.py

app.py

@@ -42,16 +42,26 @@ categories_keywords = {
 
 def categorize_question(question):
     words = question.split()
-    if len(words) == 1:
+
+    exclusion_words = {'is', 'please', 'not', 'resolved', 'problem', 'help', 'issue', 'webinar', 'office', 'leave', 'approved', 'notice', 'period'}
+    
+     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:
             return "End of Conversation"
     
+    # 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"
+    
     return "Miscellaneous"