import streamlit as st

# set page to wide
st.set_page_config(layout="wide")

st.header('ChatStudio')

def init():
    if 'chats' not in st.session_state:
        st.session_state.chats = {}

def main():
    init()

    # Create sidebar with dropdown menu with content about the app
    with st.sidebar:
        st.subheader('ChatStudio')
        page = st.selectbox("Choose a context", [
            'Wiki', 
            'Github',
            'HuggingFace',
        ])

        new_chat = st.button("New Chat", on_click=lambda: st.session_state.chats.update({page: ['N Chat']}))
        
        uploaded_docs = st.file_uploader("Upload documents")

    # Create  st.tabs
    tabs = st.tabs(
        [
            "Chat 1",
            "Chat 2",
        ]
    )

    if uploaded_docs:
        st.write(uploaded_docs)

    # Take user audio input from user


if __name__ == "__main__":
    main()