Update chat.py

chat.py

@@ -1,26 +1,43 @@
-import re
+import json
 import os
+import logging
+import re
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 class KiswahiliChatbot:
-    def __init__(self, dataset_path="trained_bert_model/salamu.txt"):
-        if not os.path.exists(dataset_path):
-            raise FileNotFoundError(f"{dataset_path} not found. Please add your dataset.")
+    def __init__(self, model_path="./trained_bert_model"):
+        # Load responses
+        response_file = os.path.join(model_path, "responses.json")
+        if not os.path.exists(response_file):
+            raise FileNotFoundError(f"{response_file} not found. Please create responses.json")
+        
+        with open(response_file, "r", encoding="utf-8") as f:
+            data = json.load(f)
+            self.responses = data.get("responses", [])
         
-        # Load greetings/responses
-        with open(dataset_path, "r", encoding="utf-8") as f:
-            self.responses = [line.strip() for line in f if line.strip()]
-        print(f"✅ Loaded {len(self.responses)} greetings")
+        logger.info(f"✅ Loaded {len(self.responses)} canned responses")
+
+    def _clean_text(self, text: str) -> str:
+        text = re.sub(r'[^\w\s?]', '', text)
+        return ' '.join(text.split()).lower()
+
+    def chat(self, user_input: str) -> str:
+        user_input_clean = self._clean_text(user_input)
 
-    def chat(self, user_input):
-        # Clean input
-        user_input_clean = re.sub(r'[^\w\s?]', '', user_input).lower()
         if not user_input_clean:
             return "Tafadhali andika ujumbe."
 
-        # Simple matching: return first response that contains a keyword
-        for response in self.responses:
-            if user_input_clean in response.lower():
-                return response
-
-        # Fallback response
-        return "Samahani, sielewi. Unaweza kuuliza kwa njia nyingine?"
+        # Very simple matching
+        if "habari" in user_input_clean:
+            return "Nzuri sana! Habari yako pia?"
+        elif "jina" in user_input_clean:
+            return "Mimi ni KiswahiliChetu."
+        elif "tanzania" in user_input_clean:
+            return "Tanzania ni nchi ya Afrika Mashariki yenye miji mikuu Dodoma na Dar es Salaam."
+        elif "hakuna matata" in user_input_clean:
+            return "Ni msemo wa Kiswahili unaomaanisha 'Hakuna shida'."
+        else:
+            # fallback
+            return self.responses[0] if self.responses else "Samahani, sielewi. Unaweza kuuliza kwa njia nyingine?"