[Update] Refactor transliteration function to accept parameters for hiragana and romaji conversion

This commit is contained in:
misyaguziya
2025-09-12 16:06:12 +09:00
parent 484d201cb0
commit 5a9a13146c
2 changed files with 32 additions and 12 deletions

View File

@@ -275,13 +275,21 @@ class Model:
self.previous_receive_message = message
return repeat_flag
def convertMessageToTransliteration(self, message: str) -> str:
def convertMessageToTransliteration(self, message: str, hiragana: bool=True, romaji: bool=True) -> str:
if hiragana is False and romaji is False:
return message
keys_to_keep = {"orig"}
if hiragana:
keys_to_keep.add("hira")
if romaji:
keys_to_keep.add("hepburn")
data_list = self.kks.convert(message)
keys_to_keep = {"orig", "hira", "hepburn"}
filtered_list = []
for item in data_list:
filtered_item = {key: value for key, value in item.items() if key in keys_to_keep}
filtered_list.append(filtered_item)
filtered_list = [
{key: value for key, value in item.items() if key in keys_to_keep}
for item in data_list
]
return filtered_list
def setOscIpAddress(self, ip_address):