# STREAMLIT STUFF
import streamlit as st
from streamlit_tags import st_tags

# PYTHON STUFF
import json

# CUSTOM
from nlp.nlp import get_nearest_tags

f = open('nlp/tags.json')        # OPEN JSON FILE
data = json.load(f)             # LOAD DATA
f.close()                       # CLOSE FILE

# --------------------------- VARIABLES / MACROS

# SPLIT DATA
GENRE_TAGS = data['genre']
MOOD_TAGS = data['mood']
INSTRUMENT_TAGS = data['instrument']

MAX_TAGS = 3

# --------------------------- TITLE
st.title("FIRE COML FALL 2022 Audio Generation Model")

# --------------------------- MAIN
# SELECT TYPE OF GENERATION
st.write("Enter Genre(s), Mood/Theme(s), and Instrument(s)")
instructions = "Press ENTER to add more"

genre = st_tags(
    label='Enter Genre(s):',
    text=instructions,
    suggestions=GENRE_TAGS,
    maxtags=MAX_TAGS,
    key='1'
)

mood = st_tags(
    label='Enter Mood/Theme(s):',
    text=instructions,
    suggestions=MOOD_TAGS,
    maxtags=MAX_TAGS,
    key='2'
)

instrument = st_tags(
    label='Enter Instrument(s):',
    text=instructions,
    suggestions=INSTRUMENT_TAGS,
    maxtags=MAX_TAGS,
    key='3'
)

# SUBMIT TAGS
if st.button("Submit Tags"):
    st.write("Tags are:")
    inp = genre + mood + instrument
    st.write("Before: " + str(inp))
    input_tags = get_nearest_tags(inp)
    st.write("After: " + str(input_tags))