Update app.py

app.py

@@ -2,28 +2,55 @@ import streamlit as st
 import numpy as np
 import pickle
 
-# Load the model
+# Load Model
 model = pickle.load(open("life_expectancy_model.pkl", "rb"))
 
-# Set page config
+# Set Page Config
 st.set_page_config(
     page_title="Life Expectancy Prediction",
     page_icon="📊",
     layout="centered",
 )
 
-# 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
+# Styling
+st.markdown("""
+    <style>
+        .stApp {
+            background-color: #E3F2FD;  /* Light 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)
+
+# Navigation State
 if "current_page" not in st.session_state:
     st.session_state.current_page = "Model Report"
 
@@ -33,20 +60,47 @@ def switch_page(page):
 
 # Sidebar Navigation
 st.sidebar.title("Navigation")
-for page_name in PAGES:
-    if st.sidebar.button(page_name):
-        switch_page(page_name)
+if st.sidebar.button("📊 Model Report"):
+    switch_page("Model Report")
+if st.sidebar.button("🛠 Hands-on Model"):
+    switch_page("Hands-on Model")
 
-# Main Page Display
+# Model Report Page
 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.")
-    
+    st.markdown("<h1 class='title'>📊 Model Report</h1>", unsafe_allow_html=True)
+    st.write("Explore different phases of the Life Expectancy project.")
+
+    if st.button("📌 Problem Statement"):
+        switch_page("Problem Statement")
+    if st.button("📥 Data Collection"):
+        switch_page("Data Collection")
+    if st.button("📊 Simple EDA"):
+        switch_page("Simple EDA")
+    if st.button("🛠 Data Pre-processing"):
+        switch_page("Data Pre-processing")
+    if st.button("📈 EDA"):
+        switch_page("EDA")
+    if st.button("⚙ Model Building"):
+        switch_page("Model Building")
+    if st.button("✅ Final Model"):
+        switch_page("Final Model")
+
+# Individual Sections
+elif st.session_state.current_page in [
+    "Problem Statement", "Data Collection", "Simple EDA", 
+    "Data Pre-processing", "EDA", "Model Building", "Final Model"
+]:
+    st.title(st.session_state.current_page)
+    st.write(f"Details about {st.session_state.current_page}...")  # Add actual content here
+
+    if st.button("🔙 Back to Model Report"):
+        switch_page("Model Report")
+
+# Hands-on Model Page
 elif st.session_state.current_page == "Hands-on Model":
-    st.title("🔍 Hands-on Model")
-    st.write("Interact with the model by providing input features.")
-    
+    st.title("🛠 Hands-on Model")
+    st.write("Provide inputs to predict Life Expectancy.")
+
     col1, col2 = st.columns(2)
     with col1:
         year = st.slider("Year", 2000, 2015, 2008)
@@ -60,7 +114,7 @@ elif st.session_state.current_page == "Hands-on 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)
@@ -81,34 +135,20 @@ elif st.session_state.current_page == "Hands-on Model":
         
         prediction = model.predict(features)[0]
 
-        st.success(f"Predicted Life Expectancy: {prediction:.2f} years")
+        st.markdown(
+            f"""
+            <div class="result-box">
+                Predicted Life Expectancy: <b>{prediction:.2f} years</b>
+            </div>
+            """,
+            unsafe_allow_html=True,
+        )
 
-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"):
+    if st.button("🔙 Back to Model Report"):
         switch_page("Model Report")
 
 
+
 # import streamlit as st
 # import numpy as np
 # import pickle