Update app.py

app.py

@@ -38,7 +38,7 @@ def signal_function(frequency, time):
     return np.sin(2 * np.pi * frequency * time)
 
 
-def generate_silence(duration, sample_rate=44100):
+def generate_silence(duration):
     return np.zeros(int(sample_rate * duration))
 
 
@@ -129,33 +129,29 @@ def encode_and_generate_audio(text):
 
 # -----------------Filter----------------- #
 
-def butter_bandpass(lowcut, highcut, fs, order=5):
-    nyquist = 0.5 * fs
-    low = lowcut / nyquist
-    high = highcut / nyquist
+def butter_bandpass(sr, order=5):
+    nyquist = 0.5 * sr
+    low = low_frequency / nyquist
+    high = high_frequency / nyquist
     coef = butter(order, [low, high], btype='band')
     b = coef[0]
     a = coef[1]
     return b, a
 
 
-def butter_bandpass_filter(data, lowcut, highcut, fs, order=5):
-    b, a = butter_bandpass(lowcut, highcut, fs, order=order)
+def butter_bandpass_filter(data, sr, order=5):
+    b, a = butter_bandpass(low_frequency, high_frequency, sr, order=order)
     y = lfilter(b, a, data)
     return y
 
 
 def main():
-    input_file = 'output_text.wav'
-    output_file = 'output_filtered_sender.wav'
-    lowcut = 18000
-    highcut = 19000
-
+    
     try:
-        fs, data = read(input_file)
+        sr, data = read(input_file)
 
-        filtered_data = butter_bandpass_filter(data, lowcut, highcut, fs)
-        write(output_file, fs, np.int16(filtered_data))
+        filtered_data = butter_bandpass_filter(data, low_frequency, high_frequency, sr)
+        write(output_file, sr, np.int16(filtered_data))
         return "Filtered Audio Generated"
     except Exception as e:
         return f"Error: {str(e)}"