understand 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2cf96bc7bb590fd85ed5e7c0ee0f0732a0678bd841392ad8294c7153014bdaa5
-  data.tar.gz: 52678bf6c6646f12a7740643623375d18f04d926a8332ebe9f6a9a7b03342ab0
+  metadata.gz: 252786af152482e56365f8e13b52340e2a48d3a9a8ef726657d75357c92891cc
+  data.tar.gz: 8a602e4f58a1320a166375289b7ba2332c4fc8f165ef24ba171602a6bbd4e3bf
 SHA512:
-  metadata.gz: d0cf7721893fca0e554e16bb8d872527e42402b186f75ce4c6c00047677c640eb53b1ac8fd90577d2df55b19f1ffe8b3d7901e08c19796846a30c394b1359f22
-  data.tar.gz: 6e0b3949a9dc3a1df9cd97251d3146b8de3952d0420f151e464d793c41f588117bcb3d457a12390fe1618c1609e3dcb81f0dd54cb5f1544ffa54bf4e4191502f
+  metadata.gz: e01bcc54fec5f2869272995d5c2a8f1f7c1de48f15452c2bb4347bc9e509a9abd275975a7a3b8d27ab70d033ffcaa91d6e60327accc9d1dc7b9a011296da73cd
+  data.tar.gz: b52894980c0a327294734be6e893c806f84af9f20a3b66da03afdc8ff3789b7c200bd0b36ff8f8b0edbc6056c40ccec744a8e6a6273d1ae3803047c42088a9da

data/bin/understand

@@ -1,7 +1,63 @@
 require "understand"
-user_prompt = if ARGV[0]
-  user_prompt
-else
-  "Help me understand my application more."
+require "bubbletea"
+require "lipgloss"
+
+class UnderstandCLI
+  include Bubbletea::Model
+  attr_accessor :ai_response_header, :ai_response
+  def initialize
+    @ai_response_header = "🔎 Understanding your codebase"
+    @ai_processing = false
+    @user_prompt = ""
+    @ai_response = ""
+  end
+
+  def init
+    [self, nil]
+  end
+
+  def update(message)
+    case message
+    when Bubbletea::KeyMessage
+      case message.to_s
+      when "enter"
+        if !@ai_processing && @user_prompt&.chars&.any?
+          @ai_processing = true
+          @user_prompt = ""
+          Understand.call(self)
+          [self, nil]
+        end
+        [self, nil]
+      when "esc", "ctrl+c"
+        [self, Bubbletea.quit]
+      when "backspace"
+        @user_prompt = @user_prompt[0..-2]
+        [self, nil]
+      when "space"
+        @user_prompt += " "
+        [self, nil]
+      else
+        @user_prompt += message.to_s
+        [self, nil]
+      end
+    end
+  end
+
+  def view
+    if @ai_processing
+      "#{@ai_response_header} \n \n AI: #{@ai_response}"
+    else
+      "💡 Understand your codebase now \n Enter your prompt: #{@user_prompt} \n\n (Press enter to submit)"
+    end
+  end
 end
-Understand.new(user_prompt).call
+
+Bubbletea.run(UnderstandCLI.new)
+
+# user_prompt = if ARGV[0]
+#   user_prompt
+# else
+#   "Help me understand my application more."
+# end
+
+#Understand.new(user_prompt).call

data/lib/understand.rb

@@ -1,21 +1,19 @@
 require "rllama"
 
 class Understand
-  attr_reader :user_prompt
-  def initialize(user_prompt)
-    @user_prompt = user_prompt
-  end
-
-  def call
+  def self.call(understand_cli)
     model = Rllama.load_model('lmstudio-community/gemma-3-1B-it-QAT-GGUF/gemma-3-1B-it-QAT-Q4_0.gguf')
     files = Dir.glob("**/*").select { |f| File.file?(f) }
-    files.each do |file|
-      prompt_for_ai = user_prompt + "--- File path: #{file} Content: #{File.read(file)}"
-      puts "💡 🔎 Understanding #{file}"
-      result = model.generate(prompt_for_ai) do |token|
-        print token
+    Thread.new do
+      files.each do |file|
+        prompt = "Help me understand my application"
+        prompt_for_ai = prompt + "--- File path: #{file} Content: #{File.read(file)}"
+        understand_cli.ai_response_header = "💡 🔎 Understanding #{file}"
+        model.generate(prompt_for_ai) do |token|
+          understand_cli.ai_response += token
+        end
       end
+      model.close
     end
-    model.close
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: understand
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Indigo Tech
@@ -23,6 +23,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.0.3
+- !ruby/object:Gem::Dependency
+  name: bubbletea
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0
+- !ruby/object:Gem::Dependency
+  name: lipgloss
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
 description: A gem to help you understand your codebases using the power of LLM's
   ( not a vibe coding tool )
 email: indigo@tech.tut