# version - ArticMonkey:19.03.24:1743

import psutil

import streamlit as st 
from audio2text.a2t import A2T
from llm.llm import LLM_chain
from streamlit_mic_recorder import mic_recorder

llmchain = LLM_chain()

def main():
    mic = mic_recorder(start_prompt="Record",stop_prompt="Stop", just_once=True)

    if mic is not None:
        a2t = A2T(mic["bytes"])
        text = a2t.predict()
        st.write(text)
        response = llmchain(entity=text, id=0)
        st.write(response if response is not None else "Nothing")
        del response
        

if __name__ == "__main__":
    print('RAM memory % used:', psutil.virtual_memory()[2])
    print('RAM Used (GB):', psutil.virtual_memory()[3]/1000000000)
    main()