Upload 3 files

app.py

@@ -0,0 +1,171 @@
+import numpy as np
+from scipy.io.wavfile import write, read
+from tqdm import tqdm
+import sounddevice as sd
+import gradio as gr
+from scipy.signal import butter, lfilter
+import reedsolo
+import random
+from IPython.display import Audio, display
+import os
+
+input_file = 'input_text.wav'
+output_file = 'output_filtered_sender.wav'
+
+low_frequency = 18000
+high_frequency = 19000
+bit_duration = 0.007
+sample_rate = 44100
+amplitude_scaling_factor = 15.0
+
+def delete_file(file_path):
+    try:
+        os.remove(file_path)
+        print(f"File '{file_path}' deleted successfully.")
+    except OSError as e:
+        print(f"Error deleting file '{file_path}': {e}")
+
+#-----------------Sender-----------------#
+
+def text_to_binary(text):
+    binary_string = ''.join(format(ord(char), '08b') for char in text)
+    return binary_string
+
+def signal_function(frequency, time):
+    return np.sin(2 * np.pi * frequency * time)
+
+def generate_silence(duration, sample_rate=44100):
+    return np.zeros(int(sample_rate * duration))
+
+def binary_signal(binary_string):
+    t = np.linspace(0, bit_duration, int(sample_rate * bit_duration), False)
+    signal = []
+
+    for bit in tqdm(binary_string, desc="Generating Signal"):
+        if bit == '0':
+            signal.append(amplitude_scaling_factor * np.sign(signal_function(low_frequency, t)))
+        else:
+            signal.append(amplitude_scaling_factor * np.sign(signal_function(high_frequency, t)))
+
+    return np.concatenate(signal)
+
+def flag_encoding(bit_value):
+    flag_duration = 6 * 0.0014
+    t_flag = np.linspace(0, flag_duration, int(sample_rate * flag_duration), False)
+    signal = []
+
+    if bit_value == 0:
+        binary_flag = "100001"
+        for bit in binary_flag:
+            if bit == '0':
+                signal.append(amplitude_scaling_factor * np.sign(signal_function(low_frequency, t_flag)))
+            else:
+                signal.append(amplitude_scaling_factor * np.sign(signal_function(high_frequency, t_flag)))
+
+        return np.concatenate(signal)
+    else:
+        binary_flag = "011110"
+        for bit in tqdm(binary_flag, desc="Generating Signal"):
+            if bit == '0':
+                signal.append(amplitude_scaling_factor * np.sign(signal_function(low_frequency, t_flag)))
+            else:
+                signal.append(amplitude_scaling_factor * np.sign(signal_function(high_frequency, t_flag)))
+
+        return np.concatenate(signal)
+
+def encode_rs(binary_string, ecc_bytes):
+    byte_data = bytearray(int(binary_string[i:i + 8], 2) for i in range(0, len(binary_string), 8))
+    rs = reedsolo.RSCodec(ecc_bytes)
+    encoded_data = rs.encode(byte_data)
+    encoded_binary_string = ''.join(format(byte, '08b') for byte in encoded_data)
+    return encoded_binary_string
+
+def manchester_encoding(binary_string):
+    encode_binary_string = encode_rs(binary_string, 20)
+
+    t = np.linspace(0, bit_duration, int(sample_rate * bit_duration), False)
+    signal = []
+
+    for bit in tqdm(encode_binary_string, desc="Generating Signal"):
+        if bit == '0':
+            signal.append(amplitude_scaling_factor * np.sign(signal_function(low_frequency, t)))
+            signal.append(amplitude_scaling_factor * np.sign(signal_function(high_frequency, t)))
+        else:
+            signal.append(amplitude_scaling_factor * np.sign(signal_function(high_frequency, t)))
+            signal.append(amplitude_scaling_factor * np.sign(signal_function(low_frequency, t)))
+
+    return np.concatenate(signal)
+
+def binary_to_signal(binary_string):
+    flag_start = flag_encoding(0)
+    flag_end = flag_encoding(1)
+    silence_duration = 0.1
+    silence_before = generate_silence(silence_duration)
+    silence_after = generate_silence(silence_duration)
+
+    signal = np.concatenate([silence_before, flag_start, manchester_encoding(binary_string), flag_end, silence_after])
+
+    return signal
+
+def encode_and_generate_audio(text):
+    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_text.wav', 44100, signal.astype(np.int16))
+    main()
+    return "WAV file generated and ready to be sent."
+
+#-----------------Filter-----------------#
+
+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')
+    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)
+    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)
+
+        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)}"
+
+
+#-----------------Player-----------------#
+
+wav_file_path = "output_filtered_sender.wav"
+
+def play_sound():
+    # You can replace 'your_sound_file.mp3' with the path to your sound file
+    return gr.Audio(wav_file_path, autoplay=True)
+
+
+#-----------------Interface-----------------#
+
+# Define the Gradio interface 
+with gr.Blocks() as demo:
+    name = gr.Textbox(label="Your Text")
+    output = gr.Textbox(label="Output")
+    submit = gr.Button("Generate Audio")
+    submit.click(fn=encode_and_generate_audio, inputs=name, outputs=output)
+
+    gr.Interface(fn=play_sound, inputs=[], outputs=gr.Audio(), live=False)
+
+demo.launch()

packages.txt

@@ -0,0 +1 @@
+libportaudio2

requirements.txt

@@ -0,0 +1,7 @@
+numpy~=1.24.3
+sounddevice~=0.4.6
+gradio~=4.8.0
+reedsolo~=1.7.0
+scipy~=1.11.4
+tqdm~=4.66.1
+ipython~=8.12.2