| 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 | |
| import os | |
| sys.path.append('../../') | |
| spectrograms = glob.glob(os.path.join(os.getcwd(),"temp*.npy")) | |
| generated_songs = glob.glob(os.path.join(os.getcwd(),"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") | |