Skip to content

davistdaniel/chempleter

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

160 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Chempleter

Demo Gif

Molecular autocomplete

PyPI - Status PyPI - Version Python Version from PEP 621 TOML PyPI - License GitHub last commit GitHub Actions Workflow Status GitHub Actions Workflow Status

Chempleter is a lightweight generative sequence model based on a multi-layer gated recurrent units (GRU) to predict syntactically valid extensions of a provided molecular fragment or bridge two molecules/molecular fragments. It operates on SELFIES token sequences, ensuring syntactically valid molecular generation and accepts SMILES notation as input. Due to its simple recurrent architecture and small vocabulary, the model runs efficiently on both CPUs and GPUs.

Demo Gif
  • What can Chempleter do?

    • Currently, Chempleter accepts an intial molecule/molecular fragment in SMILES format and generates a larger molecule with that intial structure included, while respecting chemical syntax. It also shows some interesting descriptors.

    • It can be used to generate a wide range of structural analogs which the share same core structure (by changing the sampling temperature) or decorate a core scaffold iteratively (by increasing generated token lengths)

    • It can be used to bridge two molecules/molecular fragments to explore linker chemistry.

    • In the future, it might be adapated to predict structures with a specific chemical property using a regressor to rank predictions and transition towards more "goal-directed" predictions.

See chempleter in action.

Prerequisites

  • Python ">=3.12"
  • uv (optional but recommended)

Installation

See detailed installation instructions.

Getting started

Visit Chempleter's docs.

Quick start

These commands are valid for running Chempleter on CPU. For GPU, see installation instructions.

  • Run the GUI directly without installing (via uv):

    • On windows:

      uvx --from "chempleter[cpu]" chempleter-gui.exe

    • On linux/MacOS:

      uvx --from "chempleter[cpu]" chempleter-gui

    • The very first start of the GUI on your device might be a bit slow. To know more about using the GUI and various options, see here.

    Or

  • Install using uv

    uv pip install "chempleter[cpu]"

  • Run GUI after installation

    uv run chempleter-gui

  • Use the GUI

    • Type in the SMILES notation for the starting structure or leave it empty to generate random molecules. Click on GENERATE button to generate a molecule.

    • To know more about using the GUI and various other options, see here.

  • Use as a python library

    • To use Chempleter as a python library:

      from chempleter.inference import extend
      generated_mol, generated_smiles, generated_selfies = extend(smiles="c1ccccc1")
      print(generated_smiles)
      >> C1=CC=CC=C1C2=CC=C(CN3C=NC4=CC=CC=C4C3=O)O2

      To draw the generated molecule :

      from rdkit import Chem
      Chem.Draw.MolToImage(generated_mol)
    • For details on available paramenters and inference functions, see generating molecules.

Model history and validation

See model validation reports.

Project structure

  • src/chempleter: Contains python modules relating to different functions.
  • src/chempleter/processor.py: Contains fucntions for processing csv files containing SMILES data and generating training-related files.
  • src/chempleter/dataset.py: ChempleterDataset class
  • src/chempleter/model.py: ChempleterModel class
  • src/chempleter/inference.py: Contains functions for inference
  • src/chempleter/train.py: Contains functions for training
  • src/chempleter/gui.py: Chempleter GUI built using NiceGUI
  • src/chempleter/data : Contains trained model, vocabulary files

License

MIT License

Copyright (c) 2025-2026 Davis Thomas Daniel

Contributing

Any contribution, improvements, feature ideas or bug fixes are always welcome.

About

Chempleter is a lightweight generative model to predict syntactically valid extensions of molecular structures.

Topics

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors

Languages