From 9315a2098d3f1d05474bb6e24db36a64d2b0cbc1 Mon Sep 17 00:00:00 2001 From: pstruebi Date: Thu, 13 Mar 2025 14:58:03 +0100 Subject: [PATCH] add a init state --- src/multilang_translator/translator_api/translator_api.py | 2 ++ src/multilang_translator/translator_api/translator_models.py | 1 + 2 files changed, 3 insertions(+) diff --git a/src/multilang_translator/translator_api/translator_api.py b/src/multilang_translator/translator_api/translator_api.py index 5cb97a9..e0199b0 100644 --- a/src/multilang_translator/translator_api/translator_api.py +++ b/src/multilang_translator/translator_api/translator_api.py @@ -81,6 +81,8 @@ async def make_announcement(text: str, ep_group: EndpointGroup): # Initialize all endpoints in the group if they were not initalized before for endpoint in ep_group.endpoints: + ep_group.current_state = AnnouncementStates.INIT + endpoints_db.update_group(ep_group.id, ep_group) init_endpoint(endpoint, ep_group.languages) # Translate the text for each language diff --git a/src/multilang_translator/translator_api/translator_models.py b/src/multilang_translator/translator_api/translator_models.py index 4651a1c..d0114eb 100644 --- a/src/multilang_translator/translator_api/translator_models.py +++ b/src/multilang_translator/translator_api/translator_models.py @@ -9,6 +9,7 @@ from pydantic import BaseModel class AnnouncementStates(Enum): IDLE = 0 + INIT = 0.1 TRANSLATING = 0.2 GENERATING_VOICE = 0.4 ROUTING = 0.6