ai_chatbot 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: bde3d724e066978fd481402f7aaa899041d85044b8b3ecc10574410ee79946f2
+  data.tar.gz: 8cc52ee8ec6b7e061fa6c009fdac5886b6b8e1a6e785d50abf70146bf04135ab
+SHA512:
+  metadata.gz: 115664a820b9b5c3da45d3fbb382785bd24dc0a840ad707fd8d1f0e7612d498c377ed12e86e7f997191cc83585a4245cfedda79cc22ee8ecfd44e01c131bb10b
+  data.tar.gz: f94763913a5cf902b9f940855c9f5a608a47495f776ddd3ab59e223ebad03be2bb0320c9d57317db2435a2f0662e08aca18f3143e56265a3fb7d7e86e018a6ca

data/lib/ai_chatbot/version.rb

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

data/lib/ai_chatbot.rb

@@ -0,0 +1,27 @@
+require 'open3'
+
+module AiChatbot
+  class Chatbot
+    # Method to ask a question and get prediction from Python
+    def self.ask_question(question)
+      stdout, stderr, status = Open3.capture3("python3", "#{__dir__}/ml_model.py", "predict", question)
+
+      if status.success?
+        return stdout.strip
+      else
+        raise "Error: #{stderr}"
+      end
+    end
+
+    # 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)
+
+      if status.success?
+        return stdout.strip
+      else
+        raise "Error: #{stderr}"
+      end
+    end
+  end
+end

data/lib/ml_model.py

@@ -0,0 +1,76 @@
+import sys
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.naive_bayes import MultinomialNB
+from sklearn.pipeline import make_pipeline
+from sklearn.metrics.pairwise import cosine_similarity
+import pickle
+import os
+
+# Load or initialize the dataset
+if os.path.exists("qa_model.pkl"):
+    with open("qa_model.pkl", "rb") as f:
+        model_data = pickle.load(f)
+        questions = model_data['questions']
+        answers = model_data['answers']
+else:
+    questions = [
+        "How to create a new model in Rails?",
+        "What is migration?",
+        "How to add a route?"
+    ]
+    answers = [
+        "You can create a model using 'rails generate model'.",
+        "Migration is a database schema change.",
+        "You can add a route in the config/routes.rb file."
+    ]
+
+# Create a pipeline (TF-IDF + MultinomialNB)
+model = make_pipeline(TfidfVectorizer(), MultinomialNB())
+model.fit(questions, answers)
+
+# Function to predict or retrain the model
+def main(action, query=None, new_answer=None):
+    if action == "predict":
+        return get_prediction(query)
+    elif action == "train":
+        return train_model(query, new_answer)
+
+# Function to predict the response with confidence check
+def get_prediction(query):
+    query_vec = model.named_steps['tfidfvectorizer'].transform([query])
+    question_vecs = model.named_steps['tfidfvectorizer'].transform(questions)
+
+    # Calculate cosine similarity between query and known questions
+    similarities = cosine_similarity(query_vec, question_vecs)
+    max_similarity = similarities.max()
+
+    threshold = 0.85
+    if max_similarity < threshold:
+        return "No good match found. Please provide the correct answer."
+    else:
+        prediction = model.predict([query])
+        return prediction[0]
+
+# Function to train the model with a new question and answer
+def train_model(new_question, new_answer):
+    global questions, answers
+
+    # Append new question-answer pair to the dataset
+    questions.append(new_question)
+    answers.append(new_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"Model retrained with the new question: '{new_question}' and answer: '{new_answer}'"
+
+if __name__ == "__main__":
+    # Expecting action (predict/train), question, and answer (if training)
+    action = sys.argv[1]
+    question = sys.argv[2] if len(sys.argv) > 2 else None
+    answer = sys.argv[3] if len(sys.argv) > 3 else None
+    print(main(action, question, answer))

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: ai_chatbot
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Sanet
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2024-09-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: open3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Integrates a chatbot using Python for predictions and training in a Rails
+  application.
+email:
+- sanket.tikhande@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/ai_chatbot.rb
+- lib/ai_chatbot/version.rb
+- lib/ml_model.py
+homepage: 
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.7
+signing_key: 
+specification_version: 4
+summary: A chatbot for Rails integration with AI model using Python
+test_files: []