setup.py

#!/usr/bin/env python3

import os
import sys
import subprocess

class Setup:
	def __init__(self):
		path = os.path.dirname(os.path.realpath(sys.argv[0]))

		# List of system packages to install (from apt)
		packages = [
			"git", "build-essential", "python3-dev", "flex", "bison", "mpv", "hostapd", "dnsmasq",
			"python3-smbus", "python3-evdev", "python3-setuptools", "python3-mido",
			"python3-flask", "python3-flask-socketio", "python3-flask-talisman", "python3-pip",
			"python3-psutil", "python3-pydispatch", "python3-pygame", "iw",
			"iproute2"
		]

		self.run_command("sudo apt-get update")

		# Install system packages using subprocess and handle potential errors
		self.install_packages(packages)

		# Install Python dependencies via pip with --break-system-packages
		self.install_python_packages([
			"pvporcupine", "pvrhino", "pydub", "scipy", "openai", "google-cloud-speech", "elevenlabs", "piper-tts", "pywifi", "requests"
		])

		# Set up Piper TTS models
		self.setup_piper_models()

		# Set up /etc/rc.local for wlan1 auto-connection
		self.setup_rc_local()

	def install_packages(self, packages):
		try:
			# Install packages using apt-get
			subprocess.check_call(["sudo", "apt-get", "install", "-y"] + packages)
		except subprocess.CalledProcessError as e:
			print(f"Failed to install packages: {e}")
			sys.exit(1)

	def install_python_packages(self, packages):
		try:
			for package in packages:
				# Use pip with --break-system-packages flag to allow installation
				subprocess.check_call(["sudo", sys.executable, "-m", "pip", "install", "--break-system-packages", package])
		except subprocess.CalledProcessError as e:
			print(f"Failed to install Python packages: {e}")
			sys.exit(1)

	def setup_piper_models(self):
		try:
			# Get the directory of the current script
			script_dir = os.path.dirname(os.path.abspath(__file__))

			# Download Ryan Low voice model into the script's directory
			subprocess.check_call([
				"wget",
				"-O", os.path.join(script_dir, "en_US-ryan-low.onnx"),
				"https://huggingface.co/rhasspy/piper-voices/resolve/v1.0.0/en/en_US/ryan/low/en_US-ryan-low.onnx?download=true"
			])
			subprocess.check_call([
				"wget",
				"-O", os.path.join(script_dir, "en_US-ryan-low.json"),
				"https://huggingface.co/rhasspy/piper-voices/resolve/v1.0.0/en/en_US/ryan/low/en_US-ryan-low.onnx.json?download=true"
			])

			print(f"Piper TTS models are available in {script_dir}.")
		except subprocess.CalledProcessError as e:
			print(f"Failed to set up Piper models: {e}")
			sys.exit(1)

	def setup_rc_local(self):
		"""Creates /etc/rc.local to automatically connect wlan1 to the same SSID as wlan0."""
		rc_local_content = """#!/bin/bash
# Wait until wlan1 appears
for i in {1..10}; do
	if ip link show wlan1 > /dev/null 2>&1; then
		break
	fi
	sleep 1
done

# Get the SSID from wlan0 and connect wlan1
SSID=$(iw dev wlan0 info | grep ssid | awk '{$1=""; print substr($0,2)}')
if [ -n "$SSID" ]; then
	sudo nmcli dev wifi connect "$SSID" ifname wlan1
fi

exit 0
"""
		try:
			with open("/etc/rc.local", "w") as rc_file:
				rc_file.write(rc_local_content)
			
			# Make rc.local executable
			subprocess.check_call(["sudo", "chmod", "+x", "/etc/rc.local"])

			print("/etc/rc.local has been successfully created and set up.")
		except Exception as e:
			print(f"Failed to set up /etc/rc.local: {e}")
			sys.exit(1)

	def run_command(self, command):
		try:
			subprocess.check_call(command, shell=True)
		except subprocess.CalledProcessError as e:
			print(f"Command failed: {command}\nError: {e}")
			sys.exit(1)

if __name__ == "__main__":
	Setup()