every_holiday/VRCT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

👍️[Update] Controller : controllerからuiに指示するendpointをrun_mappingに集約するように変更

Browse Source
This commit is contained in:
misyaguziya
2024-09-25 02:19:20 +09:00
parent e119647545
commit a1cb3dc029
2 changed files with 192 additions and 232 deletions
Download Patch File Download Diff File Expand all files Collapse all files

322
src-python/webui_controller.py
View File

@@ -1,4 +1,4 @@
from typing import Callable, Union from typing import Callable, Union, Any
from time import sleep from time import sleep
from subprocess import Popen from subprocess import Popen
from threading import Thread from threading import Thread
@@ -9,125 +9,146 @@ from utils import isUniqueStrings, printLog
from models.transcription.transcription_utils import device_manager from models.transcription.transcription_utils import device_manager
class Controller: class Controller:
def __call__(self, mapping:dict={}) -> None: def __call__(self) -> None:
self.mapping = mapping self.run_mapping = {}
self.run = None
def setRunMapping(self, run_mapping:dict) -> None:
self.run_mapping = run_mapping
def setRun(self, run:Callable[[int, str, Any], None]) -> None:
self.run = run
# response functions # response functions
class DownloadSoftwareProgressBar: class DownloadSoftwareProgressBar:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, progress) -> None: def set(self, progress) -> None:
printLog("Software Download Progress", progress) self.parent.run(
self.run("download", { 200,
"status":200, self.parent.run_mapping["download_software"],
"result":{ progress,
"progress":progress )
}
})
class UpdateSoftwareProgressBar: class UpdateSoftwareProgressBar:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, progress) -> None: def set(self, progress) -> None:
printLog("Software Update Progress", progress) self.parent.run(
self.run("update", { 200,
"status":200, self.parent.run_mapping["update_software"],
"result":{ progress,
"progress":progress )
}
})
class UpdateSelectedMicDevice: class UpdateSelectedMicDevice:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, host, device) -> None: def set(self, host, device) -> None:
config.SELECTED_MIC_HOST = host config.SELECTED_MIC_HOST = host
config.SELECTED_MIC_DEVICE = device config.SELECTED_MIC_DEVICE = device
printLog("Update Host/Mic Device", f"{host}/{device}") self.parent.run(
self.run("mic", { 200,
"status":200, self.parent.run_mapping["selected_mic_device"],
"result":{"host":host, "device":device} {"host":host, "device":device},
}) )
class UpdateSelectedSpeakerDevice: class UpdateSelectedSpeakerDevice:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, device) -> None: def set(self, device) -> None:
config.SELECTED_SPEAKER_DEVICE = device config.SELECTED_SPEAKER_DEVICE = device
printLog("Update Speaker Device", device) self.parent.run(
self.run("speaker", { 200,
"status":200, self.parent.run_mapping["selected_speaker_device"],
"result":device device,
}) )
class ProgressBarMicEnergy: class ProgressBarMicEnergy:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, energy) -> None: def set(self, energy) -> None:
if energy is False: if energy is False:
self.run("error_device", {"status":400,"result": {"message":"No mic device detected."}}) self.parent.run(
400,
self.parent.run_mapping["error_device"],
{"message":"No mic device detected."},
)
else: else:
self.run("mic", {"status":200, "result":energy}) self.parent.run(
200,
self.parent.run_mapping["check_mic_volume"],
energy,
)
class ProgressBarSpeakerEnergy: class ProgressBarSpeakerEnergy:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, energy) -> None: def set(self, energy) -> None:
if energy is False: if energy is False:
self.run("error_device", {"status":400,"result": {"message":"No mic device detected."}}) self.parent.run(
400,
self.parent.run_mapping["error_device"],
{"message":"No mic device detected."},
)
else: else:
self.run("speaker", {"status":200, "result":energy}) self.parent.run(
200,
self.parent.run_mapping["check_speaker_volume"],
energy,
)
class DownloadCTranslate2ProgressBar:
def __init__(self, parent) -> None:
self.parent = parent
def set(self, progress) -> None:
printLog("CTranslate2 Weight Download Progress", progress)
self.parent.run(
200,
self.parent.run_mapping["download_ctranslate2"],
progress,
)
class DownloadWhisperProgressBar: class DownloadWhisperProgressBar:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def set(self, progress) -> None: def set(self, progress) -> None:
printLog("Whisper Weight Download Progress", progress) printLog("Whisper Weight Download Progress", progress)
self.run("download", { self.parent.run(
"status":200, 200,
"result":{ self.parent.run_mapping["download_whisper"],
"progress":progress progress,
} )
})
class MicMessage: class MicMessage:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def send(self, message: Union[str, bool]) -> None: def send(self, message: Union[str, bool]) -> None:
if isinstance(message, bool) and message is False: if isinstance(message, bool) and message is False:
self.run("error_device", { self.parent.run(
"status":400, 400,
"result": { self.parent.run_mapping["error_device"],
"message":"No mic device detected." {"message":"No mic device detected."},
} )
})
elif isinstance(message, str) and len(message) > 0: elif isinstance(message, str) and len(message) > 0:
# addSentMessageLog(message) # addSentMessageLog(message)
translation = [] translation = []
transliteration = [] transliteration = []
if model.checkKeywords(message): if model.checkKeywords(message):
self.run("word_filter", { self.parent.run(
"status":200, 200,
"result": { self.parent.run_mapping["word_filter"],
"message":f"Detected by word filter:{message}" {"message":f"Detected by word filter:{message}"},
} )
})
return return
elif model.detectRepeatSendMessage(message): elif model.detectRepeatSendMessage(message):
return return
@@ -137,12 +158,11 @@ class Controller:
translation, success = model.getInputTranslate(message) translation, success = model.getInputTranslate(message)
if all(success) is not True: if all(success) is not True:
self.parent.changeToCTranslate2Process() self.parent.changeToCTranslate2Process()
self.run("error_translation_engine", { self.parent.run(
"status":400, 400,
"result": { self.parent.run_mapping["error_translation_engine"],
"message":"translation engine limit error" {"message":"translation engine limit error"},
} )
})
if config.CONVERT_MESSAGE_TO_ROMAJI is True or config.CONVERT_MESSAGE_TO_HIRAGANA is True: if config.CONVERT_MESSAGE_TO_ROMAJI is True or config.CONVERT_MESSAGE_TO_HIRAGANA is True:
if config.SELECTED_TARGET_LANGUAGES[config.SELECTED_TAB_NO]["primary"]["language"] == "Japanese": if config.SELECTED_TARGET_LANGUAGES[config.SELECTED_TAB_NO]["primary"]["language"] == "Japanese":
@@ -159,13 +179,13 @@ class Controller:
osc_message = self.parent.messageFormatter("SEND", translation, [message]) osc_message = self.parent.messageFormatter("SEND", translation, [message])
model.oscSendMessage(osc_message) model.oscSendMessage(osc_message)
self.run("mic", { self.parent.run(
"status":200, 200,
"result": { self.parent.run_mapping["transcription_mic"],
{
"message":message, "message":message,
"translation":translation, "translation":translation,
"transliteration":transliteration "transliteration":transliteration
}
}) })
if config.LOGGER_FEATURE is True: if config.LOGGER_FEATURE is True:
if len(translation) > 0: if len(translation) > 0:
@@ -179,18 +199,16 @@ class Controller:
# model.updateOverlay(overlay_image) # model.updateOverlay(overlay_image)
class SpeakerMessage: class SpeakerMessage:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def receive(self, message): def receive(self, message):
if isinstance(message, bool) and message is False: if isinstance(message, bool) and message is False:
self.run("error_device",{ self.parent.run(
"status":400, 400,
"result": { self.parent.run_mapping["error_device"],
"message":"No mic device detected." {"message":"No mic device detected."},
}, )
})
elif isinstance(message, str) and len(message) > 0: elif isinstance(message, str) and len(message) > 0:
translation = [] translation = []
transliteration = [] transliteration = []
@@ -202,12 +220,11 @@ class Controller:
translation, success = model.getOutputTranslate(message) translation, success = model.getOutputTranslate(message)
if all(success) is not True: if all(success) is not True:
self.parent.changeToCTranslate2Process() self.parent.changeToCTranslate2Process()
self.run("error_translation_engine", { self.parent.run(
"status":400, 400,
"result": { self.parent.run_mapping["error_translation_engine"],
"message":"translation engine limit error" {"message":"translation engine limit error"},
} )
})
if config.CONVERT_MESSAGE_TO_ROMAJI is True or config.CONVERT_MESSAGE_TO_HIRAGANA is True: if config.CONVERT_MESSAGE_TO_ROMAJI is True or config.CONVERT_MESSAGE_TO_HIRAGANA is True:
if config.SELECTED_TARGET_LANGUAGES[config.SELECTED_TAB_NO]["primary"]["language"] == "Japanese": if config.SELECTED_TARGET_LANGUAGES[config.SELECTED_TAB_NO]["primary"]["language"] == "Japanese":
@@ -230,13 +247,13 @@ class Controller:
# ------------Speaker2Chatbox------------ # ------------Speaker2Chatbox------------
# update textbox message log (Received) # update textbox message log (Received)
self.run("speaker",{ self.parent.run(
"status":200, 200,
"result": { self.parent.run_mapping["speaker"],
{
"message":message, "message":message,
"translation":translation, "translation":translation,
"transliteration":transliteration, "transliteration":transliteration,
}
}) })
if config.LOGGER_FEATURE is True: if config.LOGGER_FEATURE is True:
if len(translation) > 0: if len(translation) > 0:
@@ -244,9 +261,8 @@ class Controller:
model.logger.info(f"[RECEIVED] {message}{translation}") model.logger.info(f"[RECEIVED] {message}{translation}")
class ChatMessage: class ChatMessage:
def __init__(self, parent, run:Callable[[dict], None]) -> None: def __init__(self, parent) -> None:
self.parent = parent self.parent = parent
self.run = run
def send(self, data): def send(self, data):
id = data["id"] id = data["id"]
@@ -270,12 +286,11 @@ class Controller:
if all(success) is not True: if all(success) is not True:
self.parent.changeToCTranslate2Process() self.parent.changeToCTranslate2Process()
self.run("error_translation_engine", { self.parent.run(
"status":400, 400,
"result":{ self.parent.run_mapping["error_translation_engine"],
"message":"translation engine limit error" {"message":"translation engine limit error"},
} )
})
if config.CONVERT_MESSAGE_TO_ROMAJI is True or config.CONVERT_MESSAGE_TO_HIRAGANA is True: if config.CONVERT_MESSAGE_TO_ROMAJI is True or config.CONVERT_MESSAGE_TO_HIRAGANA is True:
if config.SELECTED_TARGET_LANGUAGES[config.SELECTED_TAB_NO]["primary"]["language"] == "Japanese": if config.SELECTED_TARGET_LANGUAGES[config.SELECTED_TAB_NO]["primary"]["language"] == "Japanese":
@@ -313,20 +328,6 @@ class Controller:
}, },
} }
class DownloadCTranslate2ProgressBar:
def __init__(self, parent, run:Callable[[dict], None]) -> None:
self.parent = parent
self.run = run
def set(self, progress) -> None:
printLog("CTranslate2 Weight Download Progress", progress)
self.run("download", {
"status":200,
"result":{
"progress":progress
}
})
@staticmethod @staticmethod
def getVersion(*args, **kwargs) -> dict: def getVersion(*args, **kwargs) -> dict:
return {"status":200, "result":config.VERSION} return {"status":200, "result":config.VERSION}
@@ -396,16 +397,16 @@ class Controller:
self.stopThreadingTranscriptionReceiveMessageOnOpenConfigWindow() self.stopThreadingTranscriptionReceiveMessageOnOpenConfigWindow()
return {"status":200, "result":True} return {"status":200, "result":True}
def setDisableConfigWindow(self, data, run, *args, **kwargs) -> dict: def setDisableConfigWindow(self, *args, **kwargs) -> dict:
model.stopCheckMicEnergy() model.stopCheckMicEnergy()
model.stopCheckSpeakerEnergy() model.stopCheckSpeakerEnergy()
if config.ENABLE_TRANSCRIPTION_SEND is True: if config.ENABLE_TRANSCRIPTION_SEND is True:
self.startThreadingTranscriptionSendMessageOnCloseConfigWindow(run) self.startThreadingTranscriptionSendMessageOnCloseConfigWindow()
if config.ENABLE_TRANSCRIPTION_RECEIVE is True: if config.ENABLE_TRANSCRIPTION_RECEIVE is True:
sleep(2) sleep(2)
if config.ENABLE_TRANSCRIPTION_RECEIVE is True: if config.ENABLE_TRANSCRIPTION_RECEIVE is True:
self.startThreadingTranscriptionReceiveMessageOnCloseConfigWindow(run) self.startThreadingTranscriptionReceiveMessageOnCloseConfigWindow()
return {"status":200, "result":True} return {"status":200, "result":True}
@staticmethod @staticmethod
@@ -621,9 +622,9 @@ class Controller:
def getAutoMicSelect(*args, **kwargs) -> dict: def getAutoMicSelect(*args, **kwargs) -> dict:
return {"status":200, "result":config.AUTO_MIC_SELECT} return {"status":200, "result":config.AUTO_MIC_SELECT}
def setEnableAutoMicSelect(self, data, run, *args, **kwargs) -> dict: def setEnableAutoMicSelect(self, *args, **kwargs) -> dict:
config.AUTO_MIC_SELECT = True config.AUTO_MIC_SELECT = True
update_device = self.UpdateSelectedMicDevice(self, run) update_device = self.UpdateSelectedMicDevice(self)
device_manager.setCallbackDefaultInputDevice(update_device.set) device_manager.setCallbackDefaultInputDevice(update_device.set)
device_manager.noticeDefaultDevice() device_manager.noticeDefaultDevice()
return {"status":200, "result":config.AUTO_MIC_SELECT} return {"status":200, "result":config.AUTO_MIC_SELECT}
@@ -804,9 +805,9 @@ class Controller:
def getAutoSpeakerSelect(*args, **kwargs) -> dict: def getAutoSpeakerSelect(*args, **kwargs) -> dict:
return {"status":200, "result":config.AUTO_SPEAKER_SELECT} return {"status":200, "result":config.AUTO_SPEAKER_SELECT}
def setEnableAutoSpeakerSelect(self, data, run, *args, **kwargs) -> dict: def setEnableAutoSpeakerSelect(self, *args, **kwargs) -> dict:
config.AUTO_SPEAKER_SELECT = True config.AUTO_SPEAKER_SELECT = True
update_device = self.UpdateSelectedSpeakerDevice(self, run) update_device = self.UpdateSelectedSpeakerDevice(self)
device_manager.setCallbackDefaultOutputDevice(update_device.set) device_manager.setCallbackDefaultOutputDevice(update_device.set)
device_manager.noticeDefaultDevice() device_manager.noticeDefaultDevice()
return {"status":200, "result":config.AUTO_SPEAKER_SELECT} return {"status":200, "result":config.AUTO_SPEAKER_SELECT}
@@ -1283,8 +1284,8 @@ class Controller:
model.changeMicTranscriptStatus() model.changeMicTranscriptStatus()
return {"status":200, "result":config.VRC_MIC_MUTE_SYNC} return {"status":200, "result":config.VRC_MIC_MUTE_SYNC}
def setEnableCheckSpeakerThreshold(self, data, run, *args, **kwargs) -> dict: def setEnableCheckSpeakerThreshold(self, *args, **kwargs) -> dict:
progressbar_speaker_energy = self.ProgressBarSpeakerEnergy(run) progressbar_speaker_energy = self.ProgressBarSpeakerEnergy(self)
model.startCheckSpeakerEnergy( model.startCheckSpeakerEnergy(
progressbar_speaker_energy.set, progressbar_speaker_energy.set,
) )
@@ -1297,8 +1298,8 @@ class Controller:
config.ENABLE_CHECK_ENERGY_RECEIVE = False config.ENABLE_CHECK_ENERGY_RECEIVE = False
return {"status":200, "result":config.ENABLE_CHECK_ENERGY_RECEIVE} return {"status":200, "result":config.ENABLE_CHECK_ENERGY_RECEIVE}
def setEnableCheckMicThreshold(self, data, run, *args, **kwargs) -> dict: def setEnableCheckMicThreshold(self, *args, **kwargs) -> dict:
progressbar_mic_energy = self.ProgressBarMicEnergy(self, run) progressbar_mic_energy = self.ProgressBarMicEnergy(self)
model.startCheckMicEnergy( model.startCheckMicEnergy(
progressbar_mic_energy.set, progressbar_mic_energy.set,
) )
@@ -1311,10 +1312,10 @@ class Controller:
config.ENABLE_CHECK_ENERGY_SEND = False config.ENABLE_CHECK_ENERGY_SEND = False
return {"status":200, "result":config.ENABLE_CHECK_ENERGY_SEND} return {"status":200, "result":config.ENABLE_CHECK_ENERGY_SEND}
# def updateSoftware(data, run, *args, **kwargs) -> dict: # def updateSoftware(*args, **kwargs) -> dict:
# printLog("Update callbackUpdateSoftware") # printLog("Update callbackUpdateSoftware")
# download = DownloadSoftwareProgressBar(run) # download = DownloadSoftwareProgressBar(self)
# update = UpdateSoftwareProgressBar(run) # update = UpdateSoftwareProgressBar(self)
# model.updateSoftware(restart=True, download=download.set, update=update.set) # model.updateSoftware(restart=True, download=download.set, update=update.set)
# return {"status":200, "result":True} # return {"status":200, "result":True}
@@ -1333,8 +1334,8 @@ class Controller:
Popen(['explorer', config.PATH_LOCAL.replace('/', '\\')], shell=True) Popen(['explorer', config.PATH_LOCAL.replace('/', '\\')], shell=True)
return {"status":200, "result":True} return {"status":200, "result":True}
def setEnableTranscriptionSend(self, data, run, *args, **kwargs) -> dict: def setEnableTranscriptionSend(self, *args, **kwargs) -> dict:
self.startThreadingTranscriptionSendMessage(run) self.startThreadingTranscriptionSendMessage()
config.ENABLE_TRANSCRIPTION_SEND = True config.ENABLE_TRANSCRIPTION_SEND = True
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND}
@@ -1343,8 +1344,8 @@ class Controller:
config.ENABLE_TRANSCRIPTION_SEND = False config.ENABLE_TRANSCRIPTION_SEND = False
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_SEND}
def setEnableTranscriptionReceive(self, data, run, *args, **kwargs) -> dict: def setEnableTranscriptionReceive(self, *args, **kwargs) -> dict:
self.startThreadingTranscriptionReceiveMessage(run) self.startThreadingTranscriptionReceiveMessage()
if config.OVERLAY_SMALL_LOG is True: if config.OVERLAY_SMALL_LOG is True:
if model.overlay.initialized is False and model.overlay.checkSteamvrRunning() is True: if model.overlay.initialized is False and model.overlay.checkSteamvrRunning() is True:
model.startOverlay() model.startOverlay()
@@ -1356,8 +1357,8 @@ class Controller:
config.ENABLE_TRANSCRIPTION_RECEIVE = False config.ENABLE_TRANSCRIPTION_RECEIVE = False
return {"status":200, "result":config.ENABLE_TRANSCRIPTION_RECEIVE} return {"status":200, "result":config.ENABLE_TRANSCRIPTION_RECEIVE}
def sendMessageBox(self, data, run, *args, **kwargs) -> dict: def sendMessageBox(self, data, *args, **kwargs) -> dict:
chat = self.ChatMessage(self, run) chat = self.ChatMessage(self)
response = chat.send(data) response = chat.send(data)
return response return response
@@ -1393,13 +1394,13 @@ class Controller:
} }
} }
def downloadCtranslate2Weight(self, data, run, *args, **kwargs) -> dict: def downloadCtranslate2Weight(self, *args, **kwargs) -> dict:
download = self.DownloadCTranslate2ProgressBar(self, run) download = self.DownloadCTranslate2ProgressBar(self)
self.startThreadingDownloadCtranslate2Weight(download.set) self.startThreadingDownloadCtranslate2Weight(download.set)
return {"status":200} return {"status":200}
def downloadWhisperWeight(self, data, run, *args, **kwargs) -> dict: def downloadWhisperWeight(self, *args, **kwargs) -> dict:
download = self.DownloadWhisperProgressBar(self, run) download = self.DownloadWhisperProgressBar(self)
self.startThreadingDownloadWhisperWeight(download.set) self.startThreadingDownloadWhisperWeight(download.set)
return {"status":200} return {"status":200}
@@ -1425,16 +1426,16 @@ class Controller:
def changeToCTranslate2Process() -> None: def changeToCTranslate2Process() -> None:
config.SELECTED_TRANSLATION_ENGINES[config.SELECTED_TAB_NO] = "CTranslate2" config.SELECTED_TRANSLATION_ENGINES[config.SELECTED_TAB_NO] = "CTranslate2"
def startTranscriptionSendMessage(self, run:Callable[[dict], None]) -> None: def startTranscriptionSendMessage(self) -> None:
mic_message = self.MicMessage(self, run) mic_message = self.MicMessage(self)
model.startMicTranscript(mic_message.send) model.startMicTranscript(mic_message.send)
@staticmethod @staticmethod
def stopTranscriptionSendMessage() -> None: def stopTranscriptionSendMessage() -> None:
model.stopMicTranscript() model.stopMicTranscript()
def startThreadingTranscriptionSendMessage(self, run:Callable[[dict], None]) -> None: def startThreadingTranscriptionSendMessage(self) -> None:
th_startTranscriptionSendMessage = Thread(target=self.startTranscriptionSendMessage, args=(run,)) th_startTranscriptionSendMessage = Thread(target=self.startTranscriptionSendMessage)
th_startTranscriptionSendMessage.daemon = True th_startTranscriptionSendMessage.daemon = True
th_startTranscriptionSendMessage.start() th_startTranscriptionSendMessage.start()
@@ -1444,35 +1445,34 @@ class Controller:
th_stopTranscriptionSendMessage.start() th_stopTranscriptionSendMessage.start()
th_stopTranscriptionSendMessage.join() th_stopTranscriptionSendMessage.join()
def startTranscriptionSendMessageOnCloseConfigWindow(self, run:Callable[[dict], None]) -> None: def startTranscriptionSendMessageOnCloseConfigWindow(self) -> None:
mic_message = self.MicMessage(self, run) mic_message = self.MicMessage(self)
model.startMicTranscript(mic_message.send) model.startMicTranscript(mic_message.send)
@staticmethod @staticmethod
def stopTranscriptionSendMessageOnOpenConfigWindow() -> None: def stopTranscriptionSendMessageOnOpenConfigWindow() -> None:
model.stopMicTranscript() model.stopMicTranscript()
def startThreadingTranscriptionSendMessageOnCloseConfigWindow(self, run:Callable[[dict], None]) -> None: def startThreadingTranscriptionSendMessageOnCloseConfigWindow(self) -> None:
th_startTranscriptionSendMessage = Thread(target=self.startTranscriptionSendMessageOnCloseConfigWindow, args=(run,)) th_startTranscriptionSendMessage = Thread(target=self.startTranscriptionSendMessageOnCloseConfigWindow)
th_startTranscriptionSendMessage.daemon = True th_startTranscriptionSendMessage.daemon = True
th_startTranscriptionSendMessage.start() th_startTranscriptionSendMessage.start()
@staticmethod
def stopThreadingTranscriptionSendMessageOnOpenConfigWindow(self) -> None: def stopThreadingTranscriptionSendMessageOnOpenConfigWindow(self) -> None:
th_stopTranscriptionSendMessage = Thread(target=self.stopTranscriptionSendMessageOnOpenConfigWindow) th_stopTranscriptionSendMessage = Thread(target=self.stopTranscriptionSendMessageOnOpenConfigWindow)
th_stopTranscriptionSendMessage.daemon = True th_stopTranscriptionSendMessage.daemon = True
th_stopTranscriptionSendMessage.start() th_stopTranscriptionSendMessage.start()
def startTranscriptionReceiveMessage(self, run:Callable[[dict], None]) -> None: def startTranscriptionReceiveMessage(self) -> None:
speaker_message = self.SpeakerMessage(self, run) speaker_message = self.SpeakerMessage(self)
model.startSpeakerTranscript(speaker_message.receive) model.startSpeakerTranscript(speaker_message.receive)
@staticmethod @staticmethod
def stopTranscriptionReceiveMessage() -> None: def stopTranscriptionReceiveMessage() -> None:
model.stopSpeakerTranscript() model.stopSpeakerTranscript()
def startThreadingTranscriptionReceiveMessage(self, run:Callable[[dict], None]) -> None: def startThreadingTranscriptionReceiveMessage(self) -> None:
th_startTranscriptionReceiveMessage = Thread(target=self.startTranscriptionReceiveMessage, args=(run,)) th_startTranscriptionReceiveMessage = Thread(target=self.startTranscriptionReceiveMessage)
th_startTranscriptionReceiveMessage.daemon = True th_startTranscriptionReceiveMessage.daemon = True
th_startTranscriptionReceiveMessage.start() th_startTranscriptionReceiveMessage.start()
@@ -1482,16 +1482,16 @@ class Controller:
th_stopTranscriptionReceiveMessage.start() th_stopTranscriptionReceiveMessage.start()
th_stopTranscriptionReceiveMessage.join() th_stopTranscriptionReceiveMessage.join()
def startTranscriptionReceiveMessageOnCloseConfigWindow(self, run:Callable[[dict], None]) -> None: def startTranscriptionReceiveMessageOnCloseConfigWindow(self) -> None:
speaker_message = self.SpeakerMessage(self, run) speaker_message = self.SpeakerMessage(self)
model.startSpeakerTranscript(speaker_message.receive) model.startSpeakerTranscript(speaker_message.receive)
@staticmethod @staticmethod
def stopTranscriptionReceiveMessageOnOpenConfigWindow() -> None: def stopTranscriptionReceiveMessageOnOpenConfigWindow() -> None:
model.stopSpeakerTranscript() model.stopSpeakerTranscript()
def startThreadingTranscriptionReceiveMessageOnCloseConfigWindow(self, run:Callable[[dict], None]) -> None: def startThreadingTranscriptionReceiveMessageOnCloseConfigWindow(self) -> None:
th_startTranscriptionReceiveMessage = Thread(target=self.startTranscriptionReceiveMessageOnCloseConfigWindow, args=(run,)) th_startTranscriptionReceiveMessage = Thread(target=self.startTranscriptionReceiveMessageOnCloseConfigWindow)
th_startTranscriptionReceiveMessage.daemon = True th_startTranscriptionReceiveMessage.daemon = True
th_startTranscriptionReceiveMessage.start() th_startTranscriptionReceiveMessage.start()
@@ -1559,7 +1559,7 @@ class Controller:
th_download.daemon = True th_download.daemon = True
th_download.start() th_download.start()
def init(self, run_dict:dict, *args, **kwargs) -> None: def init(self, *args, **kwargs) -> None:
printLog("Start Initialization") printLog("Start Initialization")
printLog("Start check DeepL API Key") printLog("Start check DeepL API Key")
@@ -1577,7 +1577,7 @@ class Controller:
# check Downloaded CTranslate2 Model Weight # check Downloaded CTranslate2 Model Weight
printLog("Check Downloaded CTranslate2 Model Weight") printLog("Check Downloaded CTranslate2 Model Weight")
if config.USE_TRANSLATION_FEATURE is True and model.checkCTranslatorCTranslate2ModelWeight() is False: if config.USE_TRANSLATION_FEATURE is True and model.checkCTranslatorCTranslate2ModelWeight() is False:
download = self.DownloadCTranslate2ProgressBar(self, run_dict["download_ctranslate2"]) download = self.DownloadCTranslate2ProgressBar(self)
self.startThreadingDownloadCtranslate2Weight(download.set) self.startThreadingDownloadCtranslate2Weight(download.set)
# set Transcription Engine # set Transcription Engine
@@ -1590,7 +1590,7 @@ class Controller:
# check Downloaded Whisper Model Weight # check Downloaded Whisper Model Weight
printLog("Check Downloaded Whisper Model Weight") printLog("Check Downloaded Whisper Model Weight")
if config.USE_WHISPER_FEATURE is True and model.checkTranscriptionWhisperModelWeight() is False: if config.USE_WHISPER_FEATURE is True and model.checkTranscriptionWhisperModelWeight() is False:
download = self.DownloadWhisperProgressBar(self, run_dict["download_whisper"]) download = self.DownloadWhisperProgressBar(self)
self.startThreadingDownloadWhisperWeight(download.set) self.startThreadingDownloadWhisperWeight(download.set)
# set word filter # set word filter
@@ -1616,11 +1616,11 @@ class Controller:
# init Auto device selection # init Auto device selection
printLog("Init Auto Device Selection") printLog("Init Auto Device Selection")
if config.AUTO_MIC_SELECT is True: if config.AUTO_MIC_SELECT is True:
update_mic_device = self.UpdateSelectedMicDevice(self, run_dict["update_selected_mic_device"]) update_mic_device = self.UpdateSelectedMicDevice(self)
device_manager.setCallbackDefaultInputDevice(update_mic_device.set) device_manager.setCallbackDefaultInputDevice(update_mic_device.set)
if config.AUTO_SPEAKER_SELECT is True: if config.AUTO_SPEAKER_SELECT is True:
update_speaker_device = self.UpdateSelectedSpeakerDevice(self, run_dict["update_selected_speaker_device"]) update_speaker_device = self.UpdateSelectedSpeakerDevice(self)
device_manager.setCallbackDefaultOutputDevice(update_speaker_device.set) device_manager.setCallbackDefaultOutputDevice(update_speaker_device.set)
printLog("End Initialization") printLog("End Initialization")

