# version - ArticMonkey:19.03.24:1743

import psutil

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

audio = audiorecorder("Record", "Stop")

def main():
    if len(audio) > 0:
        a2t = A2T(audio)
        print(type(audio))
        if st.button("Get text"):
            with st.spinner("Wait for it...."):
                text = a2t.predict()
                print("Text output in main : ", text)
                st.write(text)
            st.success("Done")
            del audio
            del a2t
        else:
            print("error audio is None")

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