ai_chatbot 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fc3178f23280f489a71840d4b605097868f6c0aeb35b7790701232441a06e000
-  data.tar.gz: 0f64ab1e191a453914e2b42e4bcee8dc350e93672b393c8e475e72c66ba9e061
+  metadata.gz: 06bf16d383baea57ccc0bdbdcd1e041683130f0ef85120ca5af04b24cf64bbf9
+  data.tar.gz: d98da511ebcfbc645ffd5182db2e3c31703786714f98fb0374ec203386644439
 SHA512:
-  metadata.gz: 63f32775fefb298627912c672ad243a64a8b3dd70a4e2995f87ce980d73071c11823ecede02b6a31622f70a6d5dcdb131bc1615afa68387817af778821cbd9f0
-  data.tar.gz: 21b5d01971f6d897c18c85acc55417cfdca2bfbb56f857f0427ec0e47d6c236272a604b3f7c8fe759fb9001bc7b490e0e47ac5b813276d3412a150d3e583883c
+  metadata.gz: c4e7bf28f9d9b7568aa3dbe736b546018315f1d4dabd5b2d3ceff26dc3e12fc396f8ae35f32236900b8c39f8a1eadf889c93e8ba6bb37cebc14ef7402bcc0606
+  data.tar.gz: 8fdba9ecec76779184a18098395ba4a6d3b173c4e788ed40d0c3f2c2300a4ad2fb4a05a41928e82995fe7ad39d95576b996aa5e78d17dc54826a48243d2bab64

data/lib/ai_chatbot/version.rb

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

data/lib/ai_chatbot.rb

@@ -23,5 +23,39 @@ module AiChatbot
         raise "Error: #{stderr}"
       end
     end
+
+
+       # Method to train the model with a new question-answer pair
+    def self.update_answer(existing_question, new_answer)
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "update_answer", new_question, new_answer)
+
+      if status.success?
+        return stdout.strip
+      else
+        raise "Error: #{stderr}"
+      end
+    end
+
+    def self.update_question(existing_question, new_question)
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "update_question", new_question, new_answer)
+
+      if status.success?
+        return stdout.strip
+      else
+        raise "Error: #{stderr}"
+      end
+    end
+
+    def self.list_of_questions
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "list_questions")
+
+      if status.success?
+        return stdout.strip
+      else
+        raise "Error: #{stderr}"
+      end
+    end
+
+
   end
 end

data/lib/ml_model.py

@@ -34,9 +34,11 @@ def main(action, query=None, new_answer=None):
         return get_prediction(query)
     elif action == "train":
         return train_model(query, new_answer)
-    elif action == "update":
+    elif action == "update_answer":
         return update_answer(query, new_answer)
-    elif action == "list":
+    elif action == "update_question":
+        return update_answer(query, new_question)
+    elif action == "list_questions":
         return list_questions()
 
 # Function to predict the response with confidence check
@@ -48,7 +50,7 @@ def get_prediction(query):
     similarities = cosine_similarity(query_vec, question_vecs)
     max_similarity = similarities.max()
 
-    threshold = 0.70
+    threshold = 0.65
     if max_similarity < threshold:
         return "No good match found. Please provide the correct answer."
     else:
@@ -89,6 +91,23 @@ def update_answer(existing_question, new_answer):
     else:
         return "Question not found. Please provide a valid question."    
 
+def update_question(existing_question, new_question):
+    global questions, new_question
+
+    if existing_question in questions:
+        # Find the index of the existing question
+        index = questions.index(existing_question)
+        # Update the answer
+        existing_question[index] = new_question
+        # 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"New question updated for the question: '{existing_question}'"
+    else:
+        return "Question not found. Please provide a valid question."            
+
 def list_questions():
     global questions
     return questions

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai_chatbot
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Sanket
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-09-19 00:00:00.000000000 Z
+date: 2024-09-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: open3
@@ -38,7 +38,8 @@ files:
 homepage: https://github.com/tikhandesanket/ai-chatbot
 licenses:
 - MIT
-metadata: {}
+metadata:
+  source_code_uri: https://github.com/tikhandesanket/ai-chatbot
 post_install_message: 
 rdoc_options: []
 require_paths: