import streamlit as st
from streamlit_star_rating import st_star_rating
st.set_page_config(initial_sidebar_state="collapsed")
import glob
import shutil
import sys
sys.path.append('../../')
spectrograms = glob.glob("temp*.npy")
generated_songs = glob.glob("temp*.wav")

st.markdown("# Original Song")
for s in st.session_state['song_list']:
    st.markdown(f"### {s.split('/')[-1].split('.')[0]}")
    st.audio(s, format='audio/wav')
st.markdown("# Generated Song")
st.audio(generated_songs[0], format='audio/wav')
rating = st_star_rating(label="rating", maxValue=10, defaultValue=3)


submit_rating = st.button("Submit Rating")

if submit_rating:
    shutil.copy(generated_songs[0],f"../DataSet/Song/srija_{st.session_state['model_name']}_{st.session_state['song_name']}_{st.session_state['similarity']}_{rating}.wav")
    shutil.copy(spectrograms[0],f"../DataSet/Spec/srija_{st.session_state['model_name']}_{st.session_state['song_name']}_{st.session_state['similarity']}_{rating}.npy")
    st.switch_page("app.py")