from sources.text_to_speech import Speech
from sources.utility import pretty_print
from sources.router import AgentRouter

class Interaction:
    def __init__(self, agents, tts_enabled: bool = False, recover_last_session: bool = False):
        self.tts_enabled = tts_enabled
        self.agents = agents
        self.current_agent = None
        self.router = AgentRouter(self.agents)
        self.speech = Speech()
        self.is_active = True
        self.last_query = None
        self.last_answer = None
        if tts_enabled:
            self.speech.speak("Hello Sir, we are online and ready. What can I do for you ?")
        if recover_last_session:
            self.recover_last_session()
    
    def recover_last_session(self):
        for agent in self.agents:
            agent.memory.load_memory()

    def is_active(self):
        return self.is_active
    
    def read_stdin(self) -> str:
        buffer = ""

        while buffer == "" or buffer.isascii() == False:
            try:
                buffer = input(f">>> ")
            except EOFError:
                return None
            if buffer == "exit" or buffer == "goodbye":
                return None
        return buffer

    def get_user(self):
        query = self.read_stdin()
        if query is None:
            self.is_active = False
            self.last_query = "Goodbye (exit requested by user, dont think, make answer very short)"
            return None
        self.last_query = query
        return query
    
    def think(self):
        agent = self.router.select_agent(self.last_query)
        if self.current_agent != agent:
            self.current_agent = agent
            # get history from previous agent
            self.current_agent.memory.push('user', self.last_query)
        self.last_answer, _ = agent.process(self.last_query, self.speech)
    
    def show_answer(self):
        self.current_agent.show_answer()
        if self.tts_enabled:
            self.speech.speak(self.last_answer)