Update app.py

app.py

@@ -41,16 +41,18 @@ categories_keywords = {
 
 
 def categorize_question(question):
+    words = question.split()
+    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:
+            return "End of Conversation"
+    
     for category, keywords in categories_keywords.items():
-        for keyword in keywords:
-            if keyword.lower() in question.lower():
-                # Check if the question is one word and belongs to 'End of Conversation'
-                if category == 'End of Conversation':
-                    return category
-                # If not 'End of Conversation', return the matched category
-                if category != 'End of Conversation':
-                    return category
-    return 'Miscellaneous'
+        if any(keyword.lower() in question.lower() for keyword in keywords):
+            return category
+    return "Miscellaneous"