Update app.py

app.py

@@ -101,6 +101,17 @@ if st.session_state.current_page == "Model Report":
 # Individual Sections
 elif st.session_state.current_page == "Problem Statement":
     st.markdown("<h1 class='title'>Problem Statement</h1>", unsafe_allow_html=True)
+
+    st.markdown("""
+        <p class='content'>
+            The goal of this project is to build a predictive model that estimates the <b>Life Expectancy</b> of a country 
+            based on multiple influencing factors such as health indicators, economic conditions, and social parameters.
+        </p>
+    """, unsafe_allow_html=True)
+
+    st.image("images/problem_statement.png", 
+         caption="Life Expectancy Prediction Overview", 
+         use_container_width=True)
     
     st.markdown("<h2 class='subtitle'>Predicting Life Expectancy</h2>", unsafe_allow_html=True)
     st.markdown("""
@@ -112,13 +123,6 @@ elif st.session_state.current_page == "Problem Statement":
         </p>
     """, unsafe_allow_html=True)
 
-    st.markdown("""
-        <h2 class='subtitle'>Problem Statement</h2>
-        <p class='content'>
-            The goal of this project is to build a predictive model that estimates the <b>Life Expectancy</b> of a country 
-            based on multiple influencing factors such as health indicators, economic conditions, and social parameters.
-        </p>
-    """, unsafe_allow_html=True)
 
     if st.button("🔙 Go Back to Model Report"):
         switch_page("Model Report")