Fixed typo

QueueStack -> ProcessStack
This commit is contained in:
Sviatoslav Tsariov Yurievich 2024-03-29 10:39:10 +03:00
parent 70e6e6ca90
commit 3801925b72
9 changed files with 14 additions and 14 deletions

View File

@ -5,8 +5,8 @@ import sys
import config
from queue_stack import QueueStack
from queue_stack.strategies import RecognizeAndSendStrategy
from process_stack import ProcessQueue
from process_stack.strategies import RecognizeAndSendStrategy
from recognizer import Recognizer
from recognizer.strategies import WhisperStrategy, FastWhisperStrategy
@ -22,8 +22,8 @@ fast_whisper_recognizer = Recognizer(FastWhisperStrategy())
sos_message_service = MessageService(SosMessageStrategy())
number_message_service = MessageService(NumberMessageStrategy())
queue_stack = QueueStack(RecognizeAndSendStrategy())
queue_stack.start_loop_in_thread()
process_stack = ProcessQueue(RecognizeAndSendStrategy())
process_stack.start_loop_in_thread()
@app.route("/")
def hello():
@ -39,7 +39,7 @@ def recognize_files(message_service: MessageService):
temp = NamedTemporaryFile()
handle.save(temp)
results.append(queue_stack.append_and_await((
results.append(process_stack.append_and_await((
temp,
whisper_recognizer,
message_service,

View File

@ -4,7 +4,7 @@ from dotenv import load_dotenv
load_dotenv()
HARPYIA_MODEL = os.getenv('HARPYIA_MODEL') or 'small'
HARPYIA_MODEL = os.getenv('HARPYIA_MODEL') or 'medium'
HARPYIA_LANGUAGE = os.getenv('HARPYIA_LANGUAGE') or 'ru'
HARPYIA_SAMPLE_RATE = os.getenv('HARPYIA_SAMPLE_RATE') or 160000
@ -12,7 +12,7 @@ WHISPER_NUM_WORKERS = os.getenv('WHISPER_NUM_WORKERS') or 6
WHISPER_CPU_THREADS = os.getenv('WHISPER_CPU_THREADS') or 10
WHISPER_BEAM_SIZE = os.getenv('WHISPER_BEAM_SIZE') or 5
SOS_PROMPT = os.getenv('SOS_PROMPT') or 'спасите помогите помощь пожар караул кирилл'
SOS_PROMPT = os.getenv('SOS_PROMPT') or 'спасите помогите помощь пожар караул гит'
NUMBER_PROMPT = os.getenv('NUMBER_PROMPT') or 'один два три четыре пять шесть семь восемь девять десять одинадцать двенадцать тринадцать сто сот'
RAT_URL = os.getenv('RAT_URL') or 'localhost:8081'

View File

@ -0,0 +1 @@
from process_stack.process_stack import ProcessQueue

View File

@ -2,9 +2,9 @@ import sys
from threading import Thread, Event, Lock
from queue_stack.strategies import BaseProcessStrategy
from process_stack.strategies import BaseProcessStrategy
class QueueStack:
class ProcessQueue:
def __init__(self, strategy: BaseProcessStrategy) -> None:
self._stack = []
self._strategy = strategy

View File

@ -0,0 +1,2 @@
from process_stack.strategies.base_process_strategy import BaseProcessStrategy
from process_stack.strategies.recognize_and_send_strategy import RecognizeAndSendStrategy

View File

@ -1,6 +1,6 @@
import sys
from queue_stack.strategies import BaseProcessStrategy
from process_stack.strategies import BaseProcessStrategy
from message import MessageService
from recognizer import Recognizer

View File

@ -1 +0,0 @@
from queue_stack.queue_stack import QueueStack

View File

@ -1,2 +0,0 @@
from queue_stack.strategies.base_process_strategy import BaseProcessStrategy
from queue_stack.strategies.recognize_and_send_strategy import RecognizeAndSendStrategy