Josh Knapp
44480906a4
- Implement DiarizeService with pyannote.audio speaker detection - Build PipelineService combining transcribe → diarize → merge with overlap-based speaker assignment per segment - Add pipeline.start and diarize.start IPC handlers - Add run_pipeline Tauri command for full pipeline execution - Wire frontend to use pipeline: speakers auto-created with colors, segments assigned to detected speakers - Build SpeakerManager with rename support (double-click or edit button) - Add speaker color coding throughout transcript display - Add pyannote.audio dependency - Tests: 24 Python (including merge logic), 6 Rust, 0 Svelte errors Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
63 lines
1.8 KiB
Python
63 lines
1.8 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,
|
|
hardware_detect_handler,
|
|
make_diarize_handler,
|
|
make_pipeline_handler,
|
|
make_transcribe_handler,
|
|
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)
|
|
registry.register("transcribe.start", make_transcribe_handler())
|
|
registry.register("hardware.detect", hardware_detect_handler)
|
|
registry.register("diarize.start", make_diarize_handler())
|
|
registry.register("pipeline.start", make_pipeline_handler())
|
|
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()
|