ai_chatbot 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12bcba7047f40d05685eec1c6a6714d7b8b6d57b092df1dcf6e382040af48182
-  data.tar.gz: ede7b3ccff3faecf008c681c649f7ac3732bebbd2a46f7801a313a0d64507ce4
+  metadata.gz: 06bf16d383baea57ccc0bdbdcd1e041683130f0ef85120ca5af04b24cf64bbf9
+  data.tar.gz: d98da511ebcfbc645ffd5182db2e3c31703786714f98fb0374ec203386644439
 SHA512:
-  metadata.gz: 991397292bed26854442cc630a87539fbb3ccba1cf81f6af673db57b607e56fdf150d4d905b00709b2c81ea6b4510759d7785f6ba16e44d58eab6e371b4ee581
-  data.tar.gz: f5312f608a08cffac6ef4ad93b4e7aed47191fdf0af0b7337a93bc13a9605e09db4eec91176b7dcc16c155af310d38d359fb03314bdd918b291f008579b56926
+  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.4"
+  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
@@ -90,7 +92,7 @@ def update_answer(existing_question, new_answer):
         return "Question not found. Please provide a valid question."    
 
 def update_question(existing_question, new_question):
-    global existing_question, new_question
+    global questions, new_question
 
     if existing_question in questions:
         # Find the index of the existing question

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ai_chatbot
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.5
 platform: ruby
 authors:
 - Sanket