Spaces:
Running
Running
| """ | |
| Central configuration for the telecalling agent. | |
| Edit the values in this file to match your local setup. | |
| """ | |
| import os | |
| from pathlib import Path | |
| # Project root | |
| ROOT_DIR = Path(__file__).parent | |
| DATA_DIR = ROOT_DIR / "data" | |
| MODELS_DIR = ROOT_DIR / "models" | |
| DATA_DIR.mkdir(exist_ok=True) | |
| MODELS_DIR.mkdir(exist_ok=True) | |
| # Hugging Face | |
| # Optional: set HF_TOKEN for private models or authenticated downloads. | |
| HF_TOKEN = os.environ.get("HF_TOKEN", "") | |
| # ASR: Hugging Face Moonshine | |
| TRANSCRIBE_MODEL_ID = "UsefulSensors/moonshine-tiny" | |
| TRANSCRIBE_LANGUAGE = "en" # Moonshine Tiny is English ASR | |
| TRANSCRIBE_DEVICE = "cpu" # CPU-only for zero-gpu deployment | |
| TRANSCRIBE_LOCAL_ONLY = os.environ.get("TRANSCRIBE_LOCAL_ONLY", "0").lower() in ("1", "true", "yes") | |
| # Set TRANSCRIBE_LOCAL_ONLY=1 to require a pre-downloaded Moonshine cache and disable remote Hugging Face Hub downloads. | |
| # Intent Parser: Qwen2.5-7B-Instruct (GGUF via llama-cpp-python) | |
| # Using q2_k quantization for smaller size on CPU (~2.3 GB). | |
| # Download from HF Hub on first run if not cached. | |
| QWEN_GGUF_PATH = MODELS_DIR / "qwen2.5-7b-instruct-q2_k.gguf" | |
| QWEN_N_GPU_LAYERS = 0 # CPU-only: no GPU layer offloading | |
| QWEN_N_CTX = 4096 # context window sufficient for a call transcript | |
| QWEN_MAX_TOKENS = 512 # max tokens for the structured JSON response | |
| QWEN_TEMPERATURE = 0.1 # near-deterministic for structured output | |
| # Evaluator: MiniCPM3-4B (CPU, bitsandbytes 4-bit) | |
| MINICPM_MODEL_ID = "openbmb/MiniCPM3-4B" | |
| MINICPM_DEVICE = "cpu" # runs after Qwen is done; no VRAM conflict | |
| MINICPM_MAX_TOKENS = 256 | |
| # VAD: Silero VAD (ONNX, CPU) | |
| VAD_SAMPLE_RATE = 16000 # Hz; Silero and Moonshine both use 16kHz | |
| VAD_CHUNK_MS = 250 # ms per audio chunk fed to VAD | |
| VAD_CHUNK_SAMPLES = int(VAD_SAMPLE_RATE * VAD_CHUNK_MS / 1000) # 4000 | |
| VAD_SILENCE_THRESHOLD = 0.5 | |
| VAD_SILENCE_DURATION_S = 0.8 | |
| VAD_MIN_SPEECH_S = 0.5 | |
| # SQLite database | |
| DB_PATH = DATA_DIR / "calls.db" | |
| # Scheduling rules injected into MiniCPM's system prompt. | |
| SCHEDULING_RULES = """ | |
| 1. Meetings can only be booked Monday-Friday, 09:00-18:00. | |
| 2. Minimum meeting duration is 15 minutes; maximum is 120 minutes. | |
| 3. Back-to-back meetings are not allowed; require a 15-minute gap between slots. | |
| 4. If the caller does not provide a date or time, ask for one before confirming. | |
| 5. If the requested slot is already booked, suggest the next available slot. | |
| 6. Always confirm the caller's name before booking. | |
| """ | |
| # Gradio UI | |
| APP_TITLE = "📞 AI Telecalling Agent" | |
| APP_DESCRIPTION = "Speak naturally — the agent will schedule your meeting automatically." | |
| SERVER_PORT = 7860 | |
| SERVER_NAME = "0.0.0.0" # bind to all interfaces for HF Spaces | |