Spaces:
Sleeping
Sleeping
File size: 4,047 Bytes
db70c95 |
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 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
# modules/stt_handler.py
import speech_recognition as sr
import os
import tempfile
from pydub import AudioSegment
def transcribe_audio(audio_filepath):
"""Transcribe audio with multiple fallback methods"""
if not audio_filepath or not os.path.exists(audio_filepath):
print("❌ STT Error: No audio file provided or file does not exist.")
return "Sorry, I couldn't process your audio file. Please try recording again."
print(f"🎙️ Transcribing audio file: {audio_filepath}")
recognizer = sr.Recognizer()
try:
# Try to convert audio format if needed
audio_data = None
# First try direct speech recognition
try:
with sr.AudioFile(audio_filepath) as source:
audio_data = recognizer.record(source)
print("✅ Audio file loaded successfully")
except Exception as audio_load_error:
print(f"⚠️ Direct audio loading failed: {audio_load_error}")
# Try converting with pydub first
try:
print("🔄 Converting audio format...")
audio = AudioSegment.from_file(audio_filepath)
# Export as WAV for better compatibility
with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_wav:
temp_wav_path = temp_wav.name
audio.export(temp_wav_path, format="wav")
with sr.AudioFile(temp_wav_path) as source:
audio_data = recognizer.record(source)
print("✅ Audio converted and loaded successfully")
# Clean up temp file
if os.path.exists(temp_wav_path):
os.remove(temp_wav_path)
except Exception as convert_error:
print(f"❌ Audio conversion failed: {convert_error}")
return "Sorry, I couldn't process your audio format. Please try recording again."
if not audio_data:
return "Sorry, I couldn't load your audio. Please try recording again."
# Try Whisper transcription
try:
print("🤖 Transcribing with Whisper...")
text = recognizer.recognize_whisper(audio_data, language="english")
print(f"✅ Transcription successful: {text[:100]}...")
return text if text.strip() else "I didn't catch what you said. Could you please speak more clearly?"
except sr.UnknownValueError:
print("⚠️ Whisper could not understand the audio")
return "I couldn't understand what you said. Please speak more clearly and try again."
except sr.RequestError as e:
print(f"⚠️ Whisper service error: {e}")
# Fallback to Google Web Speech API
try:
print("🔄 Falling back to Google Speech Recognition...")
text = recognizer.recognize_google(audio_data)
print(f"✅ Google transcription successful: {text[:100]}...")
return text if text.strip() else "I didn't catch what you said. Could you please try again?"
except Exception as google_error:
print(f"❌ Google fallback failed: {google_error}")
return "I'm having trouble with speech recognition. Please try again or check your microphone."
except Exception as e:
print(f"❌ Unexpected transcription error: {e}")
return f"Sorry, I encountered an error processing your audio. Please try recording again."
finally:
# Clean up the original audio file
if os.path.exists(audio_filepath):
try:
os.remove(audio_filepath)
print(f"🗑️ Cleaned up audio file: {audio_filepath}")
except OSError as e:
print(f"⚠️ Error deleting temp audio file {audio_filepath}: {e}") |