# Type all other functions here
def get_num_of_non_WS_characters(someString):
  counter = 0
  for c in someString:
    if c != ' ':
      counter += 1

  return counter

def get_num_of_words(someString):
  counter = 0
  for c in someString:
    if c == ' ':
      counter += 1

  return counter + 1

def fix_capitalization(someString):
  counter = 0
  newString = ''
  foundPunctuation = True
  foundSpace = True

  for c in someString:
    thingToAdd = c
    if foundPunctuation == True and foundSpace == True and c != ' ':
      foundPunctuation = False
      foundSpace = False
      if c.islower():
        thingToAdd = c.upper()
        counter += 1
    elif c == '.' or c == '!' or c == '?':

      foundPunctuation = True
    elif foundPunctuation == True and c == ' ':
      foundSpace = True

    newString += thingToAdd


  return newString, counter

def replace_punctuation(someString):
  counter_punct = 0
  counter_comma = 0
  newString = ''
  for c in someString:
    thingToAdd = c

    if c == '!':
      counter_punct += 1
      thingToAdd = '.'
    elif c == ';':
      counter_comma += 1
      thingToAdd = ','

    newString += thingToAdd

  return newString, counter_punct, counter_comma

def shorten_space(someString):
  last_character_was_a_space = False
  newString = ''
  for c in someString:
    if c.isspace() == True:
      if last_character_was_a_space == False:
        newString += c
        last_character_was_a_space = True
    else:
      newString += c
      last_character_was_a_space = False
  return newString

def print_menu(usrStr):
  menuOp = ''
  print('\nMENU')
  print('c - Number of non-whitespace characters')
  print('w - Number of words')
  print('f - Fix capitalization')
  print('r - Replace punctuation')
  print('s - Shorten spaces')
  print('q - Quit\n')
  menuOp = input('Choose an option:\n')

  if menuOp == 'c':
    num_char = get_num_of_non_WS_characters(usrStr)
    print("Number of non-whitespace characters:", num_char)
  elif menuOp == 'w':
    num_char = get_num_of_words(usrStr)
    print('Number of words:', num_char)
  elif menuOp == 'f':
    result = fix_capitalization(usrStr)
    print('Number of letters capitalized:', result[1])
    usrStr = result[0]
    print('Edited text:', usrStr)
  elif menuOp == 'r':
    result = replace_punctuation(usrStr)
    print('Punctuation replaced')
    print('exclamationCount:', result[1])
    print('semicolonCount:', result[2])
    usrStr = result[0]
    print('Edited text:', usrStr)
  elif menuOp == 's':
    usrStr = shorten_space(usrStr)
    print('Edited text:', usrStr)

  return menuOp, usrStr


if __name__ == '__main__':
  usrStr = input("Enter a sample text:\n")
  print('\nYou entered:', usrStr)
  lastKnownMenuOp = ''
  while lastKnownMenuOp != 'q':
    result = print_menu(usrStr)
    lastKnownMenuOp = result[0]
    usrStr = result[1]