ai_chatbot 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c1ecd8a87638329fd5c23080f6221814629a99b734c0952023d03114c1da8d9f
-  data.tar.gz: ddeb90ba2ae48a4b88adacc58ceb6cd3898498ae08ba7c0323c4e676a64343dc
+  metadata.gz: 44d4375b7617e626f0ece5d48d9951db6cbb29d40fc039ed432f57f95214ff61
+  data.tar.gz: b47d32642a89b0a8ee430d210d57fa310616828938e431cf5b9094b1720e9d92
 SHA512:
-  metadata.gz: '0176796cd767ea1f9f069e61706d61d6aa9a767c67d284b3651a0b3df08c924a915fdfd72cb24c6231c4dcce2fc36acbf352e3c0fd0377d83427416e820ecf97'
-  data.tar.gz: 4dd0ef6b1cc5eed4910d92a410a1104f77c10946e0dd0988e8d572a6672ed0bda9d4c6bc1219ea27903ee14668cdbbca5bbc1143012f247a88471c3a5ccd6e0e
+  metadata.gz: 2b85bdabf2fcf579a8355ef6b847251b6f1a715ef648a70bdbff1d4807f3615b425c50b80ce54f653477ae4c4caa163c8614e9563364a75145ff346b369077c3
+  data.tar.gz: 1fd7b59dde9c05e0ebca46fb472365786e7dbeb6440d2b30c7a0239246343efb8c3cb617aad4a83c48af2c3d18daab3726aaf6f672ea20b33bfbaa012995b94c

data/lib/ai_chatbot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiChatbot
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/ml_model.py

@@ -34,6 +34,10 @@ def main(action, query=None, new_answer=None):
         return get_prediction(query)
     elif action == "train":
         return train_model(query, new_answer)
+    elif action == "update":
+        return update_answer(query, new_answer)
+    elif action == "list":
+        return list_questions()
 
 # Function to predict the response with confidence check
 def get_prediction(query):
@@ -68,6 +72,27 @@ def train_model(new_question, new_answer):
 
     return f"Model retrained with the new question: '{new_question}' and answer: '{new_answer}'"
 
+def update_answer(existing_question, new_answer):
+    global questions, answers
+
+    if existing_question in questions:
+        # Find the index of the existing question
+        index = questions.index(existing_question)
+        # Update the answer
+        answers[index] = new_answer
+        # Retrain the model with updated data
+        model.fit(questions, answers)
+        # Save the updated model and data
+        with open("qa_model.pkl", "wb") as f:
+            pickle.dump({"questions": questions, "answers": answers}, f)
+        return f"Answer updated for the question: '{existing_question}'"
+    else:
+        return "Question not found. Please provide a valid question."    
+
+def list_questions():
+    global questions
+    return questions
+
 if __name__ == "__main__":
     # Expecting action (predict/train), question, and answer (if training)
     action = sys.argv[1]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_chatbot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Sanet