Spaces:
Running
Running
File size: 3,033 Bytes
c09dbef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
from openai import OpenAI
import streamlit as st
def audio_transcription(api_key, audio_file):
api_key = api_key
base_url = "https://api.siliconflow.cn/v1"
client = OpenAI(api_key=api_key, base_url=base_url)
transcription = client.audio.transcriptions.create(
model="FunAudioLLM/SenseVoiceSmall",
file=audio_file
)
return transcription
def audioText(api_key: str):
if "uploaded_audio" not in st.session_state:
st.session_state.uploaded_audio = None
if "input_audio" not in st.session_state:
st.session_state.input_audio = None
if st.session_state.uploaded_audio == None and st.session_state.input_audio == None:
disable = True
elif st.session_state.uploaded_audio != None and st.session_state.input_audio == None:
disable = False
elif st.session_state.uploaded_audio == None and st.session_state.input_audio != None:
disable = False
audio_uploader = st.file_uploader("Upload an Audio", type=["MP3", "WAV"], key="audio_uploader", disabled=st.session_state.input_audio!=None)
if audio_uploader is not None:
st.session_state.uploaded_audio = audio_uploader
elif audio_uploader is None:
st.session_state.uploaded_audio = None
audio_input = st.audio_input("Record an Audio", key="audio_input", disabled=st.session_state.uploaded_audio!=None)
if audio_input is not None:
st.session_state.input_audio = audio_input
elif audio_input is None:
st.session_state.input_audio = None
transcript_btn = st.button("Transcript", "transcript_btn", type="primary", disabled=disable)
transcription_str = ""
if transcript_btn:
if st.session_state.uploaded_audio is not None and st.session_state.input_audio is None:
try:
with st.spinner("Processing..."):
transcription = audio_transcription(api_key, st.session_state.uploaded_audio)
if transcription:
transcription_str = transcription.text
except Exception as e:
st.error(f"Error occured: {e}")
elif st.session_state.uploaded_audio is None and st.session_state.input_audio is not None:
try:
with st.spinner("Processing..."):
transcription = audio_transcription(api_key, st.session_state.input_audio)
if transcription:
transcription_str = transcription.text
except Exception as e:
st.error(f"Error occured: {e}")
elif st.session_state.uploaded_audio is None and st.session_state.input_audio is None:
st.info("Please upload an audio or record an audio!")
if transcription_str:
with st.container(border=True, key="trans_container"):
st.markdown(transcription_str)
if __name__ == "__main__":
audioText("sk-hvnjkojhpyitxfeqwdpynrsacitcvqffprrrzzgrwytpebrf")
|