database.py

import sqlite3
import os
from datetime import datetime

# Define the database file path
DB_FILE = os.path.join("results", "latency_metrics.db")

def init_db():
    """Initializes the database and creates the latency_metrics table if it doesn't exist."""
    os.makedirs("results", exist_ok=True)
    conn = sqlite3.connect(DB_FILE)
    cursor = conn.cursor()
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS latency_metrics (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            timestamp TEXT NOT NULL,
            room_name TEXT NOT NULL,
            transport_latency_ms INTEGER,
            stt_latency_ms INTEGER,
            llm_latency_ms INTEGER,
            tts_latency_ms INTEGER,
            total_roundtrip_ms INTEGER
        )
    """)
    conn.commit()
    conn.close()
    print(f"Database initialized at {DB_FILE}")

def log_latency(room_name, metrics):
    """Logs a new set of latency metrics to the database."""
    conn = sqlite3.connect(DB_FILE)
    cursor = conn.cursor()
    cursor.execute("""
        INSERT INTO latency_metrics (timestamp, room_name, transport_latency_ms, stt_latency_ms, llm_latency_ms, tts_latency_ms, total_roundtrip_ms)
        VALUES (?, ?, ?, ?, ?, ?, ?)
    """, (
        datetime.utcnow().isoformat(),
        room_name,
        metrics.get("transport_latency_ms"),
        metrics.get("stt_latency_ms"),
        metrics.get("llm_latency_ms"),
        metrics.get("tts_latency_ms"),
        metrics.get("total_roundtrip_ms")
    ))
    conn.commit()
    conn.close()

if __name__ == "__main__":
    init_db()