Update app.py

app.py

@@ -29,18 +29,21 @@ categories_keywords = {
     'Miscellaneous': []
 }
 
+
 def categorize_question(question):
-    
-    for keyword in categories_keywords['End of Conversation']:
-        if keyword.lower() in question.lower():
-            return 'End of Conversation'
     for category, keywords in categories_keywords.items():
         for keyword in keywords:
             if keyword.lower() in question.lower():
-                return category
+                # 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'
 
 
+
 def preprocess_data(df):
     df.rename(columns={'Question Asked': 'texts'}, inplace=True)
     df['texts'] = df['texts'].astype(str).str.lower()
@@ -207,4 +210,5 @@ interface = gr.Interface(
     description="Categorize unanswered user queries into predefined categories"
 )
 
-interface.launch(share=True)
+interface.launch(share=True)
+