ai_chatbot 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 12bcba7047f40d05685eec1c6a6714d7b8b6d57b092df1dcf6e382040af48182
-  data.tar.gz: ede7b3ccff3faecf008c681c649f7ac3732bebbd2a46f7801a313a0d64507ce4
+  metadata.gz: 726472f08ac5fcbed289fcd127215efb58823399b4c1e960ef4a9ae35a4ce368
+  data.tar.gz: 59cc5bbe59655f7863970fb32ee66279d308208af82b5ca7d3462748146972de
 SHA512:
-  metadata.gz: 991397292bed26854442cc630a87539fbb3ccba1cf81f6af673db57b607e56fdf150d4d905b00709b2c81ea6b4510759d7785f6ba16e44d58eab6e371b4ee581
-  data.tar.gz: f5312f608a08cffac6ef4ad93b4e7aed47191fdf0af0b7337a93bc13a9605e09db4eec91176b7dcc16c155af310d38d359fb03314bdd918b291f008579b56926
+  metadata.gz: 4611e846bcb29c1a0f5466a7b9269640ab8bb74ad8bb221e885f0c5fe07d2ec92e9a98aedec1b855fcc8ff6590a885c8f3522e4d8b5863724e7495ebc2a963fe
+  data.tar.gz: bb5dfc83841117e9f001d7691fa0fc4a1d439bf53cd7dc535b34a2f5a1f3b24c999d5a34d8cc407af7f4a036128fd280cf8998804c346a07364251bf885176ba

data/lib/ai_chatbot/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module AiChatbot
-  VERSION = "0.1.4"
+  VERSION = "0.1.6"
 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_question(query, new_answer)  # Corrected here, calling the right function
+    elif action == "list_questions":
         return list_questions()
 
 # Function to predict the response with confidence check
@@ -90,21 +92,21 @@ 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  # Only 'questions' is global, not '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
+        # Update the question
+        questions[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}'"
+        return f"Question updated from '{existing_question}' to '{new_question}'"
     else:
-        return "Question not found. Please provide a valid question."            
+        return "Question not found. Please provide a valid question."
 
 def list_questions():
     global questions

metadata

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