import re
import json
prompt_shortcut ={}
try:
    with open('prompt_shortcut.json') as f_obj:
        data = json.load(f_obj)
        prompt_shortcut = data
        print(data)
except IOError:
    print("prompt_shortcut.json is not found")
    
def find_words_inside_braces(string):
    result = ""
    pattern ="\{(.*?)\}"
    keywords = []
    for match in re.findall(pattern, string):
        keywords.append(match)
        result += match
    return keywords

# text = "a beautiful girl holding a cute cat {style_1} on sunny day"
# text = "a beautiful girl holding a cute cat {    style_1 } on sunny day"
text = "a beautiful girl{    }, {char1}, {painterly_style} holding a cute cat {    style_1 } on sunny day"
# text = "a beautiful girl {char1 } holding a cute cat on sunny day"


def replaceShortcut(text):
    raw_keywords = find_words_inside_braces(text)
    strip_keywords = list(map(lambda s: s.strip(),raw_keywords))

    original_substrings = list(map(lambda s: '{'+s+'}',raw_keywords)) 

    print("raw_keywords: ", raw_keywords)
    print("strip_keywords: ", strip_keywords)
    print("original_substrings: ",original_substrings)
    # print ("text:",text)

    for i, word in enumerate(strip_keywords):
        # word = word.strip()
        print("word: ",word)
        if len(word) > 0 and word in prompt_shortcut:
            prompt = prompt_shortcut[word]
            print("prompt: ",prompt)
            text = text.replace(original_substrings[i],prompt)
        # else:
            # text = text.replace(word,word.strip())
    print ("final text: ",text) 
    return text