class Ollama::Handlers::Markdown

def call(response)
  if content = response.response || response.message&.content
    if @stream
      @content << content
      markdown_content = Kramdown::ANSI.parse(@content)
      @output.print clear_screen, move_home, markdown_content
    else
      markdown_content = Kramdown::ANSI.parse(content)
      @output.print markdown_content
    end
  end
  self
end

def initialize(output: $stdout, stream: true)
  super(output:)
  @stream      = stream
  @output.sync = @stream
  @content     = ''
end