Update app.py

app.py

@@ -1,21 +1,34 @@
 import streamlit as st
-from home import show_home
-from prediction import show_prediction
-from contact import show_contact
-from about import show_about
+import home
+import prediction
+import about
+import contact
 
+# Set page config
 st.set_page_config(page_title="TransPolymer", layout="wide", page_icon="🧪")
 
-# Sidebar for navigation
+# Navigation menu
 st.sidebar.title("🔗 Navigation")
-page = st.sidebar.radio("Go to", ["Home", "Prediction", "Contact Us", "About"])
+page = st.sidebar.radio("Select a Page", ["Home", "Predictions", "About", "Contact"])
+
+# Handle URL query params (e.g., ?page=Contact)
+query_params = st.query_params
+if "page" in query_params:
+    page = query_params["page"][0].capitalize()  # Capitalize to match the radio button options
+
+# Page routing 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)
 
-# Routing logic
-if page == "Home":
-    show_home()
-elif page == "Prediction":
-    show_prediction()
-elif page == "Contact Us":
-    show_contact()
-elif page == "About":
-    show_about()