refactoring

README.md

@@ -5,7 +5,7 @@ colorFrom: blue
 colorTo: pink
 sdk: streamlit
 sdk_version: 1.21.0
-app_file: app.py
+app_file: src/app.py
 pinned: false
 license: mit
 ---

app.py

@@ -1,34 +0,0 @@
-import streamlit as st
-
-# set page to wide
-st.set_page_config(layout="wide")
-
-st.header('ChatStudio')
-
-# 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',
-    ])
-
-
-
-# Create  st.tabs
-tabs = st.tabs(
-    [
-        "Wiki",
-        "Github",
-        "HuggingFace",
-    ]
-)
-
-
-
-
-
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
-

src/app.py

@@ -0,0 +1,43 @@
+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()

start.sh

@@ -0,0 +1 @@
+streamlit run src/app.py