forked from mrq/ai-voice-cloning
Merge pull request 'websocket server: API change(!), better response format' (#334) from ben_mkiv/ai-voice-cloning:master into master
Reviewed-on: mrq/ai-voice-cloning#334
This commit is contained in:
commit
5f4215b3ef
|
@ -7,27 +7,32 @@ from websockets.server import serve
|
|||
from utils import generate, get_autoregressive_models, get_voice_list
|
||||
|
||||
|
||||
async def _handle_generate(websocket, message):
|
||||
await websocket.send(json.dumps(generate(**message)))
|
||||
|
||||
|
||||
async def _handle_get_autoregressive_models(websocket, message):
|
||||
await websocket.send(json.dumps(get_autoregressive_models()))
|
||||
|
||||
|
||||
async def _handle_get_voice_list(websocket, message):
|
||||
await websocket.send(json.dumps(get_voice_list()))
|
||||
|
||||
|
||||
# this is a not so nice workaround to set values to None if their string value is "None"
|
||||
def replaceNoneStringWithNone(message):
|
||||
ignore_fields = ['text'] # list of fields which CAN have "None" as literal String value
|
||||
|
||||
for member in message:
|
||||
if message[member] == 'None':
|
||||
if message[member] == 'None' and member not in ignore_fields:
|
||||
message[member] = None
|
||||
|
||||
return message
|
||||
|
||||
|
||||
async def _handle_generate(websocket, message):
|
||||
message['result'] = generate(**message)
|
||||
await websocket.send(json.dumps(replaceNoneStringWithNone(message)))
|
||||
|
||||
|
||||
async def _handle_get_autoregressive_models(websocket, message):
|
||||
message['result'] = get_autoregressive_models()
|
||||
await websocket.send(json.dumps(replaceNoneStringWithNone(message)))
|
||||
|
||||
|
||||
async def _handle_get_voice_list(websocket, message):
|
||||
message['result'] = get_voice_list()
|
||||
await websocket.send(json.dumps(replaceNoneStringWithNone(message)))
|
||||
|
||||
|
||||
async def _handle_message(websocket, message):
|
||||
message = replaceNoneStringWithNone(message)
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user