Update app.py

app.py

@@ -1,30 +1,30 @@
 import awesome_streamlit as ast
 import streamlit as st
-from home import write as write_home
-from pos_tagger import write as write_pos_tagger
+import home
+import pos_tagger
 
-def main():
-    st.set_page_config(
-        page_title="Hamshetsnag POS Tagger",
+st.set_page_config(page_title="Hamshetsnag POS Tagger",
         page_icon="📚",
         layout='wide'
     )
-
-    PAGES = {
-        "Home": home_page,
-        "POS Tagging": pos_tagging_page,
+PAGES = {
+        "Home": home,
+        "POS Tagging": pos_tagger,
     }
 
-    st.sidebar.title("Navigation")
-    selection = st.sidebar.radio("Pages", list(PAGES.keys()))
+st.sidebar.title("Navigation")
+selection = st.sidebar.radio("Pages", list(PAGES.keys()))
 
     # Load and display the selected page
-    PAGES[selection]()
+    page = PAGES[selection]
+    # with st.spinner(f"Loading {selection} ..."):
+ast.shared.components.write_page(page)
+
+st.sidebar.header("Info")
 
-    st.sidebar.header("Info")
-    st.sidebar.write(
+st.sidebar.write(
         "Models available on [HF Hub](https://huggingface.co/onurkeles/hamshetsnag-pos-tagger)"
     )
-    st.sidebar.write(
+st.sidebar.write(
         "Model source code will be soon available on [GitHub](https://github.com/kelesonur)"
     )