# version - ArticMonkey:19.03.24:1743

import psutil

import numpy as np

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

def main():
    audio = st_audiorec()
    
    if audio is not None:
        print(type(audio))
        if st.button("Get text"):
            with st.spinner("Wait for it...."):
                a2t = A2T(audio)
                text = a2t.predict()
                print("Text output in main : ", text)
                st.write(text)
            st.success("Done")
            del audio
        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()