Skip to main content

AI-Powered Subtitle Generation with Translation

Project description

AI Sub: AI-Powered Subtitle Generation with Translation

PyPI version Downloads


Table of Contents

Overview

AI Sub is a command-line tool that leverages Google's Gemini models to generate high-quality, audio-synchronized subtitles. It is designed to produce dual-language subtitles (Original + English translation) by analyzing both audio and visual cues.


Showcase

Please visit ai-sub-showcase.

Video Screenshot


How It Works

  1. Segmentation: Splits the input video into manageable 5-minute segments.
  2. Re-encoding (Optional): Compresses segments (e.g., 1fps, 360p) to significantly reduce bandwidth usage and upload times without sacrificing AI analysis accuracy.
  3. Upload (Optional): Uploads segments to the Gemini Files API for cloud-based processing. This bypasses local processing constraints and leverages the API's multimodal capabilities.
  4. Lyrics Search and Scene Detection (Optional): Detects scenes and performs a web search for official lyrics to improve transcription accuracy for songs.
  5. Generation: Generates synchronized subtitles and translations by using the audio as the ground truth for timing and visuals for context.
  6. Assembly: Stitches the generated subtitles back together into a final SRT file.

Installation

Prerequisites: Python 3.10 or higher.

  1. Set up a Python virtual environment:

    python -m venv venv
    source venv/bin/activate  # On Windows, use `venv\Scripts\activate.bat`
    
  2. Install AI Sub:

    pip install --upgrade ai-sub
    

Usage

You can use AI Sub a Google AI Studio API Key and an optional langsearch API key for song lyrics web search.

Option 1 (Recommended): Google API Key + Langsearch API Key

This option provides the best quality for the free tier.

  • As of 3 June 2026:
  • Gemini API provides 20 free requests/day for gemini-3.5-flash
  • Gemini API provides 500 free requests/day for gemini-3.1-flash-lite
  • Langsearch provides 1000 free request/day for web searches

The limiting factor is the daily free quota for gemini-3.5-flash. You should be able to create subtitles for between 1 hour to 1 hour 40 mins of videos per day.

  1. Obtain your API Keys:

    • Sign in to Google AI Studio.

    • Click "Create API Key".

    • Sign in to Langsearch.

    • API Keys -> Create API Key

    • Copy and securely store your keys. Never disclose your API key publicly.

  2. Run the application:

    Linux:

    GOOGLE_KEY="YOUR_GOOGLE_API_KEY"
    LANGSEARCH_KEY="YOUR_LANGSEARCH_API_KEY"
    FILE_NAME="path/to/your/video.mp4"
    
    ai-sub \
    --ai.model-lyrics=google-gla:gemini-3.1-flash-lite \
    --ai.model-subtitles=google-gla:gemini-3.5-flash \
    --ai.search.web-search-tool=langsearch \
    --ai.google.key="${GOOGLE_KEY}" \
    --ai.search.key="${LANGSEARCH_KEY}" \
    "${FILE_NAME}"
    

    Windows:

    SET "GOOGLE_KEY=YOUR_GOOGLE_API_KEY"
    SET "LANGSEARCH_KEY=YOUR_LANGSEARCH_API_KEY"
    SET "FILE_NAME=path/to/your/video.mp4"
    
    ai-sub ^
    --ai.model-lyrics=google-gla:gemini-3.1-flash-lite ^
    --ai.model-subtitles=google-gla:gemini-3.5-flash ^
    --ai.search.web-search-tool=langsearch ^
    --ai.google.key="%GOOGLE_KEY%" ^
    --ai.search.key="%LANGSEARCH_KEY%" ^
    "%FILE_NAME%"
    

Option 2: Using Google API Key Only

