sports-analyzer / tts.py
Reynold97's picture
Upload folder using huggingface_hub
3f2dca0 verified
import os
import shutil
from datetime import datetime
from dotenv import load_dotenv
from deepgram import DeepgramClient, PrerecordedOptions
load_dotenv()
API_KEY = os.getenv('DEEPGRAM_API_KEY')
deepgram = DeepgramClient(API_KEY)
def ensure_dir(directory):
"""Ensure the specified directory exists; if not, create it."""
if not os.path.exists(directory):
os.makedirs(directory)
def transcribe_audio(audio_path):
"""Transcribes audio using Deepgram's API and saves the audio file with a unique timestamp."""
try:
audio_storage_dir = "permanent_audio_files"
ensure_dir(audio_storage_dir)
# Generate a unique filename with a timestamp
timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
filename = f"audio_{timestamp}.wav"
permanent_audio_path = os.path.join(audio_storage_dir, filename)
shutil.copy(audio_path, permanent_audio_path)
print(f"Saved audio file to: {permanent_audio_path}") # Debug: Confirm file save
options = PrerecordedOptions(model="nova-2", smart_format=True, language="en-AU")
# Open the permanently stored audio file for transcription
with open(permanent_audio_path, 'rb') as audio:
source = {'buffer': audio}
response = deepgram.listen.prerecorded.v("1").transcribe_file(source, options)
transcript = response['results']['channels'][0]['alternatives'][0]['transcript']
return transcript
except Exception as e:
print("Error during transcription:", e)
raise e
return str(e)