148 lines
4.6 KiB
Python
148 lines
4.6 KiB
Python
#
|
||
# Copyright (c) 2024–2025, Daily
|
||
#
|
||
# SPDX-License-Identifier: BSD 2-Clause License
|
||
#
|
||
|
||
"""Camb.ai MARS TTS example with local audio (microphone/speakers).
|
||
|
||
This example demonstrates:
|
||
- Basic TTS synthesis with Camb.ai MARS
|
||
- Local audio input/output (no WebRTC or Daily needed)
|
||
- Handling interruptions
|
||
|
||
Requirements:
|
||
- CAMB_API_KEY environment variable
|
||
- OPENAI_API_KEY environment variable (for LLM)
|
||
- DEEPGRAM_API_KEY environment variable (for STT)
|
||
|
||
Usage:
|
||
export CAMB_API_KEY=your_camb_api_key
|
||
export OPENAI_API_KEY=your_openai_api_key
|
||
export DEEPGRAM_API_KEY=your_deepgram_api_key
|
||
python 07zb-interruptible-camb-local.py [--voice-id VOICE_ID]
|
||
"""
|
||
|
||
import argparse
|
||
import asyncio
|
||
import os
|
||
import sys
|
||
|
||
from dotenv import load_dotenv
|
||
from loguru import logger
|
||
|
||
from pipecat.audio.vad.silero import SileroVADAnalyzer
|
||
from pipecat.audio.vad.vad_analyzer import VADParams
|
||
from pipecat.frames.frames import LLMRunFrame
|
||
from pipecat.metrics.metrics import TTFBMetricsData
|
||
from pipecat.observers.loggers.metrics_log_observer import MetricsLogObserver
|
||
from pipecat.pipeline.pipeline import Pipeline
|
||
from pipecat.pipeline.runner import PipelineRunner
|
||
from pipecat.pipeline.task import PipelineParams, PipelineTask
|
||
from pipecat.processors.aggregators.llm_context import LLMContext
|
||
from pipecat.processors.aggregators.llm_response_universal import (
|
||
LLMContextAggregatorPair,
|
||
)
|
||
from pipecat.services.camb.tts import CambTTSService
|
||
from pipecat.services.deepgram.stt import DeepgramSTTService
|
||
from pipecat.services.openai.llm import OpenAILLMService
|
||
from pipecat.transports.local.audio import LocalAudioTransport, LocalAudioTransportParams
|
||
|
||
load_dotenv(override=True)
|
||
|
||
logger.remove(0)
|
||
logger.add(sys.stderr, level="DEBUG")
|
||
|
||
|
||
async def main(voice_id: int):
|
||
# Local audio transport - uses your microphone and speakers
|
||
transport = LocalAudioTransport(
|
||
LocalAudioTransportParams(
|
||
audio_in_enabled=True,
|
||
audio_out_enabled=True,
|
||
vad_analyzer=SileroVADAnalyzer(params=VADParams(stop_secs=0.2)),
|
||
)
|
||
)
|
||
|
||
# Deepgram STT for speech recognition
|
||
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
|
||
|
||
# Camb.ai TTS with MARS-flash model (uses official SDK)
|
||
tts = CambTTSService(
|
||
api_key=os.getenv("CAMB_API_KEY"),
|
||
voice_id=voice_id,
|
||
model="mars-flash",
|
||
)
|
||
|
||
# OpenAI LLM
|
||
llm = OpenAILLMService(api_key=os.getenv("OPENAI_API_KEY"))
|
||
|
||
# System prompt
|
||
messages = [
|
||
{
|
||
"role": "system",
|
||
"content": """You are a helpful voice assistant powered by Camb.ai's MARS
|
||
text-to-speech technology. Keep your responses concise and conversational since
|
||
they will be spoken aloud. Avoid special characters, emojis, or bullet points.""",
|
||
},
|
||
]
|
||
|
||
# Context management
|
||
context = LLMContext(messages)
|
||
context_aggregator = LLMContextAggregatorPair(context)
|
||
|
||
# Build the pipeline
|
||
pipeline = Pipeline(
|
||
[
|
||
transport.input(), # Microphone input
|
||
stt, # Speech-to-text
|
||
context_aggregator.user(), # User context
|
||
llm, # Language model
|
||
tts, # Camb.ai TTS
|
||
transport.output(), # Speaker output
|
||
context_aggregator.assistant(), # Assistant context
|
||
]
|
||
)
|
||
|
||
# Create pipeline task
|
||
# Use 24kHz sample rate to match Camb.ai TTS output
|
||
# Add MetricsLogObserver to track TTFB metrics
|
||
task = PipelineTask(
|
||
pipeline,
|
||
params=PipelineParams(
|
||
audio_out_sample_rate=24000,
|
||
enable_metrics=True,
|
||
enable_usage_metrics=True,
|
||
),
|
||
observers=[MetricsLogObserver(include_metrics={TTFBMetricsData})],
|
||
)
|
||
|
||
# Start the conversation when the pipeline is ready
|
||
@task.event_handler("on_pipeline_started")
|
||
async def on_pipeline_started(task, frame):
|
||
messages.append(
|
||
{
|
||
"role": "system",
|
||
"content": "Please introduce yourself briefly and ask how you can help.",
|
||
}
|
||
)
|
||
await task.queue_frames([LLMRunFrame()])
|
||
|
||
# Run the pipeline
|
||
runner = PipelineRunner()
|
||
logger.info("Starting Camb.ai TTS bot with local audio...")
|
||
logger.info("Speak into your microphone to interact with the bot.")
|
||
await runner.run(task)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
parser = argparse.ArgumentParser(description="Camb.ai TTS example with local audio")
|
||
parser.add_argument(
|
||
"--voice-id",
|
||
type=int,
|
||
default=147320,
|
||
help="Camb.ai voice ID to use (default: 147320)",
|
||
)
|
||
args = parser.parse_args()
|
||
asyncio.run(main(args.voice_id))
|