ai_chatbot 0.1.6 → 0.1.6.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 726472f08ac5fcbed289fcd127215efb58823399b4c1e960ef4a9ae35a4ce368
-  data.tar.gz: 59cc5bbe59655f7863970fb32ee66279d308208af82b5ca7d3462748146972de
+  metadata.gz: 63e218e69731be19e33a2b260fa04382dd1f3da77d9a8e20f0ee1bd8eac4fc97
+  data.tar.gz: bb380d89a7d15f29b426b33df682421412f62a41e9f1891858121dbb9c8b2de2
 SHA512:
-  metadata.gz: 4611e846bcb29c1a0f5466a7b9269640ab8bb74ad8bb221e885f0c5fe07d2ec92e9a98aedec1b855fcc8ff6590a885c8f3522e4d8b5863724e7495ebc2a963fe
-  data.tar.gz: bb5dfc83841117e9f001d7691fa0fc4a1d439bf53cd7dc535b34a2f5a1f3b24c999d5a34d8cc407af7f4a036128fd280cf8998804c346a07364251bf885176ba
+  metadata.gz: 55d0946bd34c6009ac0029a6f4c183d6059b75eef2552b9eaa6c3905277209a13bf1e6e43b5b27dbabb1e5cf5a972f13ad1ae73902465bbc0256f6b09ebcf8ab
+  data.tar.gz: 397a9e37b9c225a25eb4765e5ad5e2377e7c64566cb623f131fc1bfe8d3fd298b8f0a3214fe9661d56d2a7fd2db94fc6a26cd357bc9036ad4d22407d9dd291e7

data/lib/ai_chatbot/version.rb

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

data/lib/ai_chatbot.rb

@@ -27,7 +27,7 @@ module AiChatbot
 
        # 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)
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "update_answer", existing_question, new_answer)
 
       if status.success?
         return stdout.strip
@@ -36,8 +36,8 @@ module AiChatbot
       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)
+    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)
 
       if status.success?
         return stdout.strip
@@ -46,7 +46,7 @@ module AiChatbot
       end
     end
 
-    def self.list_of_questions
+    def self.list_questions
       stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "list_questions")
 
       if status.success?

data/lib/ml_model.py

@@ -36,8 +36,8 @@ def main(action, query=None, new_answer=None):
         return train_model(query, new_answer)
     elif action == "update_answer":
         return update_answer(query, new_answer)
-    elif action == "update_question":
-        return update_question(query, new_answer)  # Corrected here, calling the right function
+    elif action == "update_or_delete_question":
+        return update_or_delete_question(query, new_answer)  # Corrected here, calling the right function
     elif action == "list_questions":
         return list_questions()
 
@@ -91,23 +91,36 @@ def update_answer(existing_question, new_answer):
     else:
         return "Question not found. Please provide a valid question."    
 
-def update_question(existing_question, new_question):
+def update_or_delete_question(existing_question, new_question=None):
     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 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"Question updated from '{existing_question}' to '{new_question}'"
+        if new_question:
+            # Find the index of the existing question
+            index = questions.index(existing_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"Question updated from '{existing_question}' to '{new_question}'"
+        else:
+            # Remove the question if no new question is provided
+            index = questions.index(existing_question)
+            del questions[index]
+            del answers[index]  # Ensure you also delete the corresponding 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"Question '{existing_question}' deleted successfully."
     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.6
+  version: 0.1.6.1
 platform: ruby
 authors:
 - Sanket
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-09-20 00:00:00.000000000 Z
+date: 2024-09-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: open3