Update nylon.py

nylon.py

@@ -107,17 +107,22 @@ class ChatDatabase:
                 self.messages.append(message)
         
         self.messages_array = np.array(self.messages, dtype=object)
-        self.sender_array = self.messages_array[:, 0]
+        print(self.messages_array,'hihii')
+        if len(self.messages_array)==0:
+            self.sender_array = []
+        else:
+            self.sender_array = self.messages_array[:, 0]
         print(f'Database loaded. Number of messages: {len(self.messages_array)}')
 
     def add_message(self, sender, time, text, tag=None):
-        message = (sender, time, text, tag)
+        message = np.array((sender, time, text, tag)).flatten()
         self.messages.append(message)
-        self.messages_array = np.append(self.messages_array, [message], axis=0)
+        self.messages_array = np.append(self.messages_array, message, axis=0)
         self.sender_array = np.append(self.sender_array, sender)
         with open(self.filename, 'a') as f:
             f.write(f'{sender}\t{time}\t{text}\t{tag}\n')
 
+
     def predict_response_separate(self, query, sender, cache):
         if self.messages_array is None:
             print("Error: messages_array is None")
@@ -150,17 +155,16 @@ class ChatDatabase:
 
         query_keywords = query.lower().split()
 
+        #Filter by sender
+        sender_messages = self.messages_array[self.sender_array == sender]
+        print(f"Number of messages from sender {sender}: {len(sender_messages)}")
+
         # Filter messages by sender, tag, and keywords in a single line
-        if query_tag:
-            sender_messages = self.messages_array[
-                (self.sender_array == sender) & 
-                np.array([any(keyword in message.lower() for keyword in query_keywords) for message in self.messages_array[:, 2]])
-            ]
-        else:
-            sender_messages = self.messages_array[
-                (self.sender_array == sender) & 
-                np.array([any(keyword in message.lower() for keyword in query_keywords) for message in self.messages_array[:, 2]])
-            ]
+        sender_messages = self.messages_array[
+            (self.sender_array == sender) & 
+            np.array([any(keyword in message.lower() for keyword in query_keywords) for message in self.messages_array[:, 2]])
+        ]
+        
 
         if len(sender_messages) == 0:
             print(f"No messages found for sender: {sender} with the given keywords")
@@ -213,14 +217,14 @@ class ChatDatabase:
 # Usage example remains the same
 '''
 # Usage example
-db = ChatDatabase('messages.txt')
+db = ChatDatabase('memory.txt')
 
 # Example 1: Get relevant messages
-query = 'Goodnight.'
-sender = 'Alice'
+query = 'fisical policy'
+sender = 'Arcana'
 N = 10
 cache = {}
-query_tag = 'evening'
+query_tag = None
 
 relevant_messages = db.get_relevant_messages(sender, query, N, cache, query_tag)