ai_chatbot 0.1.6 → 0.1.6.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 726472f08ac5fcbed289fcd127215efb58823399b4c1e960ef4a9ae35a4ce368
-  data.tar.gz: 59cc5bbe59655f7863970fb32ee66279d308208af82b5ca7d3462748146972de
+  metadata.gz: 54c4224d297c82a20b3864dec73a0f1b5e654dfe5e08fbdfad12e09202588bde
+  data.tar.gz: 5ffb9610a7c5c13dbb2d597055c9269ca1578ceb5d16ecb940323d0d6909ed3a
 SHA512:
-  metadata.gz: 4611e846bcb29c1a0f5466a7b9269640ab8bb74ad8bb221e885f0c5fe07d2ec92e9a98aedec1b855fcc8ff6590a885c8f3522e4d8b5863724e7495ebc2a963fe
-  data.tar.gz: bb5dfc83841117e9f001d7691fa0fc4a1d439bf53cd7dc535b34a2f5a1f3b24c999d5a34d8cc407af7f4a036128fd280cf8998804c346a07364251bf885176ba
+  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"
+  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
@@ -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=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
@@ -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?
@@ -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,14 +32,17 @@ 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)
-    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()
+    elif action == "list_answers":
+        return list_answers()    
+            
 
 # Function to predict the response with confidence check
 def get_prediction(query):
@@ -91,27 +94,46 @@ 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):
     global questions  # Only 'questions' is global, not 'new_question'
-
+    if new_question=="None":
+        new_question = None
+    
     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
 
+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,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ai_chatbot
 version: !ruby/object:Gem::Version
-  version: 0.1.6
+  version: 0.1.6.2
 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