import streamlit as st
import whisper

st.title("Audio Transcript with Whisper")

audio_file = st.file_uploader("Upload Audio", type=["mp3", "wav", "m4a"])

if audio_file is not None:
    with open(audio_file.name, "wb") as f:
        f.write(audio_file.getbuffer())
    st.sidebar.success("File saved!")
    model = whisper.load_model("large")
    st.text("Whisper Model Loaded")
    
if st.button("Transcribe Audio"):
    if audio_file is not None:
        st.success("Transcribing Audio file")
        transcript = model.transcribe(audio_file.name)
        st.success("Transcription Complete")
        st.markdown(transcript["text"])
    else:
        st.error("Please upload an audio file")

st.sidebar.header("Play Audio file")
if audio_file is not None:
    st.sidebar.audio(audio_file)
else:
    st.sidebar.warning("No audio file uploaded")