Update app.py

app.py

@@ -1,40 +1,31 @@
 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
+
+# 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")
 
-# ✅ This MUST be first!
-st.set_page_config(page_title="TransPolymer", layout="wide", page_icon="🧪")
-
-# ✅ Then import your modules
-import home
-import prediction
-import about
-import contact
-
-# ✅ Navigation Sidebar
-st.sidebar.title("🔗 Navigation")
+# Navigation menu
+st.sidebar.title("Navigation")
 page = st.sidebar.radio("Select a Page", ["Home", "Predictions", "About", "Contact"])
 
-# ✅ Optional: Override from URL like ?page=Contact
-query_params = st.query_params
-if "page" in query_params:
-    param_page = query_params["page"][0].capitalize()
-    if param_page in ["Home", "Predictions", "About", "Contact"]:
-        page = param_page
-
-# ✅ Page rendering logic
-def load_page(page_name):
-    if page_name == "Home":
-        home.show()
-    elif page_name == "Predictions":
-        prediction.show()
-    elif page_name == "About":
-        about.show()
-    elif page_name == "Contact":
-        contact.show()
-    else:
-        st.error("Page not found.")
-
+# 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)