102
src-python/webui_mainloop.py
View File

@@ -1,6 +1,7 @@
import sys import sys
import json import json
import time import time
from typing import Any
from threading import Thread from threading import Thread
from queue import Queue from queue import Queue
from webui_controller import Controller from webui_controller import Controller
@@ -8,6 +9,34 @@ from utils import printLog, printResponse, encodeBase64
controller = Controller() controller = Controller()
run_mapping = {
"download_software":"/run/download_software",
"update_software":"/run/update_software",
"transcription_mic":"/run/transcription_send_mic_message",
"transcription_speaker":"/run/transcription_receive_speaker_message",
"check_mic_volume":"/run/check_mic_volume",
"check_speaker_volume":"/run/check_speaker_volume",
"error_device":"/run/error_device",
"error_translation_engine":"/run/error_translation_engine",
"word_filter":"/run/word_filter",
"download_ctranslate2":"/run/download_ctranslate2_weight",
"download_whisper":"/run/download_whisper_weight",
"selected_mic_device":"/set/data/selected_mic_host",
"selected_speaker_device":"/set/data/selected_speaker_device",
}
controller.setRunMapping(run_mapping)
def run(status:int, endpoint:str, result:Any) -> None:
printResponse(status, endpoint, result)
controller.setRun(run)
mapping = { mapping = {
# Main Window # Main Window
"/set/enable/translation": {"status": False, "variable":controller.setEnableTranslation}, "/set/enable/translation": {"status": False, "variable":controller.setEnableTranslation},
@@ -276,66 +305,6 @@ mapping = {
# "/run/restart_software": {"status": True, "variable":controller.restartSoftware}, # "/run/restart_software": {"status": True, "variable":controller.restartSoftware},
} }
run_mapping = {
"/run/update_software": {
"download":"/run/download_software",
"update":"/run/update_software"
},
"/set/disable/config_window": {
"mic":"/run/transcription_send_mic_message",
"speaker":"/run/transcription_receive_speaker_message",
"error_device":"/run/error_device",
"error_translation_engine":"/run/error_translation_engine",
"word_filter":"/run/word_filter",
},
"/set/enable/transcription_send": {
"mic":"/run/transcription_send_mic_message",
"error_device":"/run/error_device",
"error_translation_engine":"/run/error_translation_engine",
"word_filter":"/run/word_filter",
},
"/set/enable/transcription_receive": {
"speaker":"/run/transcription_receive_speaker_message",
"error_device":"/run/error_device",
"error_translation_engine":"/run/error_translation_engine",
},
"/set/enable/check_mic_threshold": {
"mic":"/run/check_mic_volume",
"error_device":"/run/error_device",
},
"/set/enable/check_speaker_threshold": {
"speaker":"/run/check_speaker_volume",
"error_device":"/run/error_device",
},
"/run/send_message_box": {
"error_translation_engine":"/run/error_translation_engine"
},
"/run/download_ctranslate2_weight": {
"download":"/run/download_ctranslate2_weight"
},
"/run/download_whisper_weight": {
"download":"/run/download_whisper_weight"
},
"/set/enable/auto_mic_select": {
"mic":"/set/data/selected_mic_host",
},
"/set/enable/auto_speaker_select": {
"speaker":"/set/data/selected_speaker_device",
}
}
class Run:
def __init__(self, endpoints:dict) -> None:
self.endpoints = endpoints
def transmit(self, key:str, data:dict) -> None:
if key not in self.endpoints:
printLog("Invalid endpoint", key)
else:
status = data.get("status", None)
result = data.get("result", None)
printResponse(status, self.endpoints[key], result)
class Main: class Main:
def __init__(self) -> None: def __init__(self) -> None:
self.queue = Queue() self.queue = Queue()
@@ -366,12 +335,8 @@ class Main:
response = "Locked endpoint" response = "Locked endpoint"
status = 423 status = 423
else: else:
run_endpoint = run_mapping.get(endpoint, None)
try: try:
if run_endpoint is not None: response = handler["variable"](data)
response = handler["variable"](data, Run(run_endpoint).transmit)
else:
response = handler["variable"](data)
status = response.get("status", None) status = response.get("status", None)
result = response.get("result", None) result = response.get("result", None)
except Exception as e: except Exception as e:
@@ -413,12 +378,7 @@ if __name__ == "__main__":
main.startReceiver() main.startReceiver()
main.startHandler() main.startHandler()
controller.init({ controller.init()
"download_ctranslate2": Run(run_mapping["/run/download_ctranslate2_weight"]).transmit,
"download_whisper": Run(run_mapping["/run/download_whisper_weight"]).transmit,
"update_selected_mic_device": Run(run_mapping["/set/enable/auto_mic_select"]).transmit,
"update_selected_speaker_device": Run(run_mapping["/set/enable/auto_speaker_select"]).transmit,
})
# mappingのすべてのstatusをTrueにする # mappingのすべてのstatusをTrueにする
for key in mapping.keys(): for key in mapping.keys():