Enhancing Whisper transcriptions: pre- & post-processing techniques
Source: openai_cookbooks

# Enhancing Whisper transcriptions: pre- & post-processing techniquesThis notebook offers a guide to improve the Whisper's transcriptions. We'll streamline your audio data via trimming and segmentation, enhancing Whisper's transcription quality. After transcriptions, we'll refine the output by adding punctuation, adjusting product terminology (e.g., 'five two nine' to '529'), and mitigating Unicode issues. These strategies will help improve the clarity of your transcriptions, but remember, customization based on your unique use-case may be beneficial.## SetupTo get started let's import a few different libraries:- [PyDub](http://pydub.com/) is a simple and easy-to-use Python library for audio processing tasks such as slicing, concatenating, and exporting audio files.- The `Audio` class from the `IPython.display` module allows you to create an audio control that can play sound in Jupyter notebooks, providing a straightforward way to play audio data directly in your notebook.- For our audio file, we'll use a fictional earnings call written by ChatGPT and read aloud by the author.This audio file is relatively short, but hopefully provides you with an illustrative idea of how these pre and post processing steps can be applied to any audio file. ```pythonfrom openai import OpenAIimport osimport urllibfrom IPython.display import Audiofrom pathlib import Pathfrom pydub import AudioSegmentimport ssl``````pythonclient = OpenAI(api_key=os.environ.get("OPENAI_API_KEY", "<your OpenAI API key if not set as env var>"))``````python# set download pathsearnings_call_remote_filepath = "https://cdn.openai.com/API/examples/data/EarningsCall.wav"# set local save locationsearnings_call_filepath = "data/EarningsCall.wav"# download example audio files and save locallyssl._create_default_https_context = ssl._create_unverified_contexturllib.request.urlretrieve(earnings_call_remote_filepath, earnings_call_filepath)```    ('data/EarningsCall.wav', <http.client.HTTPMessage at 0x11be41f50>)At times, files with long silences at the beginning can cause Whisper to transcribe the audio incorrectly. We'll use Pydub to detect and trim the silence. Here, we've set the decibel threshold of 20. You can change this if you would like.```python# Function to detect leading silence# Returns the number of milliseconds until the first sound (chunk averaging more than X decibels)def milliseconds_until_sound(sound, silence_threshold_in_decibels=-20.0, chunk_size=10):    trim_ms = 0  # ms    assert chunk_size > 0  # to avoid infinite loop    while sound[trim_ms:trim_ms+chunk_size].dBFS < silence_threshold_in_decibels and trim_ms < len(sound):        trim_ms += chunk_size    return trim_ms``````pythondef trim_start(filepath):    path = Path(filepath)    directory = path.parent    filename = path.name    audio = AudioSegment.from_file(filepath, format="wav")    start_trim = milliseconds_until_sound(audio)    trimmed = audio[start_trim:]    new_filename = directory / f"trimmed_{filename}"    trimmed.export(new_filename, format="wav")    return trimmed, new_filename``````pythondef transcribe_audio(file,output_dir):    audio_path = os.path.join(output_dir, file)    with open(audio_path, 'rb') as audio_data:        transcription = client.audio.transcriptions.create(            model="whisper-1", file=audio_data)        return transcription.text```At times, we've seen unicode character injection in transcripts, removing any non-ASCII characters should help mitigate this issue.Keep in mind you should not use this function if you are transcribing in Greek, Cyrillic, Arabic, Chinese, etc```python# Define function to remove non-ascii charactersdef remove_non_ascii(text):    return ''.join(i for i in text if ord(i)<128)```This function will add formatting and punctuation to our transcript. Whisper generates a transcript with punctuation but without formatting.```python# Define function to add punctuationdef punctuation_assistant(ascii_transcript):    system_prompt = """You are a helpful assistant that adds punctuation to text.      Preserve the original words and only insert necessary punctuation such as periods,     commas, capialization, symbols like dollar sings or percentage signs, and formatting.     Use only the context provided. If there is no context provided say, 'No context provided'\n"""    response = client.chat.completions.create(        model="gpt-3.5-turbo",        temperature=0,        messages=[            {                "role": "system",                "content": system_prompt            },            {                "role": "user",                "content": ascii_transcript            }        ]    )    return response```Our audio file is a recording from a fake earnings call that includes a lot of financial products. This function can help ensure that if Whisper transcribes these financial product names incorrectly, that they can be corrected. ```python# Define function to fix product mispellingsdef product_assistant(ascii_transcript):    system_prompt = """You are an intelligent assistant specializing in financial products;    your task is to process transcripts of earnings calls, ensuring that all references to     financial products and common financial terms are in the correct format. For each     financial product or common term that is typically abbreviated as an acronym, the full term     should be spelled out followed by the acronym in parentheses. For example, '401k' should be     transformed to '401(k) retirement savings plan', 'HSA' should be transformed to 'Health Savings Account (HSA)'    , 'ROA' should be transformed to 'Return on Assets (ROA)', 'VaR' should be transformed to 'Value at Risk (VaR)', and 'PB' should be transformed to 'Price to Book (PB) ratio'. Similarly, transform spoken numbers representing financial products into their numeric representations, followed by the full name of the product in parentheses. For instance, 'five two nine' to '529 (Education Savings Plan)' and 'four zero one k' to '401(k) (Retirement Savings Plan)'. However, be aware that some acronyms can have different meanings based on the context (e.g., 'LTV' can stand for 'Loan to Value' or 'Lifetime Value'). You will need to discern from the context which term is being referred to and apply the appropriate transformation. In cases where numerical figures or metrics are spelled out but do not represent specific financial products (like 'twenty three percent'), these should be left as is. Your role is to analyze and adjust financial product terminology in the text. Once you've done that, produce the adjusted  transcript and a list of the words you've changed"""    response = client.chat.completions.create(        model="gpt-4",        temperature=0,        messages=[            {                "role": "system",                "content": system_prompt            },            {                "role": "user",                "content": ascii_transcript            }        ]    )    return response```This function will create a new file with 'trimmed' appended to the original file name```python# Trim the start of the original audio filetrimmed_audio = trim_start(earnings_call_filepath)``````pythontrimmed_audio, trimmed_filename = trim_start(earnings_call_filepath)```Our fake earnings report audio file is fairly short in length, so we'll adjust the segments accordingly. Keep in mind you can adjust the segment length as you need.```python# Segment audiotrimmed_audio = AudioSegment.from_wav(trimmed_filename)  # Load the trimmed audio fileone_minute = 1 * 60 * 1000  # Duration for each segment (in milliseconds)start_time = 0  # Start time for the first segmenti = 0  # Index for naming the segmented filesoutput_dir_trimmed = "trimmed_earnings_directory"  # Output directory for the segmented filesif not os.path.isdir(output_dir_trimmed):  # Create the output directory if it does not exist    os.makedirs(output_dir_trimmed)while start_time < len(trimmed_audio):  # Loop over the trimmed audio file    segment = trimmed_audio[start_time:start_time + one_minute]  # Extract a segment    segment.export(os.path.join(output_dir_trimmed, f"trimmed_{i:02d}.wav"), format="wav")  # Save the segment    start_time += one_minute  # Update the start time for the next segment    i += 1  # Increment the index for naming the next file``````python# Get list of trimmed and segmented audio files and sort them numericallyaudio_files = sorted(    (f for f in os.listdir(output_dir_trimmed) if f.endswith(".wav")),    key=lambda f: int(''.join(filter(str.isdigit, f))))``````python# Use a loop to apply the transcribe function to all audio filestranscriptions = [transcribe_audio(file, output_dir_trimmed) for file in audio_files]``````python# Concatenate the transcriptionsfull_transcript = ' '.join(transcriptions)``````pythonprint(full_transcript)```    Good afternoon, everyone. And welcome to FinTech Plus Sync's second quarter 2023 earnings call. I'm John Doe, CEO of FinTech Plus. We've had a stellar Q2 with a revenue of 125 million, a 25% increase year over year. Our gross profit margin stands at a solid 58%, due in part to cost efficiencies gained from our scalable business model. Our EBITDA has surged to 37.5 million, translating to a remarkable 30% EBITDA margin. Our net income for the quarter rose to 16 million, which is a noteworthy increase from 10 million in Q2 2022. Our total addressable market has grown substantially thanks to the expansion of our high yield savings product line and the new RoboAdvisor platform. We've been diversifying our asset-backed securities portfolio, investing heavily in collateralized. debt obligations, and residential mortgage-backed securities. We've also invested $25 million in AAA rated corporate bonds, enhancing our risk adjusted returns. As for our balance sheet, total assets reached $1.5 billion with total liabilities at $900 million, leaving us with a solid equity base of $600 million. Our debt-to-equity ratio stands at 1.5, a healthy figure considering our expansionary phase. We continue to see substantial organic user growth, with customer acquisition cost dropping by 15% and lifetime value growing by 25%. Our LTVCAC ratio is at an impressive 3.5%. In terms of risk management, we have a value-at-risk model in place with a 99%... confidence level indicating that our maximum loss will not exceed 5 million in the next trading day. We've adopted a conservative approach to managing our leverage and have a healthy tier one capital ratio of 12.5%. Our forecast for the coming quarter is positive. We expect revenue to be around 135 million and 8% quarter over quarter growth driven primarily by our cutting edge blockchain solutions and AI driven predictive analytics. We're also excited about the upcoming IPO of our FinTech subsidiary Pay Plus, which we expect to raise 200 million, significantly bolstering our liquidity and paving the way for aggressive growth strategies. We thank our shareholders for their continued faith in us and we look forward to an even more successful Q3. Thank you so much.```python# Remove non-ascii characters from the transcriptascii_transcript = remove_non_ascii(full_transcript)``````pythonprint(ascii_transcript)```    Good afternoon, everyone. And welcome to FinTech Plus Sync's second quarter 2023 earnings call. I'm John Doe, CEO of FinTech Plus. We've had a stellar Q2 with a revenue of 125 million, a 25% increase year over year. Our gross profit margin stands at a solid 58%, due in part to cost efficiencies gained from our scalable business model. Our EBITDA has surged to 37.5 million, translating to a remarkable 30% EBITDA margin. Our net income for the quarter rose to 16 million, which is a noteworthy increase from 10 million in Q2 2022. Our total addressable market has grown substantially thanks to the expansion of our high yield savings product line and the new RoboAdvisor platform. We've been diversifying our asset-backed securities portfolio, investing heavily in collateralized. debt obligations, and residential mortgage-backed securities. We've also invested $25 million in AAA rated corporate bonds, enhancing our risk adjusted returns. As for our balance sheet, total assets reached $1.5 billion with total liabilities at $900 million, leaving us with a solid equity base of $600 million. Our debt-to-equity ratio stands at 1.5, a healthy figure considering our expansionary phase. We continue to see substantial organic user growth, with customer acquisition cost dropping by 15% and lifetime value growing by 25%. Our LTVCAC ratio is at an impressive 3.5%. In terms of risk management, we have a value-at-risk model in place with a 99%... confidence level indicating that our maximum loss will not exceed 5 million in the next trading day. We've adopted a conservative approach to managing our leverage and have a healthy tier one capital ratio of 12.5%. Our forecast for the coming quarter is positive. We expect revenue to be around 135 million and 8% quarter over quarter growth driven primarily by our cutting edge blockchain solutions and AI driven predictive analytics. We're also excited about the upcoming IPO of our FinTech subsidiary Pay Plus, which we expect to raise 200 million, significantly bolstering our liquidity and paving the way for aggressive growth strategies. We thank our shareholders for their continued faith in us and we look forward to an even more successful Q3. Thank you so much.```python# Use punctuation assistant functionresponse = punctuation_assistant(ascii_transcript)``````python# Extract the punctuated transcript from the model's responsepunctuated_transcript = response.choices[0].message.content``````pythonprint(punctuated_transcript)```    Good afternoon, everyone. And welcome to FinTech Plus Sync's second quarter 2023 earnings call. I'm John Doe, CEO of FinTech Plus. We've had a stellar Q2 with a revenue of $125 million, a 25% increase year over year. Our gross profit margin stands at a solid 58%, due in part to cost efficiencies gained from our scalable business model. Our EBITDA has surged to $37.5 million, translating to a remarkable 30% EBITDA margin. Our net income for the quarter rose to $16 million, which is a noteworthy increase from $10 million in Q2 2022. Our total addressable market has grown substantially thanks to the expansion of our high yield savings product line and the new RoboAdvisor platform. We've been diversifying our asset-backed securities portfolio, investing heavily in collateralized debt obligations, and residential mortgage-backed securities. We've also invested $25 million in AAA rated corporate bonds, enhancing our risk-adjusted returns. As for our balance sheet, total assets reached $1.5 billion with total liabilities at $900 million, leaving us with a solid equity base of $600 million. Our debt-to-equity ratio stands at 1.5, a healthy figure considering our expansionary phase. We continue to see substantial organic user growth, with customer acquisition cost dropping by 15% and lifetime value growing by 25%. Our LTVCAC ratio is at an impressive 3.5%. In terms of risk management, we have a value-at-risk model in place with a 99% confidence level indicating that our maximum loss will not exceed $5 million in the next trading day. We've adopted a conservative approach to managing our leverage and have a healthy tier one capital ratio of 12.5%. Our forecast for the coming quarter is positive. We expect revenue to be around $135 million and 8% quarter over quarter growth driven primarily by our cutting-edge blockchain solutions and AI-driven predictive analytics. We're also excited about the upcoming IPO of our FinTech subsidiary Pay Plus, which we expect to raise $200 million, significantly bolstering our liquidity and paving the way for aggressive growth strategies. We thank our shareholders for their continued faith in us and we look forward to an even more successful Q3. Thank you so much.```python# Use product assistant functionresponse = product_assistant(punctuated_transcript)``````python# Extract the final transcript from the model's responsefinal_transcript = response.choices[0].message.content``````pythonprint(final_transcript)```    Good afternoon, everyone. And welcome to FinTech Plus Sync's second quarter 2023 earnings call. I'm John Doe, CEO of FinTech Plus. We've had a stellar second quarter (Q2) with a revenue of $125 million, a 25% increase year over year. Our gross profit margin stands at a solid 58%, due in part to cost efficiencies gained from our scalable business model. Our Earnings Before Interest, Taxes, Depreciation, and Amortization (EBITDA) has surged to $37.5 million, translating to a remarkable 30% EBITDA margin. Our net income for the quarter rose to $16 million, which is a noteworthy increase from $10 million in second quarter (Q2) 2022. Our total addressable market has grown substantially thanks to the expansion of our high yield savings product line and the new RoboAdvisor platform. We've been diversifying our asset-backed securities portfolio, investing heavily in Collateralized Debt Obligations (CDOs), and Residential Mortgage-Backed Securities (RMBS). We've also invested $25 million in AAA rated corporate bonds, enhancing our risk-adjusted returns. As for our balance sheet, total assets reached $1.5 billion with total liabilities at $900 million, leaving us with a solid equity base of $600 million. Our Debt-to-Equity (D/E) ratio stands at 1.5, a healthy figure considering our expansionary phase. We continue to see substantial organic user growth, with Customer Acquisition Cost (CAC) dropping by 15% and Lifetime Value (LTV) growing by 25%. Our LTV to CAC (LTVCAC) ratio is at an impressive 3.5%. In terms of risk management, we have a Value at Risk (VaR) model in place with a 99% confidence level indicating that our maximum loss will not exceed $5 million in the next trading day. We've adopted a conservative approach to managing our leverage and have a healthy Tier 1 Capital ratio of 12.5%. Our forecast for the coming quarter is positive. We expect revenue to be around $135 million and 8% quarter over quarter growth driven primarily by our cutting-edge blockchain solutions and AI-driven predictive analytics. We're also excited about the upcoming Initial Public Offering (IPO) of our FinTech subsidiary Pay Plus, which we expect to raise $200 million, significantly bolstering our liquidity and paving the way for aggressive growth strategies. We thank our shareholders for their continued faith in us and we look forward to an even more successful third quarter (Q3). Thank you so much.        Words Changed:    1. Q2 -> second quarter (Q2)    2. EBITDA -> Earnings Before Interest, Taxes, Depreciation, and Amortization (EBITDA)    3. Q2 2022 -> second quarter (Q2) 2022    4. CDOs -> Collateralized Debt Obligations (CDOs)    5. RMBS -> Residential Mortgage-Backed Securities (RMBS)    6. D/E -> Debt-to-Equity (D/E)    7. CAC -> Customer Acquisition Cost (CAC)    8. LTV -> Lifetime Value (LTV)    9. LTVCAC -> LTV to CAC (LTVCAC)    10. VaR -> Value at Risk (VaR)    11. IPO -> Initial Public Offering (IPO)    12. Q3 -> third quarter (Q3)