pymolfold 0.2.2

Protein structure prediction in PyMOL

PymolFold

Inspired by ColabFold by Sergey O.
Visualization inspired by pymol-color-alphafold.
Thanks to ESMFold by Meta and the API.
Fast access to AlphaMissense predicted Human proteins provided by hegelab.

Quick Start Guide

1. Install PyMOL

This project enables structure and domain prediction directly within the PyMOL visualization software.

SO, download and install PyMOL from the official website.

---

2. Obtain API Tokens

PymolFold utilizes APIs from ESM3 and NVIDIA Boltz2.
You need to obtain API tokens from both:

• ESM3 API
• NVIDIA Boltz2

After obtaining your tokens, set them as environment variables:

• ESM_API_TOKEN
• NVCF_API_KEY
  Both must be uppercase and contain underscores.

How to set environment variables:

Windows:

1. Press Win + S and search for "Environment Variables".
2. Select "Edit the system environment variables".
3. In the System Properties window, go to "Advanced" → "Environment Variables".
4. Add or edit variables in "System variables" or "User variables".
   • Example:
     • Variable name: MY_ENV
     • Variable value: hello
5. (Windows User pay attention): In case of some strange error occurs, you also need set an extra variables PYTHONUTF8 and value set to 1.
6. Save and close.

Mac:

1. Press Command + Space, type "terminal", and open Terminal.
2. Edit your shell config:

   vim ~/.zshrc
   

3. Add your variable (example, replace with your actual token):

   export MY_ENV=hello
   

4. Save and exit Vim. (Ask ChatGPT if you duno how to do this)
5. Activate the environment variable:

   source ~/.zshrc
   

---

3. Run the Plugin

Once PyMOL and your API tokens are ready, download the run_plugin.py file from this project.
Open PyMOL and enter the full path to run_plugin.py:

run path_to/run_plugin.py

If you see the following, installation was successful:

---

4. How to Use

PymolFold provides several features: esm3, boltz2, and color_plddt.

1. Predict Monomer Protein Structure

Use the convenient esm3 command:

esm3 sequence [, name]
# Example:
esm3 MKTVRQERLKSIVRILERSKEPVSGAQLAEELSVSRQVIVQDIAYLRSLGYNIVATPRGYVLAGG

---

2. Predict Complexes, DNA, RNA, or Ligand Structures

For more complex predictions, use the boltz2 command.
Due to the number of required inputs, a web interface is provided (inspired by NVIDIA Boltz2).
Currently, conditional prediction is not supported, but may be added in the future.

To launch the web interface, enter the following in the PyMOL command line:

boltz2

You can run the provided example:

When using CCD code, you can check all the existed CCDs under pymolfold/gui/ccd_keys.json

After clicking Run on the web page, wait about 6 seconds (depending on protein size), and the structure will appear in PyMOL!

---

3. View pLDDT Scores

After prediction, enter the following to view pLDDT scores for the predicted structure:

color_plddt

4. How to evaluate the predicted results?

We utilized PXMeter to evaluate the differences between predicted structures and reference structures. PXMeter(0.1.4) now only supports PPI analysis, and more details can be seen in their repo.

But how to use in PyMolFold?

pxmeter_align full_path_to/real_struct.cif full_path_to/predict_struct.cif 

Unfortunately, we haven't fix out how to get the structure path from pymol object, so you guys have to manually provide them. Besides, only cif format is supported now.

After running the script above, you will get the metrics in csv and png format under the folder you setted (if not set, it will generate in the root path). You can use the exmaple files under pymolfold/example/, and the results should be exactly the same as pymolfold/example/metrics.

Others

Version Current version is 0.2.2, and if you are interesting in the source code, you can install pymolfold directly by pip install pymolfold==0.2.2.

Info
The PymolFold service is running on a A5000 instance (cost $100 a week), and the sequence length is limited to 1000aa.

Issues and Errors
If you encounter any errors or issues while using this project, please don't hesitate to open an issue here on GitHub. Your feedback helps us improve the project and make it more user-friendly for everyone.

PymolFold Server: A Shared Resource
Please note that the PymolFold server is a shared resource, and I request you to use it responsibly. Do not abuse the server, as it can affect the availability and performance of the service for other users.

21Sept2025: Refactor PyMOLFold, deleting unrelated module, adding boltz2 using NVIDIA API
17Jan2025: Add `esm3` to use ESM-3 for folding.
21Aug2023: As the ESMFold API is not stable, the job will be sent to PymolFold server if the job failed.
11Apr2023: `pf_plugin.py` is the PyMOL plugin and the `pf_pkg.py` is a pymol-free python package.