53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
|
"""Main entry point for the Voice to Notes Python sidecar."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import signal
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from voice_to_notes.ipc.handlers import HandlerRegistry, ping_handler
|
||
|
|
from voice_to_notes.ipc.messages import ready_message
|
||
|
|
from voice_to_notes.ipc.protocol import read_message, write_message
|
||
|
|
|
||
|
|
|
||
|
|
def create_registry() -> HandlerRegistry:
|
||
|
|
"""Set up the message handler registry."""
|
||
|
|
registry = HandlerRegistry()
|
||
|
|
registry.register("ping", ping_handler)
|
||
|
|
# TODO: Register transcribe, diarize, pipeline, ai, export handlers
|
||
|
|
return registry
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
"""Main loop: read messages from stdin, dispatch to handlers, write responses to stdout."""
|
||
|
|
|
||
|
|
# Handle clean shutdown
|
||
|
|
def shutdown(signum: int, frame: object) -> None:
|
||
|
|
print("[sidecar] Shutting down...", file=sys.stderr, flush=True)
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
signal.signal(signal.SIGTERM, shutdown)
|
||
|
|
signal.signal(signal.SIGINT, shutdown)
|
||
|
|
|
||
|
|
registry = create_registry()
|
||
|
|
|
||
|
|
# Signal to Rust that we're ready
|
||
|
|
write_message(ready_message())
|
||
|
|
print("[sidecar] Ready and waiting for messages", file=sys.stderr, flush=True)
|
||
|
|
|
||
|
|
# Message loop
|
||
|
|
while True:
|
||
|
|
msg = read_message()
|
||
|
|
if msg is None:
|
||
|
|
# EOF — parent closed stdin, time to exit
|
||
|
|
print("[sidecar] EOF on stdin, exiting", file=sys.stderr, flush=True)
|
||
|
|
break
|
||
|
|
|
||
|
|
response = registry.handle(msg)
|
||
|
|
if response is not None:
|
||
|
|
write_message(response)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|