import gradio as gr import numpy as np import io from scipy.io import wavfile import speech_recognition as sr import soundfile as sf # Morse code dictionary (ITU standard) MORSE_CODE_DICT = { 'A': '.-', 'B': '-...', 'C': '-.-.', 'D': '-..', 'E': '.', 'F': '..-.', 'G': '--.', 'H': '....', 'I': '..', 'J': '.---', 'K': '-.-', 'L': '.-..', 'M': '--', 'N': '-.', 'O': '---', 'P': '.--.', 'Q': '--.-', 'R': '.-.', 'S': '...', 'T': '-', 'U': '..-', 'V': '...-', 'W': '.--', 'X': '-..-', 'Y': '-.--', 'Z': '--..', '1': '.----', '2': '..---', '3': '...--', '4': '....-', '5': '.....', '6': '-....', '7': '--...', '8': '---..', '9': '----.', '0': '-----', ' ': '/' } MORSE_TO_CHAR = {v: k for k, v in MORSE_CODE_DICT.items()} # Morse code timing (in seconds) DIT_DURATION = 0.1 # 100ms for dit DAH_DURATION = 3 * DIT_DURATION # 300ms for dah SPACE_DURATION = 7 * DIT_DURATION # 700ms for word space CHAR_SPACE = DIT_DURATION # Space between characters SAMPLE_RATE = 44100 # Standard audio sample rate # Decode Morse from audio (simplified) def decode_morse_from_audio(audio_data): if audio_data is None: return "", "" sample_rate, data = audio_data if len(data.shape) > 1: data = data.mean(axis=1) data = data / np.max(np.abs(data)) threshold = 0.1 signal = data > threshold morse_code, decoded_text = "", "" i = 0 while i < len(signal) - int(SAMPLE_RATE * DIT_DURATION): if signal[i]: start = i while i < len(signal) and signal[i]: i += 1 duration = (i - start) / sample_rate morse_code += "-" if duration >= DAH_DURATION else "." else: start = i while i < len(signal) and not signal[i]: i += 1 pause = (i - start) / sample_rate if pause >= SPACE_DURATION and morse_code: decoded_text += " " morse_code = "" elif pause >= DIT_DURATION and morse_code: decoded_text += MORSE_TO_CHAR.get(morse_code, "?") morse_code = "" i += 1 if morse_code: decoded_text += MORSE_TO_CHAR.get(morse_code, "?") return morse_code, decoded_text.strip() # Convert text to Morse code def text_to_morse(text): text = text.upper() morse = " ".join(MORSE_CODE_DICT.get(char, "?") for char in text if char in MORSE_CODE_DICT) return morse # Generate Morse code audio from text def generate_morse_audio(morse): audio = [] frequency = 750 # Hz for Morse tone for symbol in morse.split(): if symbol == "/": audio.extend([0] * int(SAMPLE_RATE * SPACE_DURATION)) else: for char in symbol: duration = DAH_DURATION if char == "-" else DIT_DURATION t = np.linspace(0, duration, int(SAMPLE_RATE * duration), False) tone = 0.5 * np.sin(2 * np.pi * frequency * t) audio.extend(tone) audio.extend([0] * int(SAMPLE_RATE * DIT_DURATION)) # Space between dits/dahs audio.extend([0] * int(SAMPLE_RATE * CHAR_SPACE)) # Space between characters audio = np.array(audio, dtype=np.float32) buffer = io.BytesIO() sf.write(buffer, audio, SAMPLE_RATE, format="wav") buffer.seek(0) return buffer # Speech to text def speech_to_text(audio): recognizer = sr.Recognizer() with sr.AudioFile(audio) as source: audio_data = recognizer.record(source) try: return recognizer.recognize_google(audio_data) except sr.UnknownValueError: return "Could not understand audio" except sr.RequestError: return "Speech recognition service unavailable" # Highlight alphabet in UI def generate_alphabet_html(decoded_text): html = "
" for char in MORSE_CODE_DICT.keys(): color = "red" if char in decoded_text.upper() else "black" html += f"{char}: {MORSE_CODE_DICT[char]}" if char in "AEIMQUZ": html += "
" html += "
" return html # Combined processing function def process_input(text=None, speech=None, audio=None): morse, decoded_text, audio_output = "", "", None if text: # Text input morse = text_to_morse(text) decoded_text = text audio_output = generate_morse_audio(morse) elif speech: # Speech input text = speech_to_text(speech) morse = text_to_morse(text) decoded_text = text audio_output = generate_morse_audio(morse) elif audio: # Live audio input morse, decoded_text = decode_morse_from_audio(audio) alphabet_html = generate_alphabet_html(decoded_text) return morse, decoded_text, alphabet_html, audio_output # Gradio UI with Blocks with gr.Blocks(title="Morse Code Decoder & Generator") as demo: gr.Markdown("# Morse Code Decoder & Generator") gr.Markdown("Decode live Morse audio, or generate Morse from text/speech!") with gr.Tab("Decode Live Audio"): audio_input = gr.Audio(source="microphone", type="numpy", streaming=True, label="Live Audio Input") with gr.Row(): with gr.Column(): morse_output = gr.Textbox(label="Detected Morse Code", interactive=False) text_output = gr.Textbox(label="Decoded Text", interactive=False) alphabet_display = gr.HTML(label="Morse Alphabet (Highlighted)") audio_input.stream( fn=process_input, inputs=[audio_input], outputs=[morse_output, text_output, alphabet_display, gr.Audio(visible=False)], ) with gr.Tab("Generate from Text"): text_input = gr.Textbox(label="Enter Text", placeholder="Type here...") generate_btn = gr.Button("Generate Morse") with gr.Row(): with gr.Column(): morse_gen_output = gr.Textbox(label="Morse Code", interactive=False) text_gen_output = gr.Textbox(label="Original Text", interactive=False) gen_alphabet_display = gr.HTML(label="Morse Alphabet (Highlighted)") audio_playback = gr.Audio(label="Morse Audio Playback", interactive=False) generate_btn.click( fn=process_input, inputs=[text_input], outputs=[morse_gen_output, text_gen_output, gen_alphabet_display, audio_playback] ) with gr.Tab("Generate from Speech"): speech_input = gr.Audio(source="microphone", type="filepath", label="Speak Your Text") speech_btn = gr.Button("Convert Speech to Morse") with gr.Row(): with gr.Column(): morse_speech_output = gr.Textbox(label="Morse Code", interactive=False) text_speech_output = gr.Textbox(label="Transcribed Text", interactive=False) speech_alphabet_display = gr.HTML(label="Morse Alphabet (Highlighted)") speech_audio_playback = gr.Audio(label="Morse Audio Playback", interactive=False) speech_btn.click( fn=process_input, inputs=[speech_input], outputs=[morse_speech_output, text_speech_output, speech_alphabet_display, speech_audio_playback] ) # Launch the app demo.launch()