Class: AIA::UIPresenter

Inherits:
Object
  • Object
show all
Defined in:
lib/aia/ui_presenter.rb

Constant Summary collapse

USER_PROMPT = 
"Follow up (cntl-D or 'exit' to end) #=> "

Instance Method Summary collapse

Constructor Details

#initialize ⇒ UIPresenter

Returns a new instance of UIPresenter.



11
12
13
 
# File 'lib/aia/ui_presenter.rb', line 11

def initialize
  @terminal_width = TTY::Screen.width
end

Instance Method Details

#ask_question ⇒ Object

This is the follow up question in a chat session



82
83
84
85
86
87
88
89
90
91
92
93
94
 
# File 'lib/aia/ui_presenter.rb', line 82

def ask_question
  puts USER_PROMPT
  $stdout.flush  # Ensure the prompt is displayed immediately
  begin
    input = Reline.readline('', true)
    return nil if input.nil? # Handle Ctrl+D
    Reline::HISTORY << input unless input.strip.empty?
    input
  rescue Interrupt
    puts "\nChat session interrupted."
    return 'exit'
  end
end

#display_ai_response(response) ⇒ Object 



20
21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/aia/ui_presenter.rb', line 20

def display_ai_response(response)
  puts "\nAI: "
  format_chat_response(response)

  out_file = AIA.config.output.file
  if out_file && !out_file.nil?
    File.open(out_file, 'a') do |file|
      file.puts "\nAI: "
      format_chat_response(response, file)
    end
  end
end

#display_chat_end ⇒ Object 



76
77
78
 
# File 'lib/aia/ui_presenter.rb', line 76

def display_chat_end
  puts "\nChat session ended."
end

#display_chat_header ⇒ Object 



15
16
17
 
# File 'lib/aia/ui_presenter.rb', line 15

def display_chat_header
  puts "#{'═' * @terminal_width}\n"
end

#display_error(message) ⇒ Object 



100
101
102
 
# File 'lib/aia/ui_presenter.rb', line 100

def display_error(message)
  warn "ERROR: #{message}"
end

#display_info(message) ⇒ Object 



96
97
98
 
# File 'lib/aia/ui_presenter.rb', line 96

def display_info(message)
  puts "\n#{message}"
end

#display_multi_model_metrics(metrics_list) ⇒ Object 



148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
 
# File 'lib/aia/ui_presenter.rb', line 148

def display_multi_model_metrics(metrics_list)
  return unless metrics_list && !metrics_list.empty?
  
  output_lines = []
  
  # Determine table width based on whether costs are shown
  if AIA.config.flags.cost
    table_width = 80
  else
    table_width = 60
  end
  
  output_lines << "═" * table_width
  output_lines << "Multi-Model Token Usage"
  output_lines << "─" * table_width
  
  # Build header row
  if AIA.config.flags.cost
    output_lines << sprintf("%-20s %10s %10s %10s %12s %10s", 
                            "Model", "Input", "Output", "Total", "Cost", "x1000")
    output_lines << "─" * table_width
  else
    output_lines << sprintf("%-20s %10s %10s %10s", 
                            "Model", "Input", "Output", "Total")
    output_lines << "─" * table_width
  end
  
  # Process each model
  total_input = 0
  total_output = 0
  total_cost = 0.0
  
  metrics_list.each do |metrics|
    # Use display_name if available (includes role), otherwise fall back to model_id
    model_name = metrics[:display_name] || metrics[:model_id]
    # Truncate model name if too long
    model_name = model_name[0..17] + ".." if model_name.to_s.length > 19

    input_tokens = metrics[:input_tokens] || 0
    output_tokens = metrics[:output_tokens] || 0
    total_tokens = input_tokens + output_tokens
    
    if AIA.config.flags.cost
      cost_data = calculate_cost(metrics)
      if cost_data[:available]
        cost_str = "$#{'%.5f' % cost_data[:total_cost]}"
        x1000_str = "$#{'%.2f' % (cost_data[:total_cost] * 1000)}"
        total_cost += cost_data[:total_cost]
      else
        cost_str = "N/A"
        x1000_str = "N/A"
      end
      
      output_lines << sprintf("%-20s %10d %10d %10d %12s %10s",
                              model_name, input_tokens, output_tokens, total_tokens, cost_str, x1000_str)
    else
      output_lines << sprintf("%-20s %10d %10d %10d",
                              model_name, input_tokens, output_tokens, total_tokens)
    end
    
    total_input += input_tokens
    total_output += output_tokens
  end
  
  # Display totals row
  output_lines << "─" * table_width
  total_tokens = total_input + total_output
  
  if AIA.config.flags.cost && total_cost > 0
    cost_str = "$#{'%.5f' % total_cost}"
    x1000_str = "$#{'%.2f' % (total_cost * 1000)}"
    output_lines << sprintf("%-20s %10d %10d %10d %12s %10s",
                           "TOTAL", total_input, total_output, total_tokens, cost_str, x1000_str)
  else
    output_lines << sprintf("%-20s %10d %10d %10d",
                           "TOTAL", total_input, total_output, total_tokens)
  end
  
  output_lines << "═" * table_width
  
  # Output to STDOUT
  output_lines.each { |line| puts line }
  
  # Also write to file if configured
  out_file = AIA.config.output.file
  if out_file && !out_file.nil?
    File.open(out_file, 'a') do |file|
      output_lines.each { |line| file.puts line }
    end
  end
