AbuseDetection / audio_processing.py
time4et's picture
Upload 4 files
dd76fbf verified
raw
history blame
552 Bytes
import whisper
import tempfile
from pydub import AudioSegment
import os
# Set FFmpeg path manually (if needed)
AudioSegment.converter = r"C:\ffmpeg\bin\ffmpeg.exe"
def transcribe_audio(audio_file):
model = whisper.load_model("base")
# Convert to WAV if necessary
with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_wav:
audio = AudioSegment.from_file(audio_file)
audio.export(temp_wav.name, format="wav")
result = model.transcribe(temp_wav.name)
return result["text"]