app.py

import streamlit as st
from st_audiorec import st_audiorec

from nameder import init_model_ner, get_entity_labels
from speech2text import init_model_trans, transcribe
from resources import audit_elapsedtime, set_start
import subprocess

def main ():
    print("------------------------------")
    print(f"Running main")

    print(subprocess.Popen('pip list', shell=True))
    s2t = init_model_trans()
    ner = init_model_ner() #async

    print("Rendering UI...")
    start_render = set_start()
    wav_audio_data = st_audiorec()
    audit_elapsedtime(function="Rendering UI", start=start_render)

    if wav_audio_data is not None and s2t is not None:
        print("Loading data...")
        start_loading = set_start()
        st.audio(wav_audio_data, format='audio/wav')
        text = transcribe(wav_audio_data, s2t)

        if text is not None and ner is not None:    
            st.write('Entities: ', get_entity_labels(model=ner, text=text))
        audit_elapsedtime(function="Loading data", start=start_loading)

if __name__ == "__main__":
    print("IN __name__")
    main()