understand 0.0.0 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 86a05e2ff179cb908a9b5400b548a31c7fe64f4dce41ee60614107a2bcf15704
-  data.tar.gz: e527daf41b0d24bbdb7763ae96b5f9dc50c891c278b42a61527a69e079b39dbf
+  metadata.gz: 252786af152482e56365f8e13b52340e2a48d3a9a8ef726657d75357c92891cc
+  data.tar.gz: 8a602e4f58a1320a166375289b7ba2332c4fc8f165ef24ba171602a6bbd4e3bf
 SHA512:
-  metadata.gz: 9be5dd6701740e086942ce7da003c31b1285759782f8eb973198c187620e2d4a7534f5b6844556595cbcf71e664c3c388b2e0a56018da42360362b41f8bf8e80
-  data.tar.gz: ac8b2a995a24ecbddab063c9c4b7af2a5568d6919bb575e039c50848bb12c7a46f86ae68d169e608577fec271876f89e725739dbbf008c3fd00906e95a0a9038
+  metadata.gz: e01bcc54fec5f2869272995d5c2a8f1f7c1de48f15452c2bb4347bc9e509a9abd275975a7a3b8d27ab70d033ffcaa91d6e60327accc9d1dc7b9a011296da73cd
+  data.tar.gz: b52894980c0a327294734be6e893c806f84af9f20a3b66da03afdc8ff3789b7c200bd0b36ff8f8b0edbc6056c40ccec744a8e6a6273d1ae3803047c42088a9da

data/bin/understand

@@ -1,3 +1,63 @@
 require "understand"
+require "bubbletea"
+require "lipgloss"
 
-Understand.hello
+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
+
+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,5 +1,19 @@
+require "rllama"
+
 class Understand
-  def self.hello
-    puts 'Hello world'
+  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) }
+    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
   end
 end

metadata

@@ -1,14 +1,56 @@
 --- !ruby/object:Gem::Specification
 name: understand
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.0.2
 platform: ruby
 authors:
 - Indigo Tech
 bindir: bin
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rllama
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !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