appchat 0.0.4 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f741156bd18664bfb66a4700b633dad255bc2c5e41cacc6b297acbc6d67492b8
-  data.tar.gz: d4612a8a1ee88a6caf7bc4e039cf74fa44b369fe79b5d595b0e9544329749819
+  metadata.gz: 4aca406ace060a7c7b9af445e20d5be1f3605614109320792523cba09b7f92fe
+  data.tar.gz: 8c1e290b46dfca74c66c94748d305386d67703291e455c066aed1df0d1dc5cc8
 SHA512:
-  metadata.gz: '0959e29a0ed53649f7f51523e5b4c6f4089b05449f8649165ed7bf7505bdb111774e0c24661e5344f0752df1fa08a38b250a3fe911543530adb7f8a3020f2bd6'
-  data.tar.gz: '08f3d14d68ea6c2cf9c50da6842051b78bafa250bf58fc77475e09c32090344d06fe4d98205294aa08e9331bf1817e8d5b23213da001b136dd1fb2101c98aca8'
+  metadata.gz: 5aaa00c36101e08cb562924a7a3a0cca7da5ef49680eb6f9f202a8d17ebe0cbbe852fe86ef4481cb65ef8c2811d9933731074363875930ccb916e92d9fdd66d4
+  data.tar.gz: 876249a02dbec15d9d3ee4c9dbabcfbcc843970f8d85d7b4e9fad563a97af8a86c1a3600c7383ef3dea9fc2bfffef4c19f3fef70cb7009f0fc0a12401efbb462

data/lib/generators/appchat/appchat_generator.rb

@@ -9,28 +9,45 @@ class AppchatGenerator < Rails::Generators::Base
   source_root File.expand_path('templates', __dir__)
 
   def add_gems
-    unless gem_exists?('ollama-ai')
-      append_to_file 'Gemfile', "\ngem 'ollama-ai'\n"
-    end
-
-    unless gem_exists?('tailwindcss-rails')
-      append_to_file 'Gemfile', "\ngem 'tailwindcss-rails'\n"
+    gems = %w(
+      ollama-ai
+      tailwindcss-rails
+      watir
+    )
+
+    gems.each do |gem|
+      unless gem_exists?(gem)
+        append_to_file 'Gemfile', "\ngem '#{gem}'\n"
+      end
     end
 
     Bundler.with_unbundled_env do
       run 'bundle install'
     end
+  end
+
+  def install_tailwind
     run 'rails tailwindcss:install'
   end
 
-  def generate_scaffolds
-    generate "model", "Chat context:text"
+  def set_routes
     route "resources :chats"
-    generate "scaffold", "Message chat:references content:text role:integer"
+    route "resources :messages"
   end
 
-  def create_views
+  def generate_models
+    generate "model", "Chat context:text"
+    generate "model", "Message chat:references content:text role:integer status:string"
+    generate "model", "AppchatFunction name:string description:text class_name:string"
+    generate "model", "FunctionParameter appchat_function:references name:string example_value:string"
+  end
+
+  def create_controllers
     copy_file "chats_controller.rb", "app/controllers/chats_controller.rb"
+    copy_file "messages_controller.rb", "app/controllers/messages_controller.rb", force: true
+  end
+
+  def create_views
     copy_file "chats/chat.html.erb", "app/views/chats/_chat.html.erb", force: true
     copy_file "chats/index.html.erb", "app/views/chats/index.html.erb", force: true
     copy_file "chats/show.html.erb", "app/views/chats/show.html.erb", force: true
@@ -38,21 +55,30 @@ class AppchatGenerator < Rails::Generators::Base
     copy_file "messages/new.html.erb", "app/views/messages/new.html.erb", force: true
     copy_file "messages/message.html.erb", "app/views/messages/_message.html.erb", force: true
     copy_file "messages/_typing_bubbles.html.erb", "app/views/messages/_typing_bubbles.html.erb", force: true