end

#display_separator ⇒ Object 



71
72
73
 
# File 'lib/aia/ui_presenter.rb', line 71

def display_separator
  puts "\n#{'─' * @terminal_width}"
end

#display_token_metrics(metrics) ⇒ Object 



121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
 
# File 'lib/aia/ui_presenter.rb', line 121

def display_token_metrics(metrics)
  return unless metrics
  
  output_lines = []
  output_lines << "═" * 55
  output_lines << "Model: #{metrics[:model_id]}"
  
  if AIA.config.flags.cost
    output_lines.concat(format_metrics_with_cost(metrics))
  else
    output_lines.concat(format_metrics_basic(metrics))
  end
  
  output_lines << "═" * 55
  
  # Output to STDOUT
  output_lines.each { |line| puts line }
  
  # Also write to file if configured
  out_file = AIA.config.output.file
  if out_file && !out_file.nil?
    File.open(out_file, 'a') do |file|
      output_lines.each { |line| file.puts line }
    end
  end
end

#display_warning(message) ⇒ Object 



104
105
106
 
# File 'lib/aia/ui_presenter.rb', line 104

def display_warning(message)
  warn "WARNING: #{message}"
end

#format_chat_response(response, output = $stdout) ⇒ Object 



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
# File 'lib/aia/ui_presenter.rb', line 34

def format_chat_response(response, output = $stdout)
  indent = '   '

  # Convert RubyLLM::Message to string if necessary
  response_text = if response.is_a?(RubyLLM::Message)
                    response.content.to_s
                  elsif response.respond_to?(:to_s)
                    response.to_s
                  else
                    response
                  end

  in_code_block = false
  language = ''

  response_text.each_line do |line|
    line = line.chomp

    # Check for code block delimiters
    if (match = line.match(/^```(\w*)$/)) && !in_code_block
      in_code_block = true
      language = match[1]
      output.puts "#{indent}```#{language}"
    elsif line.match?(/^```$/) && in_code_block
      in_code_block = false
      output.puts "#{indent}```"
    elsif in_code_block
      # Print code with special formatting
      output.puts "#{indent}#{line}"
    else
      # Handle regular text
      output.puts "#{indent}#{line}"
    end
  end
end

#with_spinner(message = "Processing", operation_type = nil) ⇒ Object 



108
109
110
111
112
113
114
115
116
117
118
119
 
# File 'lib/aia/ui_presenter.rb', line 108

def (message = "Processing", operation_type = nil)
   = operation_type ? "#{message} #{operation_type}..." : "#{message}..."
   = TTY::Spinner.new("[:spinner] #{}", format: :bouncing_ball)
  .auto_spin

  begin
    result = yield
  ensure
    .stop
  end
  result
end