import streamlit as st
from speechbrain.inference.interfaces import foreign_class

# Initialize the classifier
classifier = foreign_class(source="speechbrain/emotion-recognition-wav2vec2-IEMOCAP", pymodule_file="custom_interface.py", classname="CustomEncoderWav2vec2Classifier")

def emotion(file):
    if file is not None:
        # Classify the file
        out_prob, score, index, text_lab = classifier.classify_file(file.name)
        # Display the output
        st.write(text_lab)
    else:
        st.write("Please upload a file.")

def main():
    st.title("Emotion Recognition")
    uploaded_file = st.file_uploader("Upload audio file", type=["wav"])
    if uploaded_file is not None:
        emotion(uploaded_file)

if __name__ == "__main__":
    main()