+  end
+
+  def create_stylesheets
     copy_file "assets/appchat.tailwind.css", "app/assets/stylesheets/appchat.tailwind.css", force: true
+  end
+
+  def create_stimulus_controllers
     copy_file "javascript/chat_message_controller.js", "app/javascript/controllers/chat_message_controller.js"
+    copy_file "javascript/speech_to_text_controller.js", "app/javascript/controllers/speech_to_text_controller.js"
   end
 
   def copy_models
     copy_file "models/message.rb", "app/models/message.rb", force: true
     copy_file "models/chat.rb", "app/models/chat.rb", force: true
+    copy_file "models/appchat_function.rb", "app/models/appchat_function.rb", force: true
   end
 
-  def serialize_context
-    inject_into_class 'app/models/chat.rb', 'Chat', "serialize :context, coder:JSON, type: Array\n"
+  def copy_services
+    copy_file "services/appchat_function_service.rb", "app/services/appchat_function_service.rb", force: true
+    copy_file "services/web_search_service.rb", "app/services/web_search_service.rb", force: true
   end
 
-  def set_associations
-    inject_into_class 'app/models/chat.rb', 'Chat', "  has_many :messages, dependent: :destroy\n"
+  def serialize_context
+    inject_into_class 'app/models/chat.rb', 'Chat', "serialize :context, coder:JSON, type: Array\n"
   end
 
   def run_migrations
@@ -63,8 +89,22 @@ class AppchatGenerator < Rails::Generators::Base
     copy_file "get_ai_response_job.rb", "app/jobs/get_ai_response_job.rb"
   end
 
-  def create_messages_controller
-    copy_file "messages_controller.rb", "app/controllers/messages_controller.rb", force: true
+  def swap_class_in_layout
+    layout_file = "app/views/layouts/application.html.erb"
+
+    if File.exist?(layout_file)
+      gsub_file layout_file, /\bmt-28\b/, "mt-10"
+    else
+      say "Layout file not found. No changes were made.", :red
+    end
+  end
+
+  def copy_rake_tasks
+    copy_file "tasks/create_appchat_function.rake", "lib/tasks/create_appchat_function.rake"
+  end
+
+  def create_functions
+    rake "appchat_function:create_web_search"
   end
 
   def show_art

data/lib/generators/appchat/templates/assets/appchat.tailwind.css

@@ -59,4 +59,18 @@
     }
   }
 
+  #microphone-button.recording {
+    animation: flash 1s infinite;
+  }
+
+  @keyframes flash {
+    0%, 100% {
+      opacity: 80;
+      background-color: #EE4B2B; /* Ensure the color remains during the flash */
+    }
+    50% {
+      opacity: 0.5;
+      background-color: #EE4B2B; /* Ensure the color remains during the flash */
+    }
+  }
 }

data/lib/generators/appchat/templates/chats/chat.html.erb

@@ -1,5 +1,5 @@
 <%= link_to chat, data: { turbo_frame: :chat } do %>
-<div class="w-full text-gray-100 hover:text-green-500">
-  <%= chat&.messages&.first&.content || "New AI chat conversation" %>
-</div>
+  <div class="w-full text-gray-100 hover:text-green-500">
+    <%= chat&.messages&.first&.content&.truncate(45) || "New AI chat conversation" %>
+  </div>
 <% end %>

data/lib/generators/appchat/templates/chats/index.html.erb

@@ -3,20 +3,23 @@
     background-color: black;
   }
 </style>
+
 <div class="flex w-full">
