422 lines
9.4 KiB
Bash
Executable File
422 lines
9.4 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Google AIY Voice Kit V1 - Installation Script
|
|
# Bilingual Voice Assistant (English/Mandarin)
|
|
#
|
|
# AI Now Inc - Del Mar Demo Unit
|
|
# Laboratory Assistant: Claw 🏭
|
|
#
|
|
|
|
set -e # Exit on error
|
|
|
|
# Colors for output
|
|
RED='\033[0;31m'
|
|
GREEN='\033[0;32m'
|
|
YELLOW='\033[1;33m'
|
|
BLUE='\033[0;34m'
|
|
NC='\033[0m' # No Color
|
|
|
|
# Configuration
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
INSTALL_DIR="/home/pi/voice-assistant"
|
|
MUSIC_DIR="/home/pi/Music"
|
|
LOG_FILE="/var/log/voice-assistant-install.log"
|
|
PYTHON_VERSION="3.9"
|
|
|
|
echo -e "${BLUE}"
|
|
echo "=========================================="
|
|
echo " 🎤 Voice Assistant Installer"
|
|
echo " AI Now Inc - Del Mar Demo Unit"
|
|
echo "=========================================="
|
|
echo -e "${NC}"
|
|
|
|
# Logging function
|
|
log() {
|
|
echo -e "[$(date '+%Y-%m-%d %H:%M:%S')] $1" | tee -a "$LOG_FILE"
|
|
}
|
|
|
|
log_error() {
|
|
echo -e "${RED}[ERROR]${NC} $1" | tee -a "$LOG_FILE"
|
|
}
|
|
|
|
log_success() {
|
|
echo -e "${GREEN}[SUCCESS]${NC} $1" | tee -a "$LOG_FILE"
|
|
}
|
|
|
|
log_info() {
|
|
echo -e "${YELLOW}[INFO]${NC} $1" | tee -a "$LOG_FILE"
|
|
}
|
|
|
|
# Check if running as root
|
|
check_root() {
|
|
if [ "$EUID" -ne 0 ]; then
|
|
log_error "Please run as root (sudo ./install.sh)"
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# Check if running on Raspberry Pi
|
|
check_raspberry_pi() {
|
|
if ! grep -q "Raspberry Pi" /proc/cpuinfo 2>/dev/null; then
|
|
log_info "Not running on Raspberry Pi (this may still work)"
|
|
else
|
|
log_success "Raspberry Pi detected"
|
|
fi
|
|
}
|
|
|
|
# Update system packages
|
|
update_system() {
|
|
log_info "Updating system packages..."
|
|
apt-get update
|
|
apt-get upgrade -y
|
|
log_success "System updated"
|
|
}
|
|
|
|
# Install system dependencies
|
|
install_system_deps() {
|
|
log_info "Installing system dependencies..."
|
|
|
|
apt-get install -y \
|
|
python3 \
|
|
python3-pip \
|
|
python3-dev \
|
|
python3-venv \
|
|
portaudio19-dev \
|
|
libffi-dev \
|
|
libssl-dev \
|
|
libjpeg-dev \
|
|
zlib1g-dev \
|
|
libfreetype6-dev \
|
|
liblcms2-dev \
|
|
libopenjp2-7 \
|
|
libtiff6 \
|
|
libblas-dev \
|
|
liblapack-dev \
|
|
libatlas-base-dev \
|
|
libgfortran5 \
|
|
swig \
|
|
libasound2-dev \
|
|
alsa-utils \
|
|
wget \
|
|
git \
|
|
curl
|
|
|
|
log_success "System dependencies installed"
|
|
}
|
|
|
|
# Install Google AIY Voice Kit
|
|
install_aiy_voice() {
|
|
log_info "Installing Google AIY Voice Kit..."
|
|
|
|
# Check if AIY already installed
|
|
if [ -d "/usr/local/bin/aiy" ]; then
|
|
log_info "AIY Voice Kit already installed"
|
|
return
|
|
fi
|
|
|
|
# Install AIY packages
|
|
cd /tmp
|
|
wget https://dl.google.com/aiyprojects/raspbian/aiyvoice-buster-20230111.zip
|
|
unzip aiyvoice-buster-*.zip
|
|
cd aiyvoice-*
|
|
./install.sh
|
|
|
|
log_success "Google AIY Voice Kit installed"
|
|
}
|
|
|
|
# Create virtual environment
|
|
create_venv() {
|
|
log_info "Creating Python virtual environment..."
|
|
|
|
cd "$INSTALL_DIR"
|
|
python3 -m venv venv
|
|
|
|
log_success "Virtual environment created"
|
|
}
|
|
|
|
# Install Python dependencies
|
|
install_python_deps() {
|
|
log_info "Installing Python dependencies..."
|
|
|
|
cd "$INSTALL_DIR"
|
|
source venv/bin/activate
|
|
|
|
# Upgrade pip
|
|
pip install --upgrade pip
|
|
|
|
# Install requirements
|
|
pip install -r requirements.txt
|
|
|
|
# Install additional dependencies for hotword detection
|
|
pip install porcupine1
|
|
pip install webrtcvad
|
|
|
|
log_success "Python dependencies installed"
|
|
}
|
|
|
|
# Create music directory
|
|
create_music_dir() {
|
|
log_info "Creating music directory..."
|
|
|
|
if [ ! -d "$MUSIC_DIR" ]; then
|
|
mkdir -p "$MUSIC_DIR"
|
|
log_success "Music directory created: $MUSIC_DIR"
|
|
else
|
|
log_info "Music directory already exists"
|
|
fi
|
|
|
|
# Set permissions
|
|
chown pi:pi "$MUSIC_DIR"
|
|
chmod 755 "$MUSIC_DIR"
|
|
}
|
|
|
|
# Configure audio
|
|
configure_audio() {
|
|
log_info "Configuring audio..."
|
|
|
|
# Create/update ALSA configuration
|
|
cat > /etc/asound.conf << 'EOF'
|
|
pcm.!default {
|
|
type plug
|
|
slave.pcm "hw:0,0"
|
|
}
|
|
|
|
ctl.!default {
|
|
type hw
|
|
card 0
|
|
}
|
|
EOF
|
|
|
|
log_success "Audio configured"
|
|
}
|
|
|
|
# Install systemd service
|
|
install_service() {
|
|
log_info "Installing systemd service..."
|
|
|
|
cat > /etc/systemd/system/voice-assistant.service << EOF
|
|
[Unit]
|
|
Description=Bilingual Voice Assistant
|
|
After=network.target sound.target
|
|
|
|
[Service]
|
|
Type=simple
|
|
User=pi
|
|
WorkingDirectory=$INSTALL_DIR
|
|
ExecStart=$INSTALL_DIR/venv/bin/python3 $INSTALL_DIR/main.py --mode run
|
|
Restart=always
|
|
RestartSec=10
|
|
Environment=PYTHONUNBUFFERED=1
|
|
Environment=GOOGLE_APPLICATION_CREDENTIALS=/home/pi/.credentials/google-credentials.json
|
|
|
|
# Logging
|
|
StandardOutput=journal
|
|
StandardError=journal
|
|
SyslogIdentifier=voice-assistant
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
EOF
|
|
|
|
# Enable service
|
|
systemctl daemon-reload
|
|
systemctl enable voice-assistant.service
|
|
|
|
log_success "Systemd service installed and enabled"
|
|
}
|
|
|
|
# Configure hotword detection
|
|
configure_hotword() {
|
|
log_info "Configuring hotword detection..."
|
|
|
|
# Create hotword configuration
|
|
cat > "$INSTALL_DIR/hotword_config.json" << 'EOF'
|
|
{
|
|
"hotwords": [
|
|
{
|
|
"keyword": "hey osiris",
|
|
"keyword_zh": "你好 osiris",
|
|
"sensitivity": 0.5,
|
|
"library_path": "resources/porcupine"
|
|
}
|
|
],
|
|
"audio": {
|
|
"sample_rate": 16000,
|
|
"frame_length": 512
|
|
}
|
|
}
|
|
EOF
|
|
|
|
log_success "Hotword detection configured"
|
|
}
|
|
|
|
# Create sample music directory structure
|
|
create_sample_music_structure() {
|
|
log_info "Creating sample music structure..."
|
|
|
|
mkdir -p "$MUSIC_DIR/samples"
|
|
mkdir -p "$MUSIC_DIR/playlists"
|
|
|
|
# Create a README for music
|
|
cat > "$MUSIC_DIR/README.md" << 'EOF'
|
|
# Music Library
|
|
|
|
Place your MP3 files here. The assistant will automatically detect and index them.
|
|
|
|
## Supported Formats
|
|
- MP3
|
|
- WAV
|
|
- OGG
|
|
- FLAC
|
|
|
|
## Organization
|
|
You can organize music by:
|
|
- Artist/Album/Song.mp3
|
|
- Genre/Song.mp3
|
|
- Or flat structure: Song.mp3
|
|
|
|
## Voice Commands
|
|
- "Play [song name]" / "播放 [歌曲名]"
|
|
- "Pause" / "暂停"
|
|
- "Resume" / "继续"
|
|
- "Next" / "下一首"
|
|
- "Volume up/down" / "音量 大/小"
|
|
EOF
|
|
|
|
chown -R pi:pi "$MUSIC_DIR"
|
|
|
|
log_success "Sample music structure created"
|
|
}
|
|
|
|
# Create startup script
|
|
create_startup_script() {
|
|
log_info "Creating startup script..."
|
|
|
|
cat > "$INSTALL_DIR/start.sh" << 'EOF'
|
|
#!/bin/bash
|
|
# Voice Assistant Startup Script
|
|
|
|
cd "$(dirname "$0")"
|
|
|
|
# Activate virtual environment
|
|
source venv/bin/activate
|
|
|
|
# Run the assistant
|
|
python3 main.py --mode run
|
|
EOF
|
|
|
|
chmod +x "$INSTALL_DIR/start.sh"
|
|
chown pi:pi "$INSTALL_DIR/start.sh"
|
|
|
|
log_success "Startup script created"
|
|
}
|
|
|
|
# Create uninstall script
|
|
create_uninstall_script() {
|
|
log_info "Creating uninstall script..."
|
|
|
|
cat > "$INSTALL_DIR/uninstall.sh" << 'EOF'
|
|
#!/bin/bash
|
|
# Uninstall Voice Assistant
|
|
|
|
echo "Uninstalling Voice Assistant..."
|
|
|
|
# Stop service
|
|
sudo systemctl stop voice-assistant
|
|
sudo systemctl disable voice-assistant
|
|
sudo rm /etc/systemd/system/voice-assistant.service
|
|
|
|
# Remove installation
|
|
sudo rm -rf /home/pi/voice-assistant
|
|
|
|
# Remove music directory (optional)
|
|
# sudo rm -rf /home/pi/Music
|
|
|
|
echo "Uninstall complete!"
|
|
EOF
|
|
|
|
chmod +x "$INSTALL_DIR/uninstall.sh"
|
|
|
|
log_success "Uninstall script created"
|
|
}
|
|
|
|
# Final configuration
|
|
final_configuration() {
|
|
log_info "Running final configuration..."
|
|
|
|
# Copy config if not exists
|
|
if [ ! -f "$INSTALL_DIR/config.local.json" ]; then
|
|
cp "$INSTALL_DIR/config.json" "$INSTALL_DIR/config.local.json"
|
|
log_info "Created local configuration: config.local.json"
|
|
fi
|
|
|
|
# Set permissions
|
|
chown -R pi:pi "$INSTALL_DIR"
|
|
chmod -R 755 "$INSTALL_DIR"
|
|
|
|
log_success "Final configuration complete"
|
|
}
|
|
|
|
# Print next steps
|
|
print_next_steps() {
|
|
echo ""
|
|
echo -e "${GREEN}=========================================="
|
|
echo " Installation Complete! 🎉"
|
|
echo "==========================================${NC}"
|
|
echo ""
|
|
echo "Next steps:"
|
|
echo "1. Edit configuration:"
|
|
echo " nano $INSTALL_DIR/config.local.json"
|
|
echo ""
|
|
echo "2. Add your MP3 files to: $MUSIC_DIR"
|
|
echo ""
|
|
echo "3. Test the assistant:"
|
|
echo " cd $INSTALL_DIR"
|
|
echo " ./start.sh"
|
|
echo ""
|
|
echo "4. Or run in demo mode:"
|
|
echo " $INSTALL_DIR/venv/bin/python3 $INSTALL_DIR/main.py --mode demo"
|
|
echo ""
|
|
echo "5. Start the service:"
|
|
echo " sudo systemctl start voice-assistant"
|
|
echo ""
|
|
echo "6. View logs:"
|
|
echo " sudo journalctl -u voice-assistant -f"
|
|
echo ""
|
|
echo "Voice commands:"
|
|
echo " - 'Hey Osiris' / '你好 Osiris' (hotword)"
|
|
echo " - 'Hello' / '你好'"
|
|
echo " - 'Play music' / '播放音乐'"
|
|
echo " - 'What time is it?' / '现在几点？'"
|
|
echo ""
|
|
echo -e "${YELLOW}Note: Make sure your microphone is connected and working!${NC}"
|
|
echo ""
|
|
}
|
|
|
|
# Main installation
|
|
main() {
|
|
log "Starting installation..."
|
|
|
|
check_root
|
|
check_raspberry_pi
|
|
update_system
|
|
install_system_deps
|
|
# install_aiy_voice # Commented out - install manually if needed
|
|
create_venv
|
|
install_python_deps
|
|
create_music_dir
|
|
configure_audio
|
|
install_service
|
|
configure_hotword
|
|
create_sample_music_structure
|
|
create_startup_script
|
|
create_uninstall_script
|
|
final_configuration
|
|
print_next_steps
|
|
|
|
log_success "Installation completed successfully!"
|
|
}
|
|
|
|
# Run main
|
|
main "$@"
|