Upload app.py

app.py

@@ -42,6 +42,7 @@ default_values = {
     'max_rt_val': 0.0,
     'new_absorption': {},
     'fig1': None,
+    'fig2': None,  # Ensure fig2 is initialized
     'df_sound_pressure': None,
     'noise_reduction': None,
     'df_current_absorption': None,
@@ -52,6 +53,7 @@ for key, value in default_values.items():
     if key not in st.session_state:
         st.session_state[key] = value
 
+
 # Helper functions
 def calculate_absorption_area(volume, reverberation_time):
     return 0.161 * volume / reverberation_time if reverberation_time else float('inf')
@@ -191,6 +193,8 @@ def display_instructions():
         '4000': [48, 49]
     })
 
+    st.write('## Instructions')
+
     instructions = """
     
     ## Introduction
@@ -331,8 +335,11 @@ def display_instructions():
     st.info("After reading the instructions, get your input data ready and proceed to the next step.")
 
 def display_initial_data_entry():
-    st.write('''The primary objective here is to provide all the initial input data needed to start RT60 analysis.
+    st.write('''## Initial Data Entry
+    
+    The primary objective here is to provide all the initial input data needed to start RT60 analysis.
     See the 'Instructions' on the left for formatting information''')
+
     st.write("#### Upload current reverberation times")
     current_rt_file = st.file_uploader("Upload CSV or Excel file for current reverberation times", type=['csv', 'xlsx'],
                                        help='Go to Instructions on the sidebar for formatting information.')
@@ -382,7 +389,9 @@ def display_initial_data_entry():
         st.warning("Please ensure all input data has been provided correctly before proceeding.")
 
 def display_initial_rt60_compliance_check():
-    st.write('''The aim of initial compliance check is to compare the current RT60 values to the 
+    st.write('''## Initial RT60 Compliance Check 
+    
+    The aim of initial compliance check is to compare the current RT60 values to the 
      standard RT60 range. For the purpose of this tool, compliance is achieved if RT60 values for all frequencies are
      within the standard range. The graph at the bottom of this page would assist you in verifying compliance.
      If at least one frequency is not in the range, you would need to apply some acoustic treatment. See the 
@@ -460,7 +469,9 @@ def display_initial_rt60_compliance_check():
         st.warning("Please provide the current reverberation times and existing materials data before proceeding.")
 
 def display_desired_rt60():
-    st.write('''At this point in the analysis, you would need to define your desired RT60, which would be used to 
+    st.write('''## Desired RT60
+    
+    At this point in the analysis, you would need to define your desired RT60, which would be used to 
     calculate the desired sound absorption needed to achieve the desired RT60. Compare total frequency values 
     in the table below to the calculated current room sound absorption on the Initial RT60 Compliance Check tab 
     to inform your selection of new materials in the next step of the analysis''')
@@ -499,7 +510,9 @@ def display_desired_rt60():
         st.warning("Please complete the previous steps before proceeding to set the desired RT60 values.")
 
 def display_acoustic_treatment():
-    st.write("""The desired sound absorption calculated on the Desired RT60 tab is the target sound absorption
+    st.write("""## Acoustic Treatment
+    
+    The desired sound absorption calculated on the Desired RT60 tab is the target sound absorption
     you are aiming to achieve. You now have to start working towards achieving the target absorptions. Based on the 
     differences between the desired sound absorptions (see Desired RT60 tab) and the calculated current room 
     sound absorptions (see Initial RT60 Compliance Check tab), you would need to introduce new materials to either 
@@ -510,6 +523,7 @@ def display_acoustic_treatment():
     concrete floor) you are allowed to make. For a retrofit project, you can change non-structural elements
     like carpet and ceiling tiles. Additionally, you can introduce new materials like curtains.     
     """)
+
     st.write("#### Upload new materials data")
     new_materials_file = st.file_uploader("Upload CSV or Excel file for new materials", type=['csv', 'xlsx'],
                                           help='Go to Instructions on the sidebar for formatting information.')
@@ -576,11 +590,13 @@ def display_acoustic_treatment():
         st.warning("Please upload new materials data before proceeding.")
 
 def display_final_rt60_compliance_check():
