import json


def save_address_book(address_book_dict):
    address_book_file = open("address_book.txt", "r+")
    address_book_string = json.dumps(address_book_dict)
    address_book_file.write(address_book_string)
    address_book_file.close()


def insert_new_persons(adress_book_dict):
    while True:
        name = input("Enter a name: ")
        if name == "exit":
            break
        phone_number = input("Enter phone number")
        adress_book_dict[name] = phone_number


address_book_file = open("address_book.txt", "r+")
address_book = json.load(address_book_file)

input1 = input("Do you want to add new persons? y/n")

if input1 == "y":
    insert_new_persons(address_book)
    save_address_book(address_book)

input2 = input("Do you want to search for a person? y/n")

if input2 == "y":
    name = input("Please provide the name of the person: ")
    if name in address_book:
        print(address_book[name])
    else:
        print("Person not found")