-
Notifications
You must be signed in to change notification settings - Fork 0
/
vox_in_out.py
113 lines (86 loc) · 3.34 KB
/
vox_in_out.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
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)