main.py

import os
import speech_recognition as sr
from pydub import AudioSegment

import logging
# Configure logging
logging.basicConfig(filename='assets/main.log', 
                    level=logging.DEBUG,
                    format='%(asctime)s - %(levelname)s - %(message)s')

r = sr.Recognizer()

def record_text():
    try:
        # Check if a microphone is available
        with sr.Microphone() as src:
            
            # Adjusts for ambient noise
            r.adjust_for_ambient_noise(source=src, duration=0.2)
            print('Listening...')

            # Captures the audio
            audio2 = r.listen(src)  
            
            # Recognize the audio using Google Speech Recognition
            text_data = r.recognize_google(audio2)
            
            # Convert the text to lowercase
            text_data = text_data.lower()

            # Save the audio
            audio_file_path = os.path.join('new_audios', f'{text_data}.wav')
            
            if not os.path.exists('audio'):
                os.makedirs('audio')
                
            with open(audio_file_path, 'wb') as f:
                f.write(audio2.get_wav_data())

            return text_data, audio_file_path

    except AssertionError as e:
        logging.error(f'Assertion error')
        print(f'Microphone access error: {e}')
    except sr.RequestError as e:
        logging.error(f'Request error')
        print(f'Request error: {e}')
    except sr.UnknownValueError:
        logging.error(f'Unknown value error')
        print('Unknown value error')

    return '', ''

def output_text(text, audio_path):
    # Only write non-empty text
    if text:  
        with open('words.txt', 'a') as f:
            f.write(f'{text} ({audio_path})')
            f.write('\n')
    return

# Main loop
try:
    while True:
        # Record speech and convert to text
        text, audio_path = record_text()
        
        # Exit the loop gracefully
        if text in ['exit', 'stop']:
            break  

        # Only output non-empty text
        if text:
            output_text(text, audio_path)
            print(f"Text: '{text}' and audio recorded at '{audio_path}' successfully written")

except KeyboardInterrupt:
    logging.error(f'Keyboard interruption')
    print('\nProgram interrupted. Exiting gracefully...')
finally:
    print('Program has stopped.')