litbot.rb

class LitBot
  def eat_file(file_name)
    file = File.open(file_name)
    self.source_text = file.read
    file.close
  end

  def speak_first_10_words
    puts source_text.split(' ').first(10)
  end

  def digest_file
    @source_text_words = source_text.split(' ')
    @word_pairs_and_probabilities = {}

    @source_text_words.each_with_index do |word, index|
      hash_key = "#{word} #{@source_text_words[index + 1]}"
      hash_value = @source_text_words[index + 2]
      if @word_pairs_and_probabilities[hash_key]
        @word_pairs_and_probabilities[hash_key] << hash_value
      else
        @word_pairs_and_probabilities[hash_key] = [hash_value]
      end
    end
  end

  def speak
    output_text = generate_random_seed_pair
    story = 0
    while story < 35 do
      word_pair = output_text.last(2).join(' ')
      next_word = @word_pairs_and_probabilities[word_pair].sample unless @word_pairs_and_probabilities[word_pair].nil?
      output_text << next_word

      if next_word && (next_word.include?(".") || next_word.include?("?"))
        story += 1
      end
    end
    output_text.join(' ')
  end

  def generate_random_seed_pair
    trial_pair = @word_pairs_and_probabilities.keys.sample
    until trial_pair[0] == trial_pair[0].upcase do
      trial_pair = @word_pairs_and_probabilities.keys.sample
    end
    trial_pair.split(' ')
  end

  def +(other_bot)
    shiny_new_litbot = LitBot.new(source_text + other_bot.source_text)
  end

  #getter
  def source_text
    @source_text
  end

  #setter
  def source_text=(text)
    @source_text = text
  end

  def initialize(text="Any string")
    self.source_text = text
  end
end