from align import AlignEngine
from ahserver.serverenv import ServerEnv
from ahserver.webapp import webapp
from ahserver.filestorage import FileStorage
from appPublic.worker import awaitify
from appPublic.jsonConfig import getConfig
from appPublic.registerfunction import RegisterFunction
from appPublic.log import debug, exception

async def lyric_align(request, params_kw, *params, **kw):
	audio_webpath = params_kw.audio_path
	text = params_kw.text
	debug(f'{params_kw=}')
	if audio_webpath is None:
		exception(f'{params_kw=}')
		return {
			"status": "error",
			"data": {
				"message": "audio_path is None"
			}
		}
	if text is None:
		exception(f'{params_kw=}')
		return {
			"status": "error",
			"data": {
				"message": "text is None"
			}
		}
	env = ServerEnv()
	fs = FileStorage()
	audio_path = fs.realPath(audio_webpath)
	align = awaitify(env.align_engine.align)
	text = text.replace(" ", "").replace('\t', "")
	text1 = text.replace("\n", "")
	s = await align(audio_path, text1)
	debug(f'{s=}, {text1=},{text=}')
	lines = text.split('\n')
	c_pos = 0
	c_max = len(s) - 1
	sentences = []
	for l in lines:
		if l:
			segment={
				'sentence': l,
				'start': s[c_pos]['start'],
				'chars':[]
			}
			for c in l:
				if c_pos <= c_max:
					segment['chars'].append(s[c_pos])
				else:
					debug(f'{l=}, {c=}, {c_max=}')
				c_pos += 1

			segment['end'] = s[c_pos -1]['end']
			sentences.append(segment)
	return sentences		   

def init():
	rf = RegisterFunction()
	rf.register('align', lyric_align)
	env = ServerEnv()
	config = getConfig()
	env.align_engine = AlignEngine(config.align_model)

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