Update app.py

app.py

@@ -1,31 +1,37 @@
 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")
+# ✅ Must be the first command
+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"])
+# ✅ Import your page modules
+import home
+import prediction
+import about
+import contact
 
-# Call the function to display the selected page
-load_page(page)
+# ✅ Navigation Sidebar
+st.sidebar.title("🔗 Navigation")
+page = st.sidebar.radio("Select a Page", ["Home", "Predictions", "About", "Contact"])
 
-# Handle page URL updates
+# ✅ Optional: Override via URL (?page=Contact)
 query_params = st.query_params
 if "page" in query_params:
-    page = query_params["page"][0]  # Get page value from URL parameters
-    load_page(page)
+    param_page = query_params["page"][0].capitalize()
+    if param_page in ["Home", "Predictions", "About", "Contact"]:
+        page = param_page
+
+# ✅ Page loading 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.")
+
+# ✅ Load the selected page
+load_page(page)