jarvis.py

"""
Jarvis Audio Controller

Wake word: "Jarvis"
Commands triggered by clap count.

1 clap  -> Coding environment
2 claps -> Gaming environment

Technologies used:
- Picovoice Porcupine (wake word detection)
- PyAudio (microphone streaming)
- SoundDevice + NumPy (clap detection)
- Subprocess (system automation)
"""

import os
import struct
import subprocess

import numpy as np
import pyaudio
import pvporcupine
import sounddevice as sd


# =================================================================
# CONFIGURATION
# =================================================================

# Wake word access key (must be set as environment variable)
ACCESS_KEY = os.getenv("PORCUPINE_ACCESS_KEY")

if not ACCESS_KEY:
    raise ValueError(
        "Porcupine access key not found. "
        "Set PORCUPINE_ACCESS_KEY as an environment variable."
    )

# Application paths (edit if needed for your system)
CHROME_PATH = "chrome"
VSCODE_PATH = "code"
GAME_LAUNCHER = "TLauncher.exe"

# Optional audio file (set to None if you don't want music)
MUSIC_FILE = None

# Clap detection settings
CLAP_THRESHOLD_LOW = 0.05
CLAP_THRESHOLD_HIGH = 0.15

CLAP_LISTEN_DURATION = 2
SAMPLE_RATE = 44100


# =================================================================
# AUDIO PROCESSING
# =================================================================

def detect_claps():
    """
    Records a short burst of audio and checks peak amplitude
    to detect clap-like spikes.
    """

    print("Listening for claps...")

    recording = sd.rec(
        int(CLAP_LISTEN_DURATION * SAMPLE_RATE),
        samplerate=SAMPLE_RATE,
        channels=1
    )

    sd.wait()

    audio = np.abs(recording.flatten())
    peak = np.max(audio)

    print(f"Peak volume detected: {peak:.4f}")

    if peak > CLAP_THRESHOLD_HIGH:
        return 2
    elif peak > CLAP_THRESHOLD_LOW:
        return 1
    else:
        return 0


# =================================================================
# AUTOMATION COMMANDS
# =================================================================

def coding_setup():
    """Launch coding environment"""

    print("Initializing Coding Environment...")

    try:
        subprocess.Popen([
            CHROME_PATH,
            "https://chat.openai.com",
            "https://claude.ai",
            "https://gemini.google.com"
        ])
    except Exception:
        print("Could not open Chrome.")

    try:
        subprocess.Popen(VSCODE_PATH)
    except Exception:
        print("Could not open VS Code.")

    # Optional music playback
    if MUSIC_FILE:
        try:
            subprocess.Popen([
                "powershell",
                "-c",
                f"Start-Process '{MUSIC_FILE}' -WindowStyle Minimized"
            ])
        except Exception:
            print("Could not play audio file.")


def gaming_setup():
    """Launch gaming setup"""

    print("Initializing Gaming Setup...")

    try:
        subprocess.Popen([
            CHROME_PATH,
            "https://youtube.com"
        ])
    except Exception:
        print("Could not open Chrome.")

    try:
        subprocess.Popen(GAME_LAUNCHER)
    except Exception:
        print("Could not launch game launcher.")


# =================================================================
# INITIALIZATION
# =================================================================

print("Initializing Jarvis...")

porcupine = pvporcupine.create(
    access_key=ACCESS_KEY,
    keywords=["jarvis"]
)

pa = pyaudio.PyAudio()

try:
    stream = pa.open(
        rate=porcupine.sample_rate,
        channels=1,
        format=pyaudio.paInt16,
        input=True,
        frames_per_buffer=porcupine.frame_length
    )
except Exception as e:
    print("Microphone initialization failed:", e)
    exit(1)


# =================================================================
# MAIN LOOP
# =================================================================

def main():

    print("\n--- Jarvis is now online ---")
    print("Wake word: Jarvis")
    print("Clap once  -> Coding setup")
    print("Clap twice -> Gaming setup\n")

    try:
        while True:

            pcm = stream.read(porcupine.frame_length)
            pcm = struct.unpack_from("h" * porcupine.frame_length, pcm)

            result = porcupine.process(pcm)

            if result >= 0:

                print("\n[!] Jarvis wake-word detected!")

                claps = detect_claps()

                print(f"Action triggered by {claps} clap(s)")

                if claps == 1:
                    coding_setup()

                elif claps == 2:
                    gaming_setup()

                else:
                    print("No action taken.")

    except KeyboardInterrupt:
        print("\nStopping Jarvis...")

    finally:
        stream.stop_stream()
        stream.close()
        pa.terminate()
        porcupine.delete()


# =================================================================
# ENTRY POINT
# =================================================================

if __name__ == "__main__":
    main()