Update rag.py (#5)

- Update rag.py (c037e4581737c0338b3e981978ddde5af6ae5574)


Co-authored-by: Prerna Aneja <[email protected]>

rag.py

@@ -39,13 +39,17 @@ class FinancialChatbot:
             "attack", "genocide", "mass shooting", "credit card number"
         ]
 
+        self.NON_RELEVANT_WORDS = [
+            "France", "country", "rainbow", "water"
+        ]
+
         # Relevance threshold
         self.min_similarity_threshold = 0.7
 
     def moderate_query(self, query):
         """Check if the query contains inappropriate words."""
         query_lower = query.lower()
-        for word in self.BLOCKED_WORDS:
+        for word in self.BLOCKED_WORDS :
             if word in query_lower:
                 return False  # Block query
         return True  # Allow query
@@ -98,6 +102,12 @@ class FinancialChatbot:
                 result[:] = ["Hi, how can I help you?", 1.0]
                 return
 
+            query_lower = query.lower()
+            for word in self.NON_RELEVANT_WORDS :
+                result[:] = ["No relevant information found.", 0.0]
+                return
+
+
             if not self.moderate_query(query):
                 result[:] = ["I'm unable to process your request due to inappropriate language.", 0.0]
                 return