Update app.py

app.py

@@ -2,81 +2,52 @@ import streamlit as st
 import numpy as np
 import pickle
 
+# Load the model
 model = pickle.load(open("life_expectancy_model.pkl", "rb"))
 
+# Set page config
 st.set_page_config(
     page_title="Life Expectancy Prediction",
     page_icon="📊",
     layout="centered",
 )
 
-# Define session state for navigation
-if 'page' not in st.session_state:
-    st.session_state.page = "home"
-
-def set_page(page):
-    st.session_state.page = page
-
+# Define pages
+PAGES = {
+    "Model Report": "model_report",
+    "Hands-on Model": "hands_on_model",
+    "Problem Statement": "problem_statement",
+    "Simple EDA": "simple_eda",
+    "Visualizations": "visualizations",
+    "Model Building": "model_building",
+    "Final Model": "final_model"
+}
+
+# Session state for navigation
+if "current_page" not in st.session_state:
+    st.session_state.current_page = "Model Report"
+
+# Function to switch pages
+def switch_page(page):
+    st.session_state.current_page = page
+
+# Sidebar Navigation
 st.sidebar.title("Navigation")
-st.sidebar.button("Model Report", on_click=set_page, args=("report",))
-st.sidebar.button("Hands-on Model", on_click=set_page, args=("model",))
-
-if st.session_state.page == "home":
-    st.markdown("""
-        <style>
-            .stApp {
-                background-color: #E3F2FD;  /* Soft blue background */
-            }
-            .title {
-                text-align: center;
-                font-size: 32px;
-                font-weight: bold;
-                color: #2C3E50;
-            }
-            .stButton > button {
-                width: 100%;
-                background-color: #1E88E5;
-                color: white;
-                font-size: 16px;
-                border-radius: 6px;
-            }
-            .result-box {
-                text-align: center;
-                font-size: 22px;
-                font-weight: bold;
-                color: white;
-                padding: 15px;
-                border-radius: 8px;
-                margin-top: 20px;
-                background-color: #388E3C; /* Green */
-            }
-            .section-title {
-                font-size: 20px;
-                font-weight: bold;
-                color: #1E88E5;
-                margin-top: 20px;
-            }
-        </style>
-    """, unsafe_allow_html=True)
-
-    st.markdown("<h1 class='title'>Life Expectancy Prediction</h1>", unsafe_allow_html=True)
-
-    st.subheader("Select a Step:")
-    if st.button("Problem Statement"):
-        set_page("problem")
-    if st.button("Simple EDA"):
-        set_page("eda")
-    if st.button("Visualizations"):
-        set_page("visuals")
-    if st.button("Model Building"):
-        set_page("model_building")
-    if st.button("Final Model"):
-        set_page("final_model")
-
-elif st.session_state.page == "model":
-    st.subheader("Hands-on Model")
+for page_name in PAGES:
+    if st.sidebar.button(page_name):
+        switch_page(page_name)
+
+# Main Page Display
+if st.session_state.current_page == "Model Report":
+    st.title("📊 Model Report")
+    st.write("Welcome to the Life Expectancy Prediction App!")
+    st.write("Navigate using the buttons on the sidebar.")
+    
+elif st.session_state.current_page == "Hands-on Model":
+    st.title("🔍 Hands-on Model")
+    st.write("Interact with the model by providing input features.")
+    
     col1, col2 = st.columns(2)
-
     with col1:
         year = st.slider("Year", 2000, 2015, 2008)
         status = st.radio("Status", ["Developing", "Developed"], horizontal=True)
@@ -89,7 +60,7 @@ elif st.session_state.page == "model":
         measles = st.slider("Measles Cases", 0, 212183, 2419)
         bmi = st.slider("BMI", 1.0, 87.3, 38.3)
         polio = st.slider("Polio Immunization (%)", 3, 99, 82)
-
+    
     with col2:
         under_five_deaths = st.slider("Under-Five Deaths", 0, 2500, 4)
         total_expenditure = st.slider("Total Healthcare Expenditure (%)", 0.37, 17.6, 5.92)
@@ -107,41 +78,35 @@ elif st.session_state.page == "model":
                               hepatitis_b, measles, bmi, under_five_deaths, polio, total_expenditure,
                               diphtheria, hiv_aids, gdp, population, thinness_1_19, thinness_5_9,
                               income_composition, schooling]])
+        
         prediction = model.predict(features)[0]
 
-        st.markdown(
-            f"""
-            <div class="result-box">
-                Predicted Life Expectancy: <b>{prediction:.2f} years</b>
-            </div>
-            """,
-            unsafe_allow_html=True,
-        )
-
-elif st.session_state.page == "report":
-    st.subheader("Model Report")
-    st.write("Detailed model performance metrics will be shown here.")
-
-elif st.session_state.page == "problem":
-    st.subheader("Problem Statement")
-    st.write("Details about the problem statement will go here.")
-
-elif st.session_state.page == "eda":
-    st.subheader("Simple EDA")
-    st.write("Exploratory Data Analysis details will be here.")
-
-elif st.session_state.page == "visuals":
-    st.subheader("Visualizations")
-    st.write("Graphs and charts will be displayed here.")
-
-elif st.session_state.page == "model_building":
-    st.subheader("Model Building")
-    st.write("Steps for building the model will be described here.")
-
-elif st.session_state.page == "final_model":
-    st.subheader("Final Model")
-    st.write("Details about the final model will be displayed here.")
+        st.success(f"Predicted Life Expectancy: {prediction:.2f} years")
 
+elif st.session_state.current_page == "Problem Statement":
+    st.title("📝 Problem Statement")
+    st.write("Define the problem related to life expectancy prediction.")
+    
+elif st.session_state.current_page == "Simple EDA":
+    st.title("📊 Simple EDA")
+    st.write("Explore the dataset with some basic exploratory data analysis.")
+
+elif st.session_state.current_page == "Visualizations":
+    st.title("📈 Visualizations")
+    st.write("Check the visualizations that help understand the data better.")
+
+elif st.session_state.current_page == "Model Building":
+    st.title("🛠 Model Building")
+    st.write("Details on how the model was built and trained.")
+
+elif st.session_state.current_page == "Final Model":
+    st.title("✅ Final Model")
+    st.write("Showcase the final selected model and its performance.")
+
+# Go Back Button
+if st.session_state.current_page != "Model Report":
+    if st.button("🔙 Go Back to Home"):
+        switch_page("Model Report")
 
 
 # import streamlit as st