Update app.py

app.py

@@ -21,48 +21,113 @@ sample_rate = 44100
 amplitude_scaling_factor = 10.0
 
 
-# -----------------Record----------------- #
+# -----------------Filter----------------- #
 
-def record(audio):
-    try:
-        sr, data = audio
-        wavio.write("recorded.wav", data, sr)
-        main()
-        return f"Audio receive correctly"
-    except Exception as e:
-        return f"Error: {e}"
+def butter_bandpass(sr, order=5):
+    """
+    This function designs a Butterworth bandpass filter.
 
+    Parameters:
+    sr (int): The sample rate of the audio.
+    order (int): The order of the filter.
 
-# -----------------Filter----------------- #
-
-def butter_bandpass(lowcut, highcut, sr, order=5):
+    Returns:
+    tuple: The filter coefficients `b` and `a`.
+    """
+    # Calculate the Nyquist frequency
     nyquist = 0.5 * sr
-    low = lowcut / nyquist
-    high = highcut / nyquist
-    coef = butter(order, [low, high], btype='band')
-    b = coef[0]
-    a = coef[1]
+
+    # Normalize the cutoff frequencies
+    low = low_frequency / nyquist
+    high = high_frequency / nyquist
+
+    # Design the Butterworth bandpass filter
+    coefficient = butter(order, [low, high], btype='band')
+
+    # Extract the filter coefficients
+    b = coefficient[0]
+    a = coefficient[1]
+
     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, sr, order=5):
+    """
+    This function applies the Butterworth bandpass filter to a given data.
+
+    Parameters:
+    data (array): The audio data to be filtered.
+    sr (int): The sample rate of the audio.
+    order (int): The order of the filter.
+
+    Returns:
+    array: The filtered audio data.
+    """
+    # Get the filter coefficients
+    b, a = butter_bandpass(sr, order=order)
+
+    # Apply the filter to the data
     y = lfilter(b, a, data)
+
     return y
 
 
-def main():
-    input_file = 'recorded.wav'
-    output_file = 'output_filtered_receiver.wav'
-    lowcut = 17500
-    highcut = 19500
+def filtered():
+    """
+    This function reads an audio file, applies the bandpass filter to the audio data,
+    and then writes the filtered data to an output file.
 
-    sr, data = read(input_file)
+    Returns:
+    str: A success message if the audio is filtered correctly, otherwise an error message.
+    """
+    try:
+        # Read the audio data from the input file
+        sr, data = read(input_file)
+
+        # Apply the bandpass filter to the audio data
+        filtered_data = butter_bandpass_filter(data, sr)
+
+        # Write the filtered data to the output file
+        write(output_file, sr, np.int16(filtered_data))
+
+        return "Filtered Audio Generated"
+    except Exception as e:
+        # If an error occurs, return an error message
+        return f"Error: {str(e)}"
+
+
+# -----------------Record----------------- #
+
+def record(audio):
+    """
+    This function records audio and writes it to a .wav file.
 
-    filtered_data = butter_bandpass_filter(data, lowcut, highcut, sr)
-    write(output_file, sr, np.int16(filtered_data))
-    return "Filtered Audio Generated"
+    Parameters:
+    audio (tuple): A tuple containing the sample rate and the audio data.
 
+    Returns:
+    str: A success message if the audio is recorded correctly, otherwise an error message.
+    """
+    try:
+        # Check if the audio tuple contains exactly two elements
+        if len(audio) != 2:
+            return f"Error: Expected a tuple with 2 elements, but got {len(audio)}"
+
+        # Unpack the sample rate and data from the audio tuple
+        sr, data = audio
+
+        # Write the audio data to a .wav file
+        wavio.write("recorded.wav", data, sr)
+
+        # Call the filtered function to apply the bandpass filter to the audio data
+        filtered()
+
+        # Return a success message
+        return f"Audio receive correctly"
+    except Exception as e:
+        # If an error occurs, return an error message
+        return f"Error: {str(e)}"
+        
 
 # -----------------Frame----------------- #