Enhance chat UI generator with improved UX and models management

- Simplify form inputs: remove labels, add placeholders, autofocus
- Add models table view with pricing information ($/1M tokens)
- Create model partial for cleaner index view
- Add refresh models functionality (synchronous, no background job)
- Improve post-install message to be concise (DHH-style)
- Remove README template in favor of inline generator message
- Fix broadcasting setup for messages
- Make forms submit on Enter key
- Follow Rails scaffold conventions for views

Author: crmne

lib/generators/ruby_llm/chat_ui/chat_ui_generator.rb

@@ -59,6 +59,8 @@ def create_views
         # Model views
         template 'views/models/index.html.erb', "app/views/#{model_model_name.tableize}/index.html.erb"
         template 'views/models/show.html.erb', "app/views/#{model_model_name.tableize}/show.html.erb"
+        template 'views/models/_model.html.erb',
+                 "app/views/#{model_model_name.tableize}/_#{model_model_name.underscore}.html.erb"
       end
 
       def create_controllers
@@ -72,7 +74,14 @@ def create_jobs
       end
 
       def add_routes
-        route "resources :#{model_model_name.tableize}, only: [:index, :show]"
+        model_routes = <<~ROUTES.strip
+          resources :#{model_model_name.tableize}, only: [:index, :show] do
+            collection do
+              post :refresh
+            end
+          end
+        ROUTES
+        route model_routes
         chat_routes = <<~ROUTES.strip
           resources :#{chat_model_name.tableize} do
             resources :#{message_model_name.tableize}, only: [:create]
@@ -95,7 +104,11 @@ def add_broadcasting_to_message_model
       end
 
       def display_post_install_message
-        readme 'README' if behavior == :invoke
+        return unless behavior == :invoke
+
+        say "\n  ✅ Chat UI installed!", :green
+        say "\n  Start your server and visit http://localhost:3000/#{chat_model_name.tableize}", :cyan
+        say "\n"
       end
     end
   end

lib/generators/ruby_llm/chat_ui/templates/README

@@ -6,4 +6,9 @@ class <%= model_model_name.pluralize %>Controller < ApplicationController
   def show
     @<%= model_model_name.underscore %> = <%= model_model_name %>.find(params[:id])
   end
+
+  def refresh
+    <%= model_model_name %>.refresh!
+    redirect_to <%= model_model_name.tableize %>_path, notice: "<%= model_model_name.pluralize %> refreshed successfully"
+  end
 end

lib/generators/ruby_llm/chat_ui/templates/controllers/models_controller.rb.tt

@@ -1,6 +1,16 @@
 <div id="<%%= dom_id <%= chat_model_name.underscore %> %>">
-  <p><strong><%= chat_model_name %> #<%%= <%= chat_model_name.underscore %>.id %></strong></p>
-  <p><%%= <%= chat_model_name.underscore %>.model_association.name %></p>
-  <p><%%= <%= chat_model_name.underscore %>.created_at.strftime("%B %d, %Y at %I:%M %p") %></p>
-  <p><%%= pluralize(<%= chat_model_name.underscore %>.messages_association.count, 'message') %></p>
+  <div>
+    <strong>Model:</strong>
+    <%%= <%= chat_model_name.underscore %>.model_association.name %>
+  </div>
+
+  <div>
+    <strong>Messages:</strong>
+    <%%= <%= chat_model_name.underscore %>.messages_association.count %>
+  </div>
+
+  <div>
+    <strong>Created:</strong>
+    <%%= <%= chat_model_name.underscore %>.created_at.strftime("%B %d, %Y at %I:%M %p") %>
+  </div>
 </div>

lib/generators/ruby_llm/chat_ui/templates/views/chats/_chat.html.erb.tt

@@ -20,11 +20,10 @@
   </div>
 
   <div style="margin-top: 15px;">
-    <%%= form.label :prompt, "Enter your prompt:", style: "display: block" %>
-    <%%= form.text_area :prompt, rows: 4, style: "width: 100%; max-width: 600px;" %>
+    <%%= form.text_field :prompt, style: "width: 100%; max-width: 600px;", placeholder: "What would you like to discuss?", autofocus: true %>
   </div>
 
