jyapayne/vocode-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add logger

Browse source
This commit is contained in:
Ajay Raj 2023-03-20 20:07:15 -07:00
commit 7202a82923
2 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
simple_turn_based_conversation.py
View file

@ -8,9 +8,11 @@ from vocode.turn_based.synthesizer.azure_synthesizer import AzureSynthesizer
from vocode.turn_based.transcriber.whisper_transcriber import WhisperTranscriber from vocode.turn_based.transcriber.whisper_transcriber import WhisperTranscriber
from vocode.turn_based.turn_based_conversation import TurnBasedConversation from vocode.turn_based.turn_based_conversation import TurnBasedConversation
load_dotenv() logging.basicConfig()
vocode.api_key = os.getenv("VOCODE_API_KEY") logger = logging.getLogger(__name__)
logger.setLevel(logging.INFO)
load_dotenv()
if __name__ == "__main__": if __name__ == "__main__":
microphone_input, speaker_output = create_microphone_input_and_speaker_output( microphone_input, speaker_output = create_microphone_input_and_speaker_output(
@ -26,8 +28,14 @@ if __name__ == "__main__":
initial_message="Hello!", initial_message="Hello!",
), ),
synthesizer=AzureSynthesizer(sampling_rate=speaker_output.sampling_rate), synthesizer=AzureSynthesizer(sampling_rate=speaker_output.sampling_rate),
logger=logger,
) )
print("Starting conversation. Press Ctrl+C to exit.")
while True: while True:
conversation.start_speech() try:
input("Press enter to end speech") input("Press enter to start recording...")
conversation.end_speech_and_respond() conversation.start_speech()
input("Press enter to end recording...")
conversation.end_speech_and_respond()
except KeyboardInterrupt:
break

5
vocode/turn_based/turn_based_conversation.py
View file

@ -1,3 +1,4 @@
import logging
from vocode.turn_based.agent.base_agent import BaseAgent from vocode.turn_based.agent.base_agent import BaseAgent
from vocode.turn_based.input_device.base_input_device import ( from vocode.turn_based.input_device.base_input_device import (
BaseInputDevice, BaseInputDevice,
@ -15,6 +16,7 @@ class TurnBasedConversation:
agent: BaseAgent, agent: BaseAgent,
synthesizer: BaseSynthesizer, synthesizer: BaseSynthesizer,
output_device: BaseOutputDevice, output_device: BaseOutputDevice,
logger: logging.Logger = None,
): ):
self.input_device = input_device self.input_device = input_device
self.transcriber = transcriber self.transcriber = transcriber
@ -22,6 +24,7 @@ class TurnBasedConversation:
self.synthesizer = synthesizer self.synthesizer = synthesizer
self.output_device = output_device self.output_device = output_device
self.maybe_play_initial_message() self.maybe_play_initial_message()
self.logger = logger or logging.getLogger(__name__)
def maybe_play_initial_message(self): def maybe_play_initial_message(self):
if self.agent.initial_message: if self.agent.initial_message:
@ -34,5 +37,7 @@ class TurnBasedConversation:
def end_speech_and_respond(self): def end_speech_and_respond(self):
human_input = self.transcriber.transcribe(self.input_device.end_listening()) human_input = self.transcriber.transcribe(self.input_device.end_listening())
self.logger.info(f"Transcription: {human_input}")
agent_response = self.agent.respond(human_input) agent_response = self.agent.respond(human_input)
self.logger.info(f"Agent response: {agent_response}")
self.output_device.send_audio(self.synthesizer.synthesize(agent_response)) self.output_device.send_audio(self.synthesizer.synthesize(agent_response))