AiChatbot

AiChatbot is a lightweight Ruby gem that provides a simple AI-based chatbot using TF-IDF vectorization and Naive Bayes classification. It supports learning new Q&A pairs and predicting answers based on cosine similarity.

Add this line to your Gemfile:

gem 'ai_chatbot', '0.1.6.5.2'

And then execute:

bundle install

Or install it directly:

gem install ai_chatbot

require 'ai_chatbot'

chatbot = AiChatbot::Bot.new

# Predict an answer
puts chatbot.predict("How to create a model in Rails?")

# Add new training data
chatbot.train("How to upload a dish in hotel?", "Use the admin panel under 'Menu Upload' to add a dish.")

• ✅ TF-IDF Vectorizer for understanding query intent
• ✅ Cosine Similarity for matching confidence
• ✅ Naive Bayes Classifier for predictions
• ✅ Add / Update / Delete Q&A pairs
• ✅ CLI Support

ruby chatbot.rb predict "How to add a route?"
ruby chatbot.rb train_model "What is RubyGems?" "A package manager for Ruby libraries."

To build and install locally:

bundle install
gem build ai_chatbot.gemspec
gem install ./ai_chatbot-x.y.z.gem

To release:

rake release

This project is licensed under the MIT License.

Sanket Tikhande
GitHub: @tikhandesanket