-  <div style="margin-top: 15px;">
+  <div>
     <%%= form.submit "Start new <%= chat_model_name.underscore.humanize.downcase %>" %>
   </div>
 <%% end %>

lib/generators/ruby_llm/chat_ui/templates/views/chats/_form.html.erb.tt

@@ -12,8 +12,7 @@
   <%% end %>
 
   <div>
-    <%%= form.label :content, "Your message:", style: "display: block" %>
-    <%%= form.text_area :content, rows: 3, style: "width: 100%; max-width: 600px;" %>
+    <%%= form.text_field :content, style: "width: 100%; max-width: 600px;", placeholder: "Message...", autofocus: true %>
   </div>
 
   <div>

lib/generators/ruby_llm/chat_ui/templates/views/messages/_form.html.erb.tt

@@ -0,0 +1,16 @@
+<tr id="<%%= dom_id <%= model_model_name.underscore %> %>">
+  <td><%%= <%= model_model_name.underscore %>.provider %></td>
+  <td><%%= <%= model_model_name.underscore %>.name %></td>
+  <td><%%= number_with_delimiter(<%= model_model_name.underscore %>.context_window) if <%= model_model_name.underscore %>.context_window %></td>
+  <td>
+    <%% if <%= model_model_name.underscore %>.pricing && <%= model_model_name.underscore %>.pricing['text_tokens'] && <%= model_model_name.underscore %>.pricing['text_tokens']['standard'] %>
+      <%% input = <%= model_model_name.underscore %>.pricing['text_tokens']['standard']['input_per_million'] %>
+      <%% output = <%= model_model_name.underscore %>.pricing['text_tokens']['standard']['output_per_million'] %>
+      <%% if input && output %>
+        $<%%= "%.2f" % input %> / $<%%= "%.2f" % output %>
+      <%% end %>
+    <%% end %>
+  </td>
+  <td><%%= link_to "Show", <%= model_model_name.underscore %> %></td>
+  <td><%%= link_to "Start <%= chat_model_name.underscore.humanize.downcase %>", new_<%= chat_model_name.underscore %>_path(model: <%= model_model_name.underscore %>.model_id) %></td>
+</tr>

lib/generators/ruby_llm/chat_ui/templates/views/models/_model.html.erb.tt

@@ -1,19 +1,30 @@
+<p style="color: green"><%%= notice %></p>
+
 <%% content_for :title, "<%= model_model_name.pluralize %>" %>
 
 <h1><%= model_model_name.pluralize %></h1>
 
-<%% @<%= model_model_name.tableize %>.each do |provider, <%= model_model_name.tableize %>| %>
-  <h2><%%= provider.upcase %></h2>
+<p>
+  <%%= button_to "Refresh <%= model_model_name.pluralize %>", refresh_<%= model_model_name.tableize %>_path, method: :post %>
+</p>
 
-  <ul>
-    <%% <%= model_model_name.tableize %>.each do |<%= model_model_name.underscore %>| %>
-      <li>
-        <strong><%%= <%= model_model_name.underscore %>.name %></strong>
-        (<%%= <%= model_model_name.underscore %>.model_id %>)
-        - <%%= link_to "Start <%= chat_model_name.underscore.humanize.downcase %>", new_<%= chat_model_name.underscore %>_path(model: <%= model_model_name.underscore %>.model_id) %>
-      </li>
-    <%% end %>
-  </ul>
-<%% end %>
+<div id="<%= model_model_name.tableize %>">
+  <table>
+    <thead>
+      <tr>
+        <th>Provider</th>
+        <th>Model</th>
+        <th>Context Window</th>
+        <th>$/1M tokens (In/Out)</th>
+        <th colspan="2"></th>
+      </tr>
+    </thead>
+    <tbody>
+      <%% @<%= model_model_name.tableize %>.values.flatten.each do |<%= model_model_name.underscore %>| %>
+        <%%= render <%= model_model_name.underscore %> %>
+      <%% end %>
+    </tbody>
+  </table>
+</div>
 
-<p><%%= link_to "Back to <%= chat_model_name.tableize.humanize.downcase %>", <%= chat_model_name.tableize %>_path %></p>
+<%%= link_to "Back to <%= chat_model_name.tableize.humanize.downcase %>", <%= chat_model_name.tableize %>_path %>