def main(arg1, arg2):
    global similarities_list
    global user_input
    global country_name

    filename = "RawCorpus2.txt"
    create_corpus(filename)

    user_input = arg1
    country_name = arg2

    similarities_list = DataBase.get_data(country_name, user_input)

    if similarities_list:
        gen_colors(similarities_list)
    else:
        similarities_list = doc_similarity(user_input)
        DataBase.update_file(country_name, user_input, similarities_list)
        gen_colors(similarities_list)
def update_similarities(updated_similarity_list):

    similarities_list = updated_similarity_list
    DataBase.update_file(country_name, user_input, similarities_list)
    gen_colors(similarities_list)