import SpeechRecognition
import Text2Speech
import SmallTalk
import Media
from Command import Command
import config
import QA

listener = SpeechRecognition.SpeechToText()
voice    = Text2Speech.Engine()
threads  = []
memory   = []
online   = True
voids    = 0
listener.listen_noise()

def reply(responce):
    if responce['text']:                                #   print answer
        print('Archie: '+responce['text'])
    if responce['voice']:                               #   say answer
        voice.generate(responce['voice']).speak()
    if responce['type'] == 'background':                #   add background thread to list
        threads.append(responce['thread'])

def check_threads():
    for thread in threads:
        if thread['finish_event'].is_set():
            responce = thread['thread'].join()
            reply(responce)
            thread['finish_event'].clear()
            del thread

while True:                                             #    main loop
    check_threads()
    print('\nYou: ', end='')
    # input
    speech = listener.listen()
    text   = speech['text']
    if speech['status'] == 'ok':
        print(text)
        voids = 0
        # set online add clean input if name in text
        if name := list(set(config.names) & set(text.split(' '))):
            online = True
            text   = text.replace(name[0], '').strip()
        # recognize and execute command
        if online:
            # repeat last answer
            if Command.isRepeat(text):
                reply(memory[0]['responce']);
                continue
            # recognize command with context
            try:
                cmd, params = memory[0]['cmd'].checkContext(text).values()
                if memory[0].get('params'): params = {**memory[0].get('params'), **params}
            except:
                cmd, params = Command.reg_find(text).values()
            # execute command
            responce = cmd.start(params)
            # say answer
            reply(responce)
            # waiting answer on question
            if responce['type'] == 'question':
                print('\nYou: ', end='')
                speech = listener.listen()
                if speech['status'] == 'ok':
                    text = speech['text']
                    print(text)
                    if responce:= responce['callback'].answer(text): reply(responce)
            # remember the command
            memory.insert(0, {
                'text': text,
                'cmd':  cmd,
                'responce': responce,
            })
    else:
        if speech['status'] == 'error': print('Отсутсвует подключение к интернету');
        elif online and speech['status']  == 'void': voids += 1;
        if online and voids >= 3: online = False; voids = 0
        if not online: listener.listen_noise()