config.py

"""
Configuration module for the mood orchestration agent.
"""
import os
from pathlib import Path
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# API Keys
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
SUNO_API_KEY = os.getenv("SUNO_API_KEY", "")

# Bulb Server Configuration
BULB_SERVER_URL = os.getenv("BULB_SERVER_URL", "http://localhost:3000")

# Audio Configuration
AUDIO_SAMPLE_RATE = 44100  # Hz
AUDIO_CHANNELS = 1  # Mono
RECORDING_DURATION = 5  # seconds

# Mood to Music Style Mappings
MOOD_TO_MUSIC_STYLE = {
    "happy": "upbeat pop music with cheerful melodies, major chords, bright instrumentation",
    "sad": "melancholic piano ballad with slow tempo, minor chords, emotional strings",
    "calm": "ambient relaxing music with soft pads, gentle piano, nature sounds",
    "energetic": "high-energy electronic dance music with driving beats, synth leads",
    "angry": "intense rock music with heavy guitars, aggressive drums, powerful vocals",
}

# File Paths
PROJECT_ROOT = Path(__file__).parent
TEMP_DIR = PROJECT_ROOT / "temp"
TEMP_DIR.mkdir(exist_ok=True)

# Suno API Configuration
SUNO_MODEL = "V4_5ALL"  # Better song structure, max 8 min