Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/sivakumar-mahalingam/subtitle-generator
๐๏ธ Automatically generating subtitles for video files using Whisper ASR model in Python
https://github.com/sivakumar-mahalingam/subtitle-generator
ai audio-model audio-processing automatic-speech-recognition openai-whisper python speech-recognition speech-to-text subtitle-generator whisper
Last synced: 3 months ago
JSON representation
๐๏ธ Automatically generating subtitles for video files using Whisper ASR model in Python
- Host: GitHub
- URL: https://github.com/sivakumar-mahalingam/subtitle-generator
- Owner: sivakumar-mahalingam
- License: mit
- Created: 2024-06-15T09:37:00.000Z (6 months ago)
- Default Branch: main
- Last Pushed: 2024-07-07T10:11:17.000Z (6 months ago)
- Last Synced: 2024-09-27T06:22:21.369Z (3 months ago)
- Topics: ai, audio-model, audio-processing, automatic-speech-recognition, openai-whisper, python, speech-recognition, speech-to-text, subtitle-generator, whisper
- Language: Python
- Homepage:
- Size: 93.1 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Subtitle Generator
This repository provides a Python script to generate subtitle files (`.srt`) for a given video file. The script extracts audio from the video, transcribes the audio using OpenAI's Whisper model, and generates an `.srt` file with the transcription and timestamps.
## SRT Files
SRT (.srt) files are the most common type of closed caption file format. SRT stands for โSubRip Subtitleโ file.
An SRT file includes:
- The number of the closed caption frame in sequence
- Beginning and end timecodes for when the closed caption frame should appear
- The closed caption itself
- A blank link to indicate the start of a new closed caption sequence![img.png](docs%2Fimg.png)
## Features
- Extracts audio from video files using `ffmpeg`.
- Transcribes audio to text with timestamps using OpenAI's Whisper model.
- Generates `.srt` subtitle files.## Requirements
- `Python 3.7+`
- `ffmpeg`
- `torch`
- `openai-whisper`## Installation
1. **Install `ffmpeg`**:
Make sure `ffmpeg` is installed on your system. You can download it from [ffmpeg.org](https://ffmpeg.org/download.html) and follow the installation instructions for your operating system.
2. **Clone the repository**:
```bash
git clone https://github.com/yourusername/subtitle-generator.git
cd subtitle-generator
```3. **Create a virtual environment and activate it**:
```bash
python -m venv venv
source venv/bin/activate # On Windows use `venv\Scripts\activate`
```4. **Install the required Python packages**:
```bash
pip install -r requirements.txt
```## Usage
1. **Provide the path to the video file**:
Ensure you have the video file path ready. The script will process this file to generate subtitles.
2. **Run the script**:
Update all file paths (.srt, .wav, .mp4) in the script
```bash
python generate_subtitles.py
```3. **Output**:
The script will create a `.srt` file with the same name as the video file in the same directory.
## Script Overview
### `generate_subtitles.py`
This is the main script that performs the following steps:
1. **Extract audio from the video file**:
Uses `ffmpeg` to extract audio from the provided video file and save it as a `.wav` file.2. **Transcribe the audio file**:
Utilizes OpenAI's Whisper model to transcribe the audio into text with timestamps.3. **Create the `.srt` file**:
Generates an `.srt` file with the transcribed text and timestamps.## License
This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for details.
## Contributing
Contributions are welcome! Please open an issue or submit a pull request for any improvements or bug fixes.
## Acknowledgements
- [OpenAI Whisper](https://github.com/openai/whisper) for the transcription model.
- [ffmpeg](https://ffmpeg.org/) for audio extraction.