Update sections.py

sections.py

@@ -602,34 +602,79 @@ def section_final_checks() -> None:
         for f in FREQUENCIES
     }
 
+    # ── RT60 final check ───────────────────────────────────────────────────
+    # with t_rt:
+    #     if st.session_state.room_volume == 0 or not combined:
+    #         st.warning("⚠️ Provide volume + materials, then run Acoustic Treatment.")
+    #     else:
+    #         new_rt = {f: calc_abs_area(st.session_state.room_volume, a)
+    #                   for f, a in combined.items()}
+    #         st.markdown("New RT60 values after acoustic treatment")
+    #         st.dataframe(pd.DataFrame([new_rt], index=["New RT60"]),
+    #                      use_container_width=True)
+
+    #         y_cur = ([float(st.session_state.df_current_rt.iloc[0][f])
+    #                   for f in FREQUENCIES]
+    #                  if st.session_state.df_current_rt is not None else [None]*6)
+
+    #         fig = plot_rt_band(
+    #             y_cur,
+    #             [st.session_state.rt_min]*6,
+    #             [st.session_state.rt_max]*6,
+    #             "Final RT60 vs Standard Range",
+    #         )
+    #         fig.add_trace(go.Scatter(
+    #             x=FREQUENCIES,
+    #             y=[new_rt[f] for f in FREQUENCIES],
+    #             mode="lines+markers",
+    #             name="New",
+    #             marker_color="#d62728",
+    #         ))
+    #         st.plotly_chart(fig, use_container_width=True)
+    #         st.session_state.fig_rt_final = fig
+
+    # ── RT60 final check ───────────────────────────────────────────────────
     # ── RT60 final check ───────────────────────────────────────────────────
     with t_rt:
-        if st.session_state.room_volume == 0 or not combined:
-            st.warning("⚠️ Provide volume + materials, then run Acoustic Treatment.")
+        if st.session_state.room_volume == 0 or not st.session_state.new_absorption:
+            st.warning("⚠️ Provide room volume and at least one new material, then run Acoustic Treatment.")
         else:
-            new_rt = {f: calc_abs_area(st.session_state.room_volume, a)
-                      for f, a in combined.items()}
-            st.markdown("New RT60 values after acoustic treatment")
-            st.dataframe(pd.DataFrame([new_rt], index=["New RT60"]),
-                         use_container_width=True)
-
-            y_cur = ([float(st.session_state.df_current_rt.iloc[0][f])
-                      for f in FREQUENCIES]
-                     if st.session_state.df_current_rt is not None else [None]*6)
-
+            # --------------------------------------------------------------
+            # Use *only* the absorption contributed by the new materials
+            # (st.session_state.new_absorption) for Sabine RT60 calculation
+            # --------------------------------------------------------------
+            new_rt = {
+                f: (0.16 * st.session_state.room_volume / a) if a > 0 else float("inf")
+                for f, a in st.session_state.new_absorption.items()
+            }
+    
+            st.markdown("New RT60 values after acoustic treatment (based on new materials only)")
+            st.dataframe(
+                pd.DataFrame([new_rt], index=["New RT60"]), use_container_width=True
+            )
+    
+            # Plot current vs standard band vs new RT60
+            y_cur = (
+                [float(st.session_state.df_current_rt.iloc[0][f]) for f in FREQUENCIES]
+                if st.session_state.df_current_rt is not None
+                else [None] * 6
+            )
+    
             fig = plot_rt_band(
                 y_cur,
-                [st.session_state.rt_min]*6,
-                [st.session_state.rt_max]*6,
+                [st.session_state.rt_min] * 6,
+                [st.session_state.rt_max] * 6,
                 "Final RT60 vs Standard Range",
             )
-            fig.add_trace(go.Scatter(
-                x=FREQUENCIES,
-                y=[new_rt[f] for f in FREQUENCIES],
-                mode="lines+markers",
-                name="New",
-                marker_color="#d62728",
-            ))
+            fig.add_trace(
+                go.Scatter(
+                    x=FREQUENCIES,
+                    y=[new_rt[f] for f in FREQUENCIES],
+                    mode="lines+markers",
+                    name="New",
+                    marker_color="#d62728",
+                )
+            )
             st.plotly_chart(fig, use_container_width=True)
             st.session_state.fig_rt_final = fig