Update app.py

app.py

@@ -55,7 +55,7 @@ class AudioProcessor:
         reverb_room_size = min(max(avg_volume, 0.1), 0.5)
         return highpass_cutoff, compressor_threshold, reverb_room_size
     
-    def master_audio(self):
+    def master_audio(self, freq_low=200, freq_high=2000, reduction_db=-18):
         # Load the mixed and generated 24bit audio file
         print(f"Loading 24bit audio file: {self.mastered_24bit}")
         with AudioFile(self.mastered_24bit) as f:
@@ -81,8 +81,8 @@ class AudioProcessor:
         # Apply custom stereo widening
         processed_audio = self.stereo_widen(processed_audio)
 
-        # Reduce piano volume
-        processed_audio = self.reduce_piano_volume(processed_audio, sample_rate)
+        # Reduce piano volume with user-provided or default values
+        processed_audio = self.reduce_piano_volume(processed_audio, sample_rate, freq_low, freq_high, reduction_db)
 
         # Save the processed audio to a new file
         output_file = 'final_mastered_audio.wav'
@@ -131,10 +131,15 @@ if uploaded_file is not None:
     # Initialize the processor
     processor = AudioProcessor(target_file=audio_file_path, reference_file=reference_file)
 
+    # User input for reduce_piano_volume parameters
+    freq_low = st.number_input('Low Frequency Cut-off (Hz)', value=200)
+    freq_high = st.number_input('High Frequency Cut-off (Hz)', value=2000)
+    reduction_db = st.number_input('Volume Reduction (dB)', value=-18)
+
     if st.button('Master Audio'):
         with st.spinner('Mixing, generating, and mastering audio...'):
             processor.mix_and_generate()
-            final_output = processor.master_audio()
+            final_output = processor.master_audio(freq_low=freq_low, freq_high=freq_high, reduction_db=reduction_db)
             
             st.success('Audio mastering complete!')
             st.audio(final_output)