Update app.py

app.py

@@ -1,31 +1,21 @@
 import streamlit as st
-import home  # Import home.py for the home page
-import prediction  # Import predictions.py for the predictions page
-import about  # Import about.py for the about page
-import contact  # Import contact.py for the contact page
+from home import show_home
+from prediction import show_prediction
+from contact import show_contact
+from about import show_about
 
-# Set up navigation logic
-def load_page(page):
-    if page == "Home":
-        home.show()  # Call home.py's show() function
-    elif page == "Predictions":
-        prediction.show()  # Call predictions.py's show() function
-    elif page == "About":
-        about.show()  # Call about.py's show() function
-    elif page == "Contact":
-        contact.show()  # Call contact.py's show() function
-    else:
-        st.error("Page not found")
+st.set_page_config(page_title="TransPolymer", layout="wide", page_icon="🧪")
 
-# Navigation menu
-st.sidebar.title("Navigation")
-page = st.sidebar.radio("Select a Page", ["Home", "Predictions", "About", "Contact"])
+# Sidebar for navigation
+st.sidebar.title("🔗 Navigation")
+page = st.sidebar.radio("Go to", ["Home", "Prediction", "Contact Us", "About"])
 
-# Call the function to display the selected page
-load_page(page)
-
-# Handle page URL updates
-query_params = st.query_params
-if "page" in query_params:
-    page = query_params["page"][0]  # Get page value from URL parameters
-    load_page(page)
+# Routing logic
+if page == "Home":
+    show_home()
+elif page == "Prediction":
+    show_prediction()
+elif page == "Contact Us":
+    show_contact()
+elif page == "About":
+    show_about()