import streamlit as st
from transformers import pipeline
import io
from pydub import AudioSegment





pipe1 = pipeline("automatic-speech-recognition", model="jonatasgrosman/wav2vec2-large-xlsr-53-english")
pipe2 = pipeline("text-classification", model="Mintiny/CustomModel_yelp1.1")

st.title("Customer Review Audio Analysis🍔🍣🍝🍜")
with st.form(key="my_form"):
    f = st.file_uploader("Upload Recordings", type = ["mp3"])
    submit_button = st.form_submit_button(label="Start")
            
        
if f is not None:
    audio_data =f.getvalue()
    st.audio(f,format="mp3")
    text=pipe1(audio_data)['text']
    label=pipe2(text)[0]['label']
    if label == 'LABEL_1':
        st.write('The review is positive.')
    else:
        st.write('The review is negative.')