Update app.py

app.py

@@ -18,8 +18,7 @@ sample_rate = 44100
 amplitude_scaling_factor = 15.0
 
 
-# ----------------Useless----------------  #
-
+# ----------------Useless---------------- #
 def delete_file(file_path):
     try:
         os.remove(file_path)
@@ -39,7 +38,7 @@ def signal_function(frequency, time):
     return np.sin(2 * np.pi * frequency * time)
 
 
-def generate_silence(duration):
+def generate_silence(duration, sample_rate=44100):
     return np.zeros(int(sample_rate * duration))
 
 
@@ -119,19 +118,19 @@ def binary_to_signal(binary_string):
 
 
 def encode_and_generate_audio(text):
-    #delete_file(input_file)
-    #delete_file(output_file)
+    # delete_file("output_text.wav")
+    # delete_file("output_filtered.wav")
     binary_string_to_send = text_to_binary(text)
     signal = binary_to_signal(binary_string_to_send)
-    write(output_file, 44100, signal.astype(np.int16))
+    write('output_text.wav', 44100, signal.astype(np.int16))
     main()
     return "WAV file generated and ready to be sent."
 
 
 # -----------------Filter----------------- #
 
-def butter_bandpass(lowcut, highcut, sr, order=5):
-    nyquist = 0.5 * sr
+def butter_bandpass(lowcut, highcut, fs, order=5):
+    nyquist = 0.5 * fs
     low = lowcut / nyquist
     high = highcut / nyquist
     coef = butter(order, [low, high], btype='band')
@@ -140,21 +139,23 @@ def butter_bandpass(lowcut, highcut, sr, order=5):
     return b, a
 
 
-def butter_bandpass_filter(data, lowcut, highcut, sr, order=5):
-    b, a = butter_bandpass(lowcut, highcut, sr, order=order)
+def butter_bandpass_filter(data, lowcut, highcut, fs, order=5):
+    b, a = butter_bandpass(lowcut, highcut, fs, 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:
-        sr, data = read(input_file)
+        fs, data = read(input_file)
 
-        filtered_data = butter_bandpass_filter(data, lowcut, highcut, sr)
-        write(output_file, sr, np.int16(filtered_data))
+        filtered_data = butter_bandpass_filter(data, lowcut, highcut, fs)
+        write(output_file, fs, np.int16(filtered_data))
         return "Filtered Audio Generated"
     except Exception as e:
         return f"Error: {str(e)}"
@@ -167,9 +168,9 @@ def play_sound():
     return gr.Audio(output_file, autoplay=True)
 
 
-# -----------------Interface-----------------#
+# -----------------Interface----------------- #
 
-# Define the Gradio interface 
+# Define the Gradio interface
 with gr.Blocks() as demo:
     name = gr.Textbox(label="Your Text")
     output = gr.Textbox(label="Output")