-    st.write("""
+    st.write("""## Final RT60 Compliance Check
+    
     This is the last step in the RT60 analysis. To pass the final compliance check, the new RT60 for each 
     frequency must be within the standard RT60 range. If compliance fails, you need to go back to the Acoustic Treatment
     stage to revise your materials. Update your new materials spreadsheet and upload on the Acoustic Treatment tab. 
     """)
+
     if st.session_state.new_absorption and st.session_state.frequencies:
         min_rt = {freq: st.session_state.min_rt_val for freq in st.session_state.frequencies}
         max_rt = {freq: st.session_state.max_rt_val for freq in st.session_state.frequencies}
@@ -614,14 +630,18 @@ def display_final_rt60_compliance_check():
         fig2.update_layout(title='Current RT60 and New RT60 vs. Standard Range', xaxis_title='Frequency (Hz)',
                            yaxis_title='Reverberation Time (s)')
         st.plotly_chart(fig2)
+        st.session_state.fig2 = fig2  # Save fig2 in session state
 
         st.success(
             "After reflecting on the results, decide what to do next based on the expected outcome specified in your project brief.")
     else:
         st.warning("Please complete the previous steps before proceeding to the final RT60 compliance check.")
 
+
 def display_intelligibility_noise_reduction():
-    st.write('''On this tab, you will calculate noise reduction using existing and new materials sound absorption
+    st.write('''## Intelligibility Noise Reduction
+    
+    On this tab, you will calculate noise reduction using existing and new materials sound absorption
     coefficients and surface areas that were provided for reverberation time analysis. The relevant data 
     have copied to this tab for ease of reference. Note that you can update the new materials data by adding new rows
     or deleting existing rows. Your can also directly edit the current new materials data information. All updates
@@ -707,10 +727,11 @@ def display_intelligibility_noise_reduction():
             st.session_state.noise_reduction = noise_reduction
             st.write(f"##### Noise Reduction: {noise_reduction:.2f} dB")
             st.info('''The above noise reduction value is calculated total sound absorption for existing and new 
-            material above. The noise reduction formula is shown below:\n 
+            material above. The noise reduction formula is shown below:
             NR = 10 * Log10(new materials total absorption/existing materials total absorption)
             
             The above formula is the same as 10 * Log10(S alpha after/S alpha before''')
+
         else:
             st.error("Current total absorption is zero, which is not possible. Please check your data.")
 
@@ -718,7 +739,7 @@ def display_intelligibility_noise_reduction():
         st.write("#### Updated Sound Pressure Levels")
         st.info('''The values in the table below have been updated to account for the calculated noise reduction.
         Note that these are logarithm deductions given that sound pressure level is in decibels. Plot the 
-        updated sound pressure levels on your speech intelligibility Dot chart''')
+        updated sound pressure levels on your speech intelligibility Dot chart.''')
         updated_df = st.session_state.df_sound_pressure.copy()
 
         # Ensure all values are numeric and fill NaN with 0
@@ -734,12 +755,12 @@ def display_intelligibility_noise_reduction():
 
     # PDF Download Button
     st.write("""
-    **IMPORTANT NOTE ABOUT PDF REPORT:** The purpose of the PDF report is to provide you detailed record of all
-    your input data and the graphs generated. Graphs in PDF report are of low resolutions. Hence, remember to download 
-    individual graphs and tables to include in your report. Hover your mouse over a graph and several icons would
-    appear at the top right corner of the graph, click the first icon to download the graph. Repeat the same process 
-    to download tables.
-    """)
+        **IMPORTANT NOTE ABOUT PDF REPORT:** The purpose of the PDF report is to provide you detailed record of all
+        your input data and the graphs generated. Graphs in PDF report are of low resolutions. Hence, remember to download 
+        individual graphs and tables to include in your report. Hover your mouse over a graph and several icons would
+        appear at the top right corner of the graph, click the first icon to download the graph. Repeat the same process
+        to download tables.
+        """)
     if st.button("Download PDF Report"):
         pdf = generate_pdf(
             st.session_state.volume,