ai_chatbot 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c1ecd8a87638329fd5c23080f6221814629a99b734c0952023d03114c1da8d9f
-  data.tar.gz: ddeb90ba2ae48a4b88adacc58ceb6cd3898498ae08ba7c0323c4e676a64343dc
+  metadata.gz: fc3178f23280f489a71840d4b605097868f6c0aeb35b7790701232441a06e000
+  data.tar.gz: 0f64ab1e191a453914e2b42e4bcee8dc350e93672b393c8e475e72c66ba9e061
 SHA512:
-  metadata.gz: '0176796cd767ea1f9f069e61706d61d6aa9a767c67d284b3651a0b3df08c924a915fdfd72cb24c6231c4dcce2fc36acbf352e3c0fd0377d83427416e820ecf97'
-  data.tar.gz: 4dd0ef6b1cc5eed4910d92a410a1104f77c10946e0dd0988e8d572a6672ed0bda9d4c6bc1219ea27903ee14668cdbbca5bbc1143012f247a88471c3a5ccd6e0e
+  metadata.gz: 63f32775fefb298627912c672ad243a64a8b3dd70a4e2995f87ce980d73071c11823ecede02b6a31622f70a6d5dcdb131bc1615afa68387817af778821cbd9f0
+  data.tar.gz: 21b5d01971f6d897c18c85acc55417cfdca2bfbb56f857f0427ec0e47d6c236272a604b3f7c8fe759fb9001bc7b490e0e47ac5b813276d3412a150d3e583883c

data/lib/ai_chatbot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiChatbot
-  VERSION = "0.1.1"
+  VERSION = "0.1.3"
 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,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai_chatbot
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
-- Sanet
+- Sanket
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-09-17 00:00:00.000000000 Z
+date: 2024-09-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: open3
@@ -35,7 +35,7 @@ files:
 - lib/ai_chatbot.rb
 - lib/ai_chatbot/version.rb
 - lib/ml_model.py
-homepage: 
+homepage: https://github.com/tikhandesanket/ai-chatbot
 licenses:
 - MIT
 metadata: {}