Update app.py

app.py

@@ -582,15 +582,16 @@ if st.session_state.app_step == "dev":
         """, unsafe_allow_html=True)
 
         # 2-column layout, big gap (prevents overlap)
-        col_cross, col_track = st.columns([3, 2], gap="large")
-        with col_cross:
-            st.pyplot(cross_plot_static(df[TARGET], df["UCS_Pred"]), use_container_width=False)
+        col_track, col_cross = st.columns([2, 3], gap="large")
         with col_track:
             st.plotly_chart(
                 track_plot(df, include_actual=True),
                 use_container_width=False, # Set to False to honor the width in track_plot()
                 config={"displayModeBar": False, "scrollZoom": True}
             )
+        with col_cross:
+            st.pyplot(cross_plot_static(df[TARGET], df["UCS_Pred"]), use_container_width=False)
+
 
     if "Train" in st.session_state.results or "Test" in st.session_state.results:
         tab1, tab2 = st.tabs(["Training", "Testing"])
@@ -612,7 +613,7 @@ if st.session_state.app_step == "validate":
             st.sidebar.caption(f"**Data loaded:** {up.name} • {df0.shape[0]} rows × {df0.shape[1]} cols")
     if st.sidebar.button("Preview data", use_container_width=True, disabled=(up is None)):
         st.session_state.show_preview_modal = True # Set state to show modal
-    go_btn = st.sidebar.button("Predict", type="primary", use_container_width=True)
+    go_btn = st.sidebar.button("Predict & Validate", type="primary", use_container_width=True)
     if st.sidebar.button("⬅ Back to Case Building", use_container_width=True): st.session_state.app_step="dev"; st.rerun()
     if st.sidebar.button("Proceed to Prediction ▶", use_container_width=True): st.session_state.app_step="predict"; st.rerun()
 
@@ -658,20 +659,20 @@ if st.session_state.app_step == "validate":
                 <strong>MAE:</strong> Mean Absolute Error
             </div>
         """, unsafe_allow_html=True)
-       
-        col_cross, col_track = st.columns([3, 2], gap="large")
-        with col_cross:
-            st.pyplot(
-                cross_plot_static(st.session_state.results["Validate"][TARGET],
-                                 st.session_state.results["Validate"]["UCS_Pred"]),
-                use_container_width=False
-            )
+        
+        col_track, col_cross = st.columns([2, 3], gap="large")
         with col_track:
             st.plotly_chart(
                 track_plot(st.session_state.results["Validate"], include_actual=True),
                 use_container_width=False, # Set to False to honor the width in track_plot()
                 config={"displayModeBar": False, "scrollZoom": True}
             )
+        with col_cross:
+            st.pyplot(
+                cross_plot_static(st.session_state.results["Validate"][TARGET],
+                                  st.session_state.results["Validate"]["UCS_Pred"]),
+                use_container_width=False
+            )
 
         sv = st.session_state.results["sv_val"]
         if sv["oor"] > 0: st.markdown('<div class="st-message-box st-warning">Some inputs fall outside **training min–max** ranges.</div>', unsafe_allow_html=True)