screenclipper/apps/worker/worker/main.py

import asyncio

from . import db
from .chat_capture import start_chat_capture
from .signal_detection import start_signal_detection_worker
from .stt_scoring import start_stt_scoring_worker
from .video_render import start_video_render_worker

SESSION_WATCH_INTERVAL_SEC = 10


async def watch_sessions_for_chat_capture():
    while True:
        try:
            sessions = await db.get_active_sessions()
            for s in sessions:
                start_chat_capture(s["id"], s["live_video_id"])
        except Exception as exc:
            print(f"[main] session watcher error: {exc}")
        await asyncio.sleep(SESSION_WATCH_INTERVAL_SEC)


async def main():
    signal_worker = start_signal_detection_worker()
    stt_worker = start_stt_scoring_worker()
    render_worker = start_video_render_worker()
    print("[main] worker started: signal-detection + stt-scoring + video-render queues")

    try:
        await watch_sessions_for_chat_capture()
    finally:
        await signal_worker.close()
        await stt_worker.close()
        await render_worker.close()


if __name__ == "__main__":
    asyncio.run(main())