added local test
This commit is contained in:
144
examples/foundational/07zb-interruptible-camb-local.py
Normal file
144
examples/foundational/07zb-interruptible-camb-local.py
Normal file
@@ -0,0 +1,144 @@
|
|||||||
|
#
|
||||||
|
# Copyright (c) 2024–2025, Daily
|
||||||
|
#
|
||||||
|
# SPDX-License-Identifier: BSD 2-Clause License
|
||||||
|
#
|
||||||
|
|
||||||
|
"""Camb.ai MARS-8 TTS example with local audio (microphone/speakers).
|
||||||
|
|
||||||
|
This example demonstrates:
|
||||||
|
- Basic TTS synthesis with Camb.ai MARS-8
|
||||||
|
- 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
|
||||||
|
"""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import aiohttp
|
||||||
|
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.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():
|
||||||
|
# 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"))
|
||||||
|
|
||||||
|
# Create HTTP session for Camb.ai TTS
|
||||||
|
async with aiohttp.ClientSession() as session:
|
||||||
|
# Camb.ai TTS with MARS-8-flash model
|
||||||
|
tts = CambTTSService(
|
||||||
|
api_key=os.getenv("CAMB_API_KEY"),
|
||||||
|
aiohttp_session=session,
|
||||||
|
voice_id=2681, # Attic voice
|
||||||
|
model="mars-8-flash",
|
||||||
|
params=CambTTSService.InputParams(
|
||||||
|
speed=1.0,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
# 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-8
|
||||||
|
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
|
||||||
|
task = PipelineTask(
|
||||||
|
pipeline,
|
||||||
|
params=PipelineParams(
|
||||||
|
enable_metrics=True,
|
||||||
|
enable_usage_metrics=True,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
# 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.")
|
||||||
|
|
||||||
|
# Start the conversation with a greeting after a short delay
|
||||||
|
async def start_greeting():
|
||||||
|
await asyncio.sleep(1) # Wait for pipeline to start
|
||||||
|
messages.append(
|
||||||
|
{
|
||||||
|
"role": "system",
|
||||||
|
"content": "Please introduce yourself briefly and ask how you can help.",
|
||||||
|
}
|
||||||
|
)
|
||||||
|
await task.queue_frames([LLMRunFrame()])
|
||||||
|
|
||||||
|
# Run greeting and pipeline concurrently
|
||||||
|
await asyncio.gather(
|
||||||
|
runner.run(task),
|
||||||
|
start_greeting(),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
asyncio.run(main())
|
||||||
Reference in New Issue
Block a user