import streamlit as st
audioldm_exmples_input_paths = []
audioldm_exmples_original_paths = []
exmples_prompts = ["wiwi", "pipi","wifi"]


st.title('Drums generation in different models')
add_selectbox = st.sidebar.selectbox(
    'How would you like to be contacted?',
    ('Email', 'Home phone', 'Mobile phone')
)

# Add a slider to the sidebar:
add_slider = st.sidebar.slider(
    'Select a range of values',
    0.0, 100.0, (25.0, 75.0)
)
st.subheader('This is a subheader with a divider', divider='rainbow')
with st.container(border = True, height=300):
    st.title("Examples")
    col1, col2, col3, col4 = st.columns(4)
    with col1:
        st.subheader("original audio")
        # with st.container(border = True,height=120):

            # st.write("some prompts...")
            # st.audio("goodres.wav", format="audio/mpeg", loop=False)
            # st.write("some prompts...")
            # st.audio("goodres.wav", format="audio/mpeg", loop=False)
    with col2:
        st.subheader("Separated Audio")
        # with st.container(border = True,height=120):

        #     # st.write("some prompts...")
        #     st.audio("goodres.wav", format="audio/mpeg", loop=False)
        #     # st.write("some prompts...")
        #     st.audio("goodres.wav", format="audio/mpeg", loop=False)
    with col3:
        st.subheader("AudioLDM2")
        # with st.container(border = True,height=120):

        #     st.write("some prompts...")
        #     st.audio("goodres.wav", format="audio/mpeg", loop=False)
        #     st.write("some prompts...")
        #     st.audio("goodres.wav", format="audio/mpeg", loop=False)
    with col4:
        st.subheader("StableAudio")
        # with st.container(border = True,height=120):
        #     st.write("some prompts...")
        #     st.audio("goodres.wav", format="audio/mpeg", loop=False)
        #     st.write("some prompts...")
        #     st.audio("goodres.wav", format="audio/mpeg", loop=False)
            
    with st.container(border = True,height=120):
        for p in exmples_prompts:
            st.write(p)
            col1, col2, col3, col4 = st.columns(4)
            with col1:
                # st.write("some prompts...")
                st.audio("goodres.wav", format="audio/mpeg", loop=False)
                # st.write("some prompts...")
                # st.audio("goodres.wav", format="audio/mpeg", loop=False)
            with col2:
                # st.subheader("Separated Audio")
                # with st.container(border = True,height=120):
        
                    # st.write("some prompts...")
                st.audio("goodres.wav", format="audio/mpeg", loop=False)
                    # st.write("some prompts...")
                    # st.audio("goodres.wav", format="audio/mpeg", loop=False)
            with col3:
                # st.subheader("AudioLDM2")
                # with st.container(border = True,height=120):
        
                    # st.write("some prompts...")
                st.audio("goodres.wav", format="audio/mpeg", loop=False)
                    # st.write("some prompts...")
                    # st.audio("goodres.wav", format="audio/mpeg", loop=False)
            with col4:
                # st.subheader("StableAudio")
                # with st.container(border = True,height=120):
                    # st.write("some prompts...")
                st.audio("goodres.wav", format="audio/mpeg", loop=False)
                    # st.write("some prompts...")
                    # st.audio("goodres.wav", format="audio/mpeg", loop=False)