# version - ArticMonkey:19.03.24:1743

import psutil

import streamlit as st 
from audio2text.a2t import A2T
from audiorecorder import audiorecorder

def main():
    audio = audiorecorder("Record", "Stop")
    
    if len(audio) > 0:
        a2t = A2T(audio)
        if st.button("Get text"):
            with st.spinner("Wait for it...."):
                text = a2t.predict()
                st.write(text)
            st.success("Done")

    del audio
    del a2t

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