Update streamlit_simulation/app.py

streamlit_simulation/app.py

@@ -242,13 +242,30 @@ def render_simulation_view(timestamp, prediction, actual, progress, fig, paused=
         unsafe_allow_html=True
     )
 
-    with info_container:
-        st.markdown("---")
+def render_simulation_view(timestamp, prediction, actual, progress, fig, paused=False):
+    """Displays the simulation plot and metrics in the UI."""
+    title = "Actual vs. Prediction (Paused)" if paused else "Actual vs. Prediction"
+    plot_title.markdown(
+        f"<div style='text-align: center; font-size: 20pt; font-weight: bold; color: {TEXT_COLOR}; margin-bottom: -0.7rem; margin-top: 0rem;'>"
+        f"{title}</div>",
+        unsafe_allow_html=True
+    )
+    plot_container.pyplot(fig)
+
+    st.markdown("<div style='margin-bottom: 0.5rem;'></div>", unsafe_allow_html=True)
+    x_axis_label.markdown(
+        f"<div style='text-align: center; font-size: 14pt; color: {TEXT_COLOR}; margin-top: -0.5rem;'>"
+        f"Time</div>",
+        unsafe_allow_html=True
+    )
+
+    with info_container.container():
+        st.markdown("<div style='margin-top: 5rem;'></div>", unsafe_allow_html=True)
         st.markdown(
             f"<span style='font-size: 24px; font-weight: 600; color: {HEADER_COLOR} !important;'>Time: {timestamp}</span>",
             unsafe_allow_html=True
         )
-    
+
         st.metric("Prediction", f"{prediction:,.0f} MW" if prediction is not None else "–")
         st.metric("Actual", f"{actual:,.0f} MW" if actual is not None else "–")
         st.caption("Simulation Progress")