from appPublic.worker import get_event_loop
from fw.init import load_fastwhisper
from ahserver.webapp import webapp
from ahserver.serverenv import ServerEnv

from appPublic.registerfunction import RegisterFunction

async def transcrible(request, *args, **kw):
	env = request._run_ns
	ret = await env.fastwhister.submit_task(env.params_kw)
	return ret

async def get_status(request, *args, **kw):
	env = request._run_ns
	ret = await env.fastwhister.get_status(env.params_kw.task_id)
	return ret
	
async def ahapp_built(app):
	env = ServerEnv()
	app.on_startup.append(env.fastwhisper.run)

def init():
	rf = RegisterFunction()
	rf.register('transcrible', transcrible)
	rf.register('ahapp_built', ahapp_built)
	rf.register('get_status', get_status)
	load_fastwhisper()

if __name__ == '__main__':
	loop = get_event_loop()
	webapp(init)