This is the easiest option if you don't want to bother with creating a Langsearch API key. We can use duckduckgo to try to lookup lyrics, but it is not as reliable.

  1. Obtain your API Key:

    • Sign in to Google AI Studio.
    • Click "Create API Key".
    • Copy and securely store your key. Never disclose your API key publicly.
  2. Run the application:

    GOOGLE_KEY="YOUR_GOOGLE_API_KEY"
    FILE_NAME="path/to/your/video.mp4"
    
    ai-sub \
    --ai.model-lyrics=google-gla:gemini-3.1-flash-lite \
    --ai.model-subtitles=google-gla:gemini-3.5-flash \
    --ai.search.web-search-tool=duckduckgo \
    --ai.google.key="${GOOGLE_KEY}" \
    "${FILE_NAME}"
    

    Windows:

    SET "GOOGLE_KEY=YOUR_GOOGLE_API_KEY"
    SET "FILE_NAME=path/to/your/video.mp4"
    
    ai-sub ^
    --ai.model-lyrics=google-gla:gemini-3.1-flash-lite ^
    --ai.model-subtitles=google-gla:gemini-3.5-flash ^
    --ai.search.web-search-tool=duckduckgo ^
    --ai.google.key="%GOOGLE_KEY%" ^
    "%FILE_NAME%"
    

Configuration

For a detailed list of all configuration options, including AI models, re-encoding settings, and concurrency controls, please refer to CONFIGURATION.md.

All settings can be configured via command-line arguments (e.g., --ai.model=google-gla:gemini-3.5-flash) or environment variables with the AISUB_ prefix (e.g., AISUB_AI_MODEL=google-gla:gemini-3.5-flash).


Release Notes

For the latest updates and bug fixes, please refer to RELEASE_NOTES.md.


Known Limitations

  1. Timestamp Accuracy: Subtitle timestamps may occasionally be inaccurate due to limitations of the Gemini AI model. Shorter video segments generally yield better accuracy. Experiment with the --split.max-seconds setting.
  2. AI Hallucinations: Like all LLMs, Gemini may occasionally produce "hallucinations" or inaccurate information.

If you encounter issues, consider re-processing specific video segments as detailed below.


Advanced: Re-processing Segments

Intermediate files and job states are stored in a temporary directory (default: tmp_<input_file_name>). You can customize this location using the --dir.tmp flag.

The application creates separate state files for each processing stage (e.g., lyrics detection, subtitle generation). To re-process a specific segment, you must delete the state file for the stage you want to re-run.

File naming format: part_XXX.<stage>.<model_name>.json

Example: To re-run subtitle generation for the third segment:

  1. Navigate to the temporary directory.
  2. Identify the model name used for subtitles (e.g., gemini-3.5-flash).
  3. Delete the corresponding state file (e.g., part_002.subtitles.gemini-3-5-flash.json).
  4. Re-run the script. It will detect the missing subtitle job state and re-process only that segment, using any existing lyrics data.

To re-run the entire pipeline for that segment (including lyrics search), delete both the lyrics and subtitles JSON files for that part.

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ai_sub-3.3.0.tar.gz (49.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ai_sub-3.3.0-py3-none-any.whl (52.8 kB view details)

Uploaded Python 3

File details

Details for the file ai_sub-3.3.0.tar.gz.

File metadata

  • Download URL: ai_sub-3.3.0.tar.gz
  • Upload date:
  • Size: 49.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for ai_sub-3.3.0.tar.gz
Algorithm Hash digest
SHA256 9938fe2aac1b9b31d354e4ca17303bf4fdd1178c1219e4799679561ad28da6e5
MD5 810f4a6aa6a65aacb8c007ffd57fa78d
BLAKE2b-256 0642822e719438b99fe182febbe06ff83cea256ce3d5cff985d157e24af0221c

See more details on using hashes here.

Provenance

The following attestation bundles were made for ai_sub-3.3.0.tar.gz:

Publisher: publish.yml on FlippFuzz/ai-sub

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file ai_sub-3.3.0-py3-none-any.whl.

File metadata

  • Download URL: ai_sub-3.3.0-py3-none-any.whl
  • Upload date:
  • Size: 52.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for ai_sub-3.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 0a0f53bb0e072d45141388283c473208b131b04c6c85d19652febc1a8e2f250b
MD5 3f5f7f19ccf517390263e98dcfd2e4a4
BLAKE2b-256 a51c3eeaf444cfa51dfc2dc6e7933d9933634828d1398e2621e854fdab3e2034

See more details on using hashes here.

Provenance

The following attestation bundles were made for ai_sub-3.3.0-py3-none-any.whl:

Publisher: publish.yml on FlippFuzz/ai-sub

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page