from audiocraft.models import MusicGen
import streamlit as st
import os 
import torch
import torchaudio
import numpy as np
import base64

@st.cache_resource
def load_model():
    model=MusicGen.get_pretrained("facebook/musicgen-small")
    return model

st.set_page_config(
    page_icon=":musical_note:",
    page_title="Music Gen"
)

def main():
    st.title("Your Music")

    with st.expander("See Explanation"):
        st.write("App is developed by using Meta's Audiocraft Music Gen model. Write your text and we will generate audio")
    text_area=st.text_area("Enter description")
    time_slider=st.slider("Select time duration(s)",2,5,20)
    
    if text_area and time_slider:
        st.json(
            {
                "Description":text_area,
                "Selected duration:":time_slider
            }
        )
        st.subheader("Generated Music")

if __name__=="__main__":
    main()