vox_in_out.py

from gtts import gTTS
from insult_response import get_insult
from loader_recorder import set_command, get_response, get_command_list
from os import path, system
import pygame
from speech_recognition import Recognizer, UnknownValueError, Microphone

COMMAND_LIST = ['ADICIONAR', 'PARAR']

def record_audio(audio):
    """
    Função que grava a o aúdio de responsta para o usuário.
    """
    # Cria a pasta raiz dos audios caso a mesma não exista
    if not path.exists('audios'):
        system('mkdir audios')

    # Atribui a entrada de aúdio e sua linguagem. 
    # O gTTS é um módulo da google para traduzir a escrita para a fala do robo.
    tts = gTTS(audio, lang='pt-br')
    # Salva a entrada no formato mp3.
    tts.save('audios/audio_response.mp3')

    response_audio()


def get_audio():
    """
    Função que resolve a fala do usuário.
    """
    # Função que reconhece a entrada da fala.
    mic_in = Recognizer()
    # Pega a entrada do microfone.
    # Ele vai ouvir o comando e traduzir em escrita para que as demais funções possam iterar com o comando.
    with Microphone() as source:
        # Ajusta a entrada eliminando ruidos.
        mic_in.adjust_for_ambient_noise(source)
        
        # Limpa informações de inicialização do terminal.
        system('clear')
        
        # Mostra na tela os comandos padrões.
        print('Comandos do sistema:')
        for system_commands in COMMAND_LIST:
            print(system_commands)

        user_commands = [info for info in get_command_list()]
        
        if user_commands:
            print('Comandos do usuário:')
            for comand in user_commands:
                print(comand)

        # Armazena a informação de audio na variável.
        audio = mic_in.listen(source)

    try:
        system('clear')
        print('Processando ...')
        # Passa o aúdio para o reconhecedor de padrões do google trasformar em frase.
        phrase = mic_in.recognize_google(audio, language='pt-BR')
        print(f'Entrada de aúdio: {phrase}')
        
        # Condição para o caso de o usuário use o comando ADD.
        if str(phrase).upper() == COMMAND_LIST[0]:
            set_command()

            return get_audio()
        
        if str(phrase).upper() == COMMAND_LIST[1]:
            return system('exit')
        
        if str(phrase).upper() not in get_command_list():
            insult = get_insult(phrase)
            record_audio(insult)

            return get_audio()

        record_audio(get_response(phrase))

        return get_audio()

    except UnknownValueError:
        insult = get_insult('Vox')
        record_audio(insult)

        return get_audio()


def response_audio():
    """
    Responde o usuário com o aúdio da resposta.
    """
    # Limpa o terminal.
    system('clear')

    print('Gerando resposta ...')
    pygame.init()  # pylint: disable=no-member

    # Carrega o aúdio no mixer.
    pygame.mixer.music.load('audios/audio_response.mp3')
    # seleciona o volume da saída do mixer.
    pygame.mixer.music.set_volume(1)
    # Inicia a saída de aúdio.
    pygame.mixer.music.play()

    # Cria um temporizador.
    clock = pygame.time.Clock()

    while pygame.mixer.music.get_busy():
        # Inicia um evento para o caso de o mixer estiver ocupado o proxímo retorno aguarde 10 segundos
        pygame.event.poll()
        clock.tick(10)