-<%= turbo_stream_from :chats %>
-<div class="max-w-sm w-full h-full flex flex-col gap-4 p-2 w-full flex-shrink-0 bg-[#2e2e2e] rounded-lg" id="chats">
-  <%= link_to  chats_path, data: { turbo_method: :post, turbo_frame: :chat }, class: "flex gap-2 justify-center items-center bg-gray-500 text-gray-100 rounded-lg p-2 text-center mr-auto" do %>
-    <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" class="size-6">
-      <path stroke-linecap="round" stroke-linejoin="round" d="M12 4.5v15m7.5-7.5h-15" />
-    </svg>
-    <span> Create Chat </span>
-  <% end %>
-  <% @chats.each do |chat| %>
-    <%= render "chat", chat: chat %>
-  <% end %>
-</div>
-<div class="w-full ml-4">
-  <%= turbo_frame_tag :chat, src: ( chat_path(@chats.first) if @chats.any? )  %>
-</div>
+  <%= turbo_stream_from :chats %>
+  <div class="max-w-sm w-full h-full flex flex-col gap-4 p-2 w-full flex-shrink-0 bg-[#2e2e2e] rounded-lg" id="chats">
+    <%= link_to  chats_path, data: { turbo_method: :post, turbo_frame: :chat }, class: "flex gap-2 justify-center items-center bg-gray-500 text-gray-100 rounded-lg p-2 text-center mr-auto" do %>
+      <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" class="size-6">
+        <path stroke-linecap="round" stroke-linejoin="round" d="M12 4.5v15m7.5-7.5h-15" />
+      </svg>
+      <span> Create Chat </span>
+    <% end %>
+    <div class="h-[70vh] overflow-y-hidden overflow-y-scroll">
+      <% @chats.each do |chat| %>
+        <%= render "chat", chat: chat %>
+      <% end %>
+    </div>
+  </div>
+  <div class="w-full ml-4">
+    <%= turbo_frame_tag :chat, src: ( chat_path(@chats.first) if @chats.any? )  %>
+  </div>
 </div>

data/lib/generators/appchat/templates/chats_controller.rb

@@ -3,7 +3,7 @@ class ChatsController < ApplicationController
     @chat = Chat.find(params[:id])
   end
   def index
-    @chats = Chat.all
+    @chats = Chat.all.reverse
   end
   def create
     @chat = Chat.create

data/lib/generators/appchat/templates/get_ai_response_job.rb

@@ -1,27 +1,33 @@
 class GetAiResponseJob < ApplicationJob
   queue_as :default
 
-  attr_reader :chat, :user_prompt
+  attr_reader :client, :chat, :user_prompt, :informed_prompt, :message
 
   def perform(chat_id, user_prompt)
+    @client = new_client
     @chat = Chat.find(chat_id)
     @user_prompt = user_prompt
+    @message = chat.messages.create(role: 'assistant')
+    appchat_functions
     call_ollama
   end
 
   private
 
-  def call_ollama
-    client = Ollama.new(
+  def new_client
+    Ollama.new(
       credentials: { address: 'http://localhost:11434' },
       options: { server_sent_events: true }
     )
-    message = chat.messages.create(role: 'assistant')
+  end
+
+  def call_ollama
+    prompt = informed_prompt || user_prompt
 
     response = client.generate(
       {
         model: 'llama3.1',
-        prompt: user_prompt,
+        prompt: prompt,
         context: chat.context,
         stream: true,
       }
@@ -35,4 +41,20 @@ class GetAiResponseJob < ApplicationJob
       end
     chat.update(context: response.last["context"])
   end
+
+  def appchat_functions
+    appchat_function_service = AppchatFunctionService.new(chat.id, user_prompt).run
+    if appchat_function_service["match"] == "true" && appchat_function_service["appchat_function"].in?(AppchatFunction.pluck(:class_name))
+      puts "Function Match Found! --> #{appchat_function_service}"
+      function_class = appchat_function_service["appchat_function"].constantize
+      function_response = function_class.new(appchat_function_service["parameters"]).run do |status|
+        message.update(status: status)
+      end
+
+      return if function_response.nil?
+      @informed_prompt = "#{ user_prompt }, base your response on this data:
+        #{ appchat_function_service["name"] } responded with: #{ function_response },
+        :current_time => #{Date.current}"
+    end
+  end
 end

data/lib/generators/appchat/templates/javascript/speech_to_text_controller.js

@@ -0,0 +1,65 @@
+import { Controller } from "@hotwired/stimulus";
+
+// Connects to data-controller="speech-to-text"
+export default class extends Controller {
+  static targets = ["microphoneButton", "chatInput"]
+
+  connect() {
+    const SpeechRecognition = window.SpeechRecognition || window.webkitSpeechRecognition;
+    if (SpeechRecognition) {
+      this.microphoneButtonTarget.classList.remove("hidden")
+    } else {
+      return
+    }
+
+    this.recognition = new SpeechRecognition();
+    this.recognition.continuous = false;
+    this.recognition.lang = 'en-US';
+    this.recognition.interimResults = false;
+    this.recognition.maxAlternatives = 1;
+
+    this.recognition.onresult = this.handleResult.bind(this);
+    this.recognition.onerror = this.handleError.bind(this);
+    this.recognition.onstart = this.handleStart.bind(this);
+    this.recognition.onend = this.handleEnd.bind(this);
+
+    this.isRecognizing = false; // Track if recognition is active
+  }
+
+  startRecognition(event) {
+    event.preventDefault();
+    if (this.isRecognizing) {
+      console.log("Recognition already started");
+      return;
+    }
+
+    try {
+      this.recognition.start();
+      this.isRecognizing = true;
+    } catch (error) {
+      console.error("Recognition start failed:", error);
+      this.isRecognizing = false;
+    }
+  }
+
+  handleStart() {
+    this.microphoneButtonTarget.classList.add("recording");
+  }
+
+  handleEnd() {
+    this.microphoneButtonTarget.classList.remove("recording");
+    this.isRecognizing = false;
+  }
+
+  handleResult(event) {
+    const transcript = event.results[0][0].transcript;
+    this.chatInputTarget.value += ` ${transcript}`;
+    this.recognition.stop();
+  }
+
+  handleError(event) {
+    console.error(`Error occurred in recognition: ${event.error}`);
+    this.isRecognizing = false;
+    alert(`Speech recognition error: ${event.error}`);
+  }
+}

data/lib/generators/appchat/templates/messages/message.html.erb

@@ -1,5 +1,7 @@
 <div data-chat-message-target="message" id="<%= dom_id(message) %>" class="bg-gray-200 text-gray-900 shadow-lg p-4 rounded-lg">
-  <% if message.content? %>
+  <% if message.status? && !message.content? %>
+    <%= message.status %>
+  <% elsif message.content? %>
     <%=  message.content %>
   <% else %>
     <%= render 'messages/typing_bubbles' %>

data/lib/generators/appchat/templates/messages/new.html.erb

@@ -1,16 +1,39 @@
 <%= turbo_frame_tag :new_message do %>
   <%= form_with model: @chat.messages.new, url: messages_path(chat_id: @chat.id) do |f| %>
-   <div class="w-full flex items-center gap-2"> 
-   <div class="flex flex-col w-full">
-      <%= f.label :content, "What do you want to ask?" %>
-      <%= f.text_area :content, class: "w-full" %>
-    </div>
-    <%= button_tag class: "mt-4 flex p-2 rounded-lg bg-gradient-to-r from-indigo-500 to-blue-500 text-blue-100 font-semibold" do %>
-      Send Message
-      <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" class="size-6">
-        <path stroke-linecap="round" stroke-linejoin="round" d="M6 12 3.269 3.125A59.769 59.769 0 0 1 21.485 12 59.768 59.768 0 0 1 3.27 20.875L5.999 12Zm0 0h7.5" />
-      </svg>
-    <% end %>
-    </div>
+   <div class="w-full flex items-center gap-2 mt-4">
+      <div class="flex flex-col w-full" data-controller="speech-to-text">
+        <div class="relative w-full">
+          <%= f.text_area :content, class: "w-full pr-32 rounded-lg", "data-speech-to-text-target":"chatInput" %>
+          <button id="microphone-button"
+            data-speech-to-text-target="microphoneButton"
+            data-speech-to-text-recording-class="recording"
+            data-action="click->speech-to-text#startRecognition"
+            class="absolute right-14 top-1/2 transform -translate-y-1/2 p-1 mr-2 bg-gray-200 rounded-full hidden">
+            <svg
+              fill="#000000"
+              height="40px"
+              width="40px"
+              version="1.1"
+              xmlns="http://www.w3.org/2000/svg"
+              viewBox="0 0 512 512"
+              xmlns:xlink="http://www.w3.org/1999/xlink"
+              enable-background="new 0 0 512 512">
+              <g>
+                <g>
+                  <path d="m439.5,236c0-11.3-9.1-20.4-20.4-20.4s-20.4,9.1-20.4,20.4c0,70-64,126.9-142.7,126.9-78.7,0-142.7-56.9-142.7-126.9 0-11.3-9.1-20.4-20.4-20.4s-20.4,9.1-20.4,20.4c0,86.2 71.5,157.4 163.1,166.7v57.5h-23.6c-11.3,0-20.4,9.1-20.4,20.4 0,11.3 9.1,20.4 20.4,20.4h88c11.3,0 20.4-9.1 20.4-20.4 0-11.3-9.1-20.4-20.4-20.4h-23.6v-57.5c91.6-9.3 163.1-80.5 163.1-166.7z"/>
+                  <path d="m256,323.5c51,0 92.3-41.3 92.3-92.3v-127.9c0-51-41.3-92.3-92.3-92.3s-92.3,41.3-92.3,92.3v127.9c0,51 41.3,92.3 92.3,92.3zm-52.3-220.2c0-28.8 23.5-52.3 52.3-52.3s52.3,23.5 52.3,52.3v127.9c0,28.8-23.5,52.3-52.3,52.3s-52.3-23.5-52.3-52.3v-127.9z"/>
+                </g>
+              </g>
+            </svg>
+          </button>
+          <%= button_tag class: "absolute top-1 right-0 flex p-2 rounded-lg text-gray-900 font-semibold" do %>
+            <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" class="w-10">
+              <path stroke-linecap="round" stroke-linejoin="round" d="M6 12 3.269 3.125A59.769 59.769 0 0 1 21.485 12 59.768 59.768 0 0 1 3.27 20.875L5.999 12Zm0 0h7.5" />
+            </svg>
+          <% end %>
+        </div>
+      </div>
+
+      </div>
   <% end %>
 <% end %>

data/lib/generators/appchat/templates/models/appchat_function.rb

@@ -0,0 +1,18 @@
+class AppchatFunction < ApplicationRecord
+  has_many :function_parameters, dependent: :destroy
+
+  def to_prompt_hash
+    {
+      name: name,
+      description: description,
+      class_name: class_name,
+      parameters: function_parameters.each_with_object({}) do |param, hash|
+        hash[param.name] = param.example_value
+      end
+    }
+  end
+
+  def self.all_to_prompt_json
+    all.includes(:function_parameters).map(&:to_prompt_hash).to_json
+  end
+end

data/lib/generators/appchat/templates/services/appchat_function_service.rb

@@ -0,0 +1,47 @@
+class AppchatFunctionService
+  attr_reader :chat, :user_prompt, :response_json
+
+  def initialize(chat_id, user_prompt)
+    @chat = Chat.find(chat_id)
+    @user_prompt = user_prompt
+  end
+
+  def run
+    call_ollama
+    response_json
+  end
+
+  def call_ollama
+    client = Ollama.new(
+      credentials: { address: 'http://localhost:11434' },
+      options: { server_sent_events: true }
+    )
+    response = client.generate(
+      {
+        model: 'llama3.1',
+        prompt: prompt,
+        context: chat.context,
+        "format": "json"
+      }
+    )
+    @response_json = JSON.parse(response.map { |r| r["response"] }.join)
+  end
+  def prompt
+    <<-PROMPT
+      Evaluate the following user prompt in chat context.
+      Your goal is to determine whether the user prompt is requesting or requires additional information from any of the available services.
+        - If the prompt is a general greeting, casual remark, or something that can be answered without further information, respond with JSON { 'match' => 'false' }.
+        - If the prompt asks for specific information, requires a search, or needs a service to generate the correct response, then respond with JSON like this example:
+        {
+          'match' => 'true',
+          'appchat_function' => 'WebSearchService',
+          'parameters' => {
+            'query' => 'a query based on the prompt and context'
+          }
+        }
+
+      Here is the user's prompt: #{user_prompt}
+      Here are the Available Services: #{ AppchatFunction.all_to_prompt_json }
+    PROMPT
+  end
+end

data/lib/generators/appchat/templates/services/web_search_service.rb

@@ -0,0 +1,23 @@
+require 'watir'
+require 'cgi'
+
+class WebSearchService
+  attr_reader :query
+
+  def initialize(args)
+    @query = args["query"]
+  end
+
+  def run
+    yield("Searching Google for #{@query}") if block_given?
+
+    browser = Watir::Browser.new :chrome, headless: true
+    search_url = "https://www.google.com/search?q=#{CGI.escape(query)}"
+    browser.goto(search_url)
+    response = "browser_text: #{browser.text} browser_links: #{browser.links}"
+    browser.close
+    response
+  rescue Selenium::WebDriver::Error::UnknownError => e
+    "Error: #{e.message}"
+  end
+end

data/lib/generators/appchat/templates/tasks/create_appchat_function.rake

@@ -0,0 +1,13 @@
+# lib/tasks/create_appchat_function.rake
+namespace :appchat_function do
+  desc "Create an AppchatFunction with name 'Web Search', class name 'WebSearchService', and a description"
+  task create_web_search: :environment do
+    af = AppchatFunction.create!(
+      name: "Web Search",
+      class_name: "WebSearchService",
+      description: "Searches Google with a user's query and returns the page text and links"
+    )
+    puts "AppchatFunction 'Web Search' created successfully."
+    af.function_parameters.create(name: 'query', example_value: 'Dog friendly vegan resturants in Austin, TX')
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: appchat
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.6
 platform: ruby
 authors:
 - hackliteracy
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-26 00:00:00.000000000 Z
+date: 2024-09-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -66,7 +66,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.3.0
-description: The best and easiest framework for adding chats
+description: The best and easiest framework for adding AI chats
 email: hackliteracy@gmail.com
 executables: []
 extensions: []
@@ -80,13 +80,18 @@ files:
 - lib/generators/appchat/templates/chats_controller.rb
 - lib/generators/appchat/templates/get_ai_response_job.rb
 - lib/generators/appchat/templates/javascript/chat_message_controller.js
+- lib/generators/appchat/templates/javascript/speech_to_text_controller.js
 - lib/generators/appchat/templates/messages/_typing_bubbles.html.erb
 - lib/generators/appchat/templates/messages/index.html.erb
 - lib/generators/appchat/templates/messages/message.html.erb
 - lib/generators/appchat/templates/messages/new.html.erb
 - lib/generators/appchat/templates/messages_controller.rb
+- lib/generators/appchat/templates/models/appchat_function.rb
 - lib/generators/appchat/templates/models/chat.rb
 - lib/generators/appchat/templates/models/message.rb
+- lib/generators/appchat/templates/services/appchat_function_service.rb
+- lib/generators/appchat/templates/services/web_search_service.rb
+- lib/generators/appchat/templates/tasks/create_appchat_function.rake
 homepage: https://rubygems.org/gems/appchat
 licenses:
 - MIT
@@ -110,5 +115,5 @@ requirements: []
 rubygems_version: 3.5.17
 signing_key:
 specification_version: 4
-summary: Appchat makes it easy to add a chat to your app
+summary: Appchat makes it easy to add an AI chat to your app
 test_files: []