Update app.py

app.py

@@ -1,3 +1,4 @@
+import awesome_streamlit as ast
 import streamlit as st
 import pandas as pd
 from io import StringIO
@@ -8,19 +9,49 @@ from sentence_transformers import SentenceTransformer, util
 #import lmdeploy
 #import turbomind as tm 
 
+from backend.utils import get_current_ram_usage, ga
+#import backend.aragpt
+import backend.home
+import backend.processor
+#import backend.sa
+#import backend.qa
+
 import os
 os.getenv("HF_TOKEN")
 
+st.set_page_config(
+    page_title="TEST", page_icon="📖", initial_sidebar_state="expanded", layout="wide"
+)
+
+ga(st.__file__)
+
 PAGES = {
-    "Home": Home,
+    "Home": backend.home,
     "Demo": Demo,
-    "About": About
+    "About": backend.home
 }
 
 st.sidebar.title("SBSmapper")
 selection = st.sidebar.radio("Pages", list(PAGES.keys()))
 
-
+page = PAGES[selection]
+# with st.spinner(f"Loading {selection} ..."):
+#ast.shared.components.write_page(page)
+
+st.sidebar.header("Info")
+st.sidebar.write("Project by JA RAD")
+#st.sidebar.write(
+#    "Pre-trained models are available on [HF Hub](https://huggingface.co/)"
+#)
+#st.sidebar.write(
+#    "Models source code available on [GitHub](https://github.com/)"
+#)
+#st.sidebar.write(
+#    "App source code available on [GitHub](https://github.com/)"
+#)
+if st.sidebar.checkbox("Show RAM usage"):
+    ram = get_current_ram_usage()
+    st.sidebar.write("Ram usage: {:.2f}/{:.2f} GB".format(ram[0], ram[1]))
 
 
 def on_click():