ez audio transcription tool with flexible processing and post-processing options https://pypi.org/project/wscribe/
Find a file Use this template
2023-11-09 18:03:21 +05:30
.github chore(deps): bump actions/checkout from 3 to 4 2023-10-01 18:03:53 +00:00
docs chore: readme updates 2023-08-06 12:56:46 +05:30
examples chore: add sample output files 2023-08-06 01:35:11 +05:30
scripts Use curl to fetch model only if git lfs is not available (#4) 2023-11-08 11:45:20 +05:30
src/wscribe feat: add vad filter flag 2023-11-09 18:03:21 +05:30
tests chore: refactor and add types 2023-08-05 12:48:37 +05:30
.envrc.example chore: remove unused directories from template 2023-07-25 19:41:59 +05:30
.gitignore chore: squash all wip commits into this >.< 2023-07-25 19:35:14 +05:30
LICENSE Initial commit 2023-07-21 13:29:01 +05:30
Makefile chore: squash all wip commits into this >.< 2023-07-25 19:35:14 +05:30
Makefile.common Initial commit 2023-07-21 13:29:01 +05:30
poetry.lock chore: update faster-whisper version 2023-11-08 13:11:57 +05:30
pyproject.toml chore: publish v0.1.5 2023-11-08 13:32:05 +05:30
README.org fixup! chore: readme updates 2023-08-06 17:32:21 +05:30

wscribe

Getting started

wscribe is yet another easy to use front-end for whisper specifically for transcription. It aims to be modular so that it can support multiple audio sources, processing backends and inference interfaces. It can run both on CPU and GPU based on the processing backend. Once transcript is generated, editing/correction/visualization of the transcript can be done manually with the wscribe-editor.

It was created at sochara because we have a large volume of audio recordings that need to be transcribed and eventually archived. Another important need was that we needed to verify and manually edit the generated transcript, I could not find any open-source tool that checked all the boxes. Suggested workflow is generating word-level transcript(only supported in json export) and then editing the transcript with the wscribe-editor.

Currently, it supports the following. Check roadmap for upcoming support.

  • Processing backend: faster-whisper
  • Audio sources: Local files (Audio/Video)
  • Inference interfaces: Python CLI
  • File exports: JSON, SRT, WebVTT

Installation

These instructions were tested on NixOS:Python3.10 and ArchLinux:Python3.10 but should work for any other OS, if you face any installation issues please feel free to create issues. I'll try to put out a docker image sometime.

1. Set required env var
  • WSCRIBE_MODELS_DIR : Path to the directory where whisper models should be downloaded to.
export WSCRIBE_MODELS_DIR=$XDG_DATA_HOME/whisper-models # example
2. Download the models
Recommended
  • Recommended way for downloading the models is to use the helper script, it'll download the models to WSCRIBE_MODELS_DIR.

    cd /tmp # temporary script, only needed to download the models
    curl https://raw.githubusercontent.com/geekodour/wscribe/main/scripts/fw_dw_hf_wo_lfs.sh
    chmod u+x fw_dw_hf_wo_lfs.sh
    ./fw_dw_hf_wo_lfs.sh tiny # other models: tiny, small, medium and large-v2
Manual

You can download the models directly from here using git lfs, make sure you download/copy them to WSCRIBE_MODELS_DIR

3. Install wscribe

Assuming you already have a working python>=3.10 setup

pip install wscribe

Usage

# wscribe transcribe [OPTIONS] SOURCE DESTINATION

# cpu
wscribe transcribe audio.mp3 transcription.json
# use gpu
wscribe transcribe video.mp4 transcription.json --gpu
# use gpu, srt format
wscribe transcribe video.mp4 transcription.srt -g -f srt
# use gpu, srt format, tiny model
wscribe transcribe video.mp4 transcription.vtt -g -f vtt -m tiny
wscribe transcribe --help # all help info

Numbers

device quant model original playback transcription playback/transcription
cuda float16 tiny 6.3m 0.1m 68x
cuda float16 small 6.3m 0.2m 29x
cuda float16 medium 6.3m 0.4m 14x
cuda float16 large-v2 6.3m 0.8m 7x
cpu int8 tiny 6.3m 0.2m 25x
cpu int8 small 6.3m 1.3m 4x
cpu int8 medium 6.3m 3.6m ~1.7x
cpu int8 large-v2 6.3m 3.6m ~0.9x

Roadmap

Processing Backends

Transcription Features

  • Add support for diarization
  • Add translation
  • Add VAD/other de-noising stuff etc.
  • Add local llm integration with llama.cpp or something similar for summary and othe possible things. It can be also used to generate more accurate transcript. Whisper mostly generates sort of a subtitle, for converting subtitle into transcription we need to group the subtitle. This can be done in various ways. Eg. By speaker if diarization is supported, by time chunks etc. By using LLMs or maybe other NLP techniques we'll also be able to do this with things like break in dialogue etc. Have to explore.

Inference interfaces

  • Python CLI

    • Basic CLI
    • Improve summary statistics
  • REST endpoint

    • Basic server to run wscribe via an API.
    • Possibly add glue code to expose it via CFtunnels or something similar
  • GUI

Audio sources

  • Local files
  • Youtube
  • Google drive

Distribution

  • Python packaging
  • Docker/Podman
  • Package for Nix
  • Package for Arch(AUR)

Contributing

All contribution happens through PRs, any contributions is greatly appreciated, bugfixes are welcome, features are welcome, tests are welcome, suggestions & criticism are welcome.

Testing

  • make test
  • See other helper commands in Makefile