ai_chatbot 0.1.6.1 → 0.1.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 63e218e69731be19e33a2b260fa04382dd1f3da77d9a8e20f0ee1bd8eac4fc97
-  data.tar.gz: bb380d89a7d15f29b426b33df682421412f62a41e9f1891858121dbb9c8b2de2
+  metadata.gz: 54c4224d297c82a20b3864dec73a0f1b5e654dfe5e08fbdfad12e09202588bde
+  data.tar.gz: 5ffb9610a7c5c13dbb2d597055c9269ca1578ceb5d16ecb940323d0d6909ed3a
 SHA512:
-  metadata.gz: 55d0946bd34c6009ac0029a6f4c183d6059b75eef2552b9eaa6c3905277209a13bf1e6e43b5b27dbabb1e5cf5a972f13ad1ae73902465bbc0256f6b09ebcf8ab
-  data.tar.gz: 397a9e37b9c225a25eb4765e5ad5e2377e7c64566cb623f131fc1bfe8d3fd298b8f0a3214fe9661d56d2a7fd2db94fc6a26cd357bc9036ad4d22407d9dd291e7
+  metadata.gz: cc4dedd9eed6dd5c5e3543969fab14e29014e9363369e33d5c932e2faf445b916343c87ad1412bd3da338895b739f13e2a2344f6ca9e8f6d2a0cffa548dbc04f
+  data.tar.gz: '093e12aa5db131a81ea46bc1d8eafee4794a99d13775da6d8cb84fefa2238bbc2b5edc1885e647c10b0f2c9d95cd9d9745723af7efd5468331cf5a028d0b2b90'

data/lib/ai_chatbot/version.rb

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

data/lib/ai_chatbot.rb

@@ -15,7 +15,7 @@ module AiChatbot
 
     # Method to train the model with a new question-answer pair
     def self.train_model(new_question, new_answer)
-      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "train", new_question, new_answer)
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "train_model", new_question, new_answer)
 
       if status.success?
         return stdout.strip
@@ -36,8 +36,8 @@ module AiChatbot
       end
     end
 
-    def self.update_or_delete_question(existing_question, new_question)
-      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "update_or_delete_question", existing_question, new_answer)
+    def self.update_or_delete_question(existing_question, new_question=nil)
+         stdout, stderr, status = new_question == nil ? Open3.capture3("python3", "#{__dir__}/ml_model.py", "update_or_delete_question", existing_question,"None") : Open3.capture3("python3", "#{__dir__}/ml_model.py", "update_or_delete_question", existing_question, new_question)
 
       if status.success?
         return stdout.strip
@@ -56,6 +56,16 @@ module AiChatbot
       end
     end
 
+    def self.list_answers
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "list_answers")
+
+      if status.success?
+        return stdout.strip
+      else
+        raise "Error: #{stderr}"
+      end
+    end
+
 
   end
 end

data/lib/ml_model.py

@@ -32,7 +32,7 @@ model.fit(questions, answers)
 def main(action, query=None, new_answer=None):
     if action == "predict":
         return get_prediction(query)
-    elif action == "train":
+    elif action == "train_model":
         return train_model(query, new_answer)
     elif action == "update_answer":
         return update_answer(query, new_answer)
@@ -40,6 +40,9 @@ def main(action, query=None, new_answer=None):
         return update_or_delete_question(query, new_answer)  # Corrected here, calling the right function
     elif action == "list_questions":
         return list_questions()
+    elif action == "list_answers":
+        return list_answers()    
+            
 
 # Function to predict the response with confidence check
 def get_prediction(query):
@@ -91,8 +94,10 @@ def update_answer(existing_question, new_answer):
     else:
         return "Question not found. Please provide a valid question."    
 
-def update_or_delete_question(existing_question, new_question=None):
+def update_or_delete_question(existing_question, new_question):
     global questions  # Only 'questions' is global, not 'new_question'
+    if new_question=="None":
+        new_question = None
     
     if existing_question in questions:
         if new_question:
@@ -125,6 +130,10 @@ def list_questions():
     global questions
     return questions
 
+def list_answers():
+    global answers
+    return answers    
+
 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.6.1
+  version: 0.1.6.2
 platform: ruby
 authors:
 - Sanket