chembfn-webui 1.0.0__tar.gz → 2.1.3__tar.gz

{chembfn_webui-1.0.0 → chembfn_webui-2.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chembfn_webui
-Version: 1.0.0
+Version: 2.1.3
 Summary: WebUI for ChemBFN
 Home-page: https://github.com/Augus1999/ChemBFN-WebUI
 Author: Nianze A. Tao
@@ -15,15 +15,16 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Topic :: Scientific/Engineering :: Chemistry
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: bayesianflow_for_chem>=2.2.2
-Requires-Dist: mol2chemfigPy3>=1.5.11
-Requires-Dist: gradio>=5.32.1
-Requires-Dist: torch>=2.7.0
+Requires-Dist: bayesianflow_for_chem>=2.4.0
+Requires-Dist: mol2chemfigPy3>=1.5.12
+Requires-Dist: gradio<7.0.0,>=6.0.0
+Requires-Dist: torch>=2.9.0
 Requires-Dist: selfies>=2.2.0
 Dynamic: author
 Dynamic: author-email
@@ -39,10 +40,20 @@ Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-## ChemBFN WebUI: WebUI to generate and visualise molecules for ChemBFN method
+## Web-based UI visualisation tool for ChemBFN method
 
+[![PyPI](https://img.shields.io/pypi/v/chembfn-webui?color=green)](https://pypi.org/project/chembfn-webui/)
+![CI](https://github.com/Augus1999/ChemBFN-WebUI/actions/workflows/pytest.yml/badge.svg)
+![black](https://img.shields.io/badge/code%20style-black-black)
+[![Stand With Ukraine](https://raw.githubusercontent.com/vshymanskyy/StandWithUkraine/main/badges/StandWithUkraine.svg)](https://stand-with-ukraine.pp.ua)
 
 
+
+> Important:
+>
+> For the security concerning, it is not recommended to use this application as a public service.
+> When deploying on a local host as a shared application, it is better to install this application in a container or VM, to prevent this application from accessing the Internet, and to limit the premissions of read, create, and delete loacal files and directories.
+
 ### 1. Install
 
 ```bash
@@ -83,6 +94,22 @@ For example,
             └───moses_selfies_vocab.txt
 ```
 
+> Note:
+>
+> >The file `config.json` is automatically saved by CLI tool `Madmol` provided in `bayesianflow-for-chem` package. If you train models via Python API, you need to manually create that file for your models by filling in the tempate:
+> >```json
+> >{
+> >    "padding_index": 0,
+> >    "start_index": 1,
+> >    "end_index": 2,
+> >    "padding_strategy": "static",
+> >    "padding_length": PADDING_LENGTH,
+> >    "label": [LABEL_NAME_I, LABEL_NAME_II, ...],
+> >    "name": JOB_NAME
+> >}
+> >```
+> >The configureation file for base models can be downloaded [here](https://huggingface.co/suenoomozawa/ChemBFN/resolve/main/config.json).
+
 If placed correctly, all these files can be seen in the "model explorer" tab.
 
 > You can use an external folder to host the models if it follows the same structure as [`chembfn_webui/model`](https://github.com/Augus1999/ChemBFN-WebUI/tree/main/chembfn_webui/model). See the next section for the method.
@@ -131,6 +158,7 @@ Under "advanced control" tab
 
 * You can control semi-autoregressive behaviours by key in `F` for switching off SAR, `T` for switching on SAR, and prompt like `F,F,T,...` to individually control the SAR in an ensemble model.
 * You can add unwanted tokens, e.g., `[Cu],p,[Si]`.
+* You can customise the result preprocessing function, e.g., the model output  a reaction SMILES "CCI.C[O-]>>COCC" which couldn't be recognised by RDKit; you can pass `lambda x: x.split(">>")[-1]` to force the program only looking at the products.
 
 ### 6. Generate molecules
 

{chembfn_webui-1.0.0 → chembfn_webui-2.1.3}/README.md

@@ -1,13 +1,23 @@
-## ChemBFN WebUI: WebUI to generate and visualise molecules for ChemBFN method
+## Web-based UI visualisation tool for ChemBFN method
+
+[![PyPI](https://img.shields.io/pypi/v/chembfn-webui?color=green)](https://pypi.org/project/chembfn-webui/)
+![CI](https://github.com/Augus1999/ChemBFN-WebUI/actions/workflows/pytest.yml/badge.svg)
+![black](https://img.shields.io/badge/code%20style-black-black)
+[![Stand With Ukraine](https://raw.githubusercontent.com/vshymanskyy/StandWithUkraine/main/badges/StandWithUkraine.svg)](https://stand-with-ukraine.pp.ua)
 
 <p align="left">
-<img src="image/screenshot_0.jpeg" alt="screenshot 0" width="400" height="auto">
-<img src="image/screenshot_1.jpeg" alt="screenshot 1" width="400" height="auto">
-<img src="image/screenshot_2.jpeg" alt="screenshot 2" width="400" height="auto">
-<img src="image/screenshot_3.jpeg" alt="screenshot 3" width="400" height="auto">
-<img src="image/screenshot_4.jpeg" alt="screenshot 4" width="400" height="auto">
+<img src="image/screenshot_0.jpeg" alt="screenshot 0" width="360" height="auto">
+<img src="image/screenshot_1.jpeg" alt="screenshot 1" width="360" height="auto">
+<img src="image/screenshot_2.jpeg" alt="screenshot 2" width="360" height="auto">
+<img src="image/screenshot_3.jpeg" alt="screenshot 3" width="360" height="auto">
+<img src="image/screenshot_4.jpeg" alt="screenshot 4" width="360" height="auto">
 </p>
 
+> [!IMPORTANT]
+>
+> For the security concerning, it is not recommended to use this application as a public service.
+> When deploying on a local host as a shared application, it is better to install this application in a container or VM, to prevent this application from accessing the Internet, and to limit the premissions of read, create, and delete loacal files and directories.
+
 ### 1. Install
 
 ```bash
@@ -48,6 +58,22 @@ For example,
             └───moses_selfies_vocab.txt
 ```
 
+> [!NOTE]
+>
+> >The file `config.json` is automatically saved by CLI tool `Madmol` provided in `bayesianflow-for-chem` package. If you train models via Python API, you need to manually create that file for your models by filling in the tempate:
+> >```json
+> >{
+> >    "padding_index": 0,
+> >    "start_index": 1,
+> >    "end_index": 2,
+> >    "padding_strategy": "static",
+> >    "padding_length": PADDING_LENGTH,
+> >    "label": [LABEL_NAME_I, LABEL_NAME_II, ...],
+> >    "name": JOB_NAME
+> >}
+> >```
+> >The configureation file for base models can be downloaded [here](https://huggingface.co/suenoomozawa/ChemBFN/resolve/main/config.json).
+
 If placed correctly, all these files can be seen in the "model explorer" tab.
 
 > You can use an external folder to host the models if it follows the same structure as [`chembfn_webui/model`](./chembfn_webui/model). See the next section for the method.
@@ -96,6 +122,7 @@ Under "advanced control" tab
 
 * You can control semi-autoregressive behaviours by key in `F` for switching off SAR, `T` for switching on SAR, and prompt like `F,F,T,...` to individually control the SAR in an ensemble model.
 * You can add unwanted tokens, e.g., `[Cu],p,[Si]`.
+* You can customise the result preprocessing function, e.g., the model output  a reaction SMILES "CCI.C[O-]>>COCC" which couldn't be recognised by RDKit; you can pass `lambda x: x.split(">>")[-1]` to force the program only looking at the products.
 
 ### 6. Generate molecules
 

{chembfn_webui-1.0.0 → chembfn_webui-2.1.3}/chembfn_webui/bin/app.py

@@ -6,8 +6,9 @@ Define application behaviours.
 import sys
 import argparse
 from pathlib import Path
+from copy import deepcopy
 from functools import partial
-from typing import Tuple, List, Dict, Union
+from typing import Tuple, List, Dict, Optional, Union, Literal
 
 sys.path.append(str(Path(__file__).parent.parent))
 from rdkit.Chem import Draw, MolFromSmiles  # type: ignore
@@ -32,18 +33,21 @@ from bayesianflow_for_chem.tool import (
     quantise_model_,
 )
 from lib.utilities import (
+    sys_info,
     find_model,
     find_vocab,
     parse_prompt,
     parse_exclude_token,
     parse_sar_control,
+    build_result_prep_fn,
 )
 from lib.version import __version__
 
 vocabs = find_vocab()
 models = find_model()
-lora_selected = False  # lora select flag
 cache_dir = Path(__file__).parent.parent / "cache"
+favicon_dir = Path(__file__).parent / "favicon.png"
+_RESULT_COUNT = 0
 
 HTML_STYLE = gr.InputHTMLAttributes(
     autocapitalize="off",
@@ -74,7 +78,7 @@ def selfies2vec(sel: str, vocab_dict: Dict[str, int]) -> List[int]:
     return [vocab_dict.get(i, unknown_id) for i in s]
 
 
-def refresh(
+def _refresh(
     model_selected: str, vocab_selected: str, tokeniser_selected: str
 ) -> Tuple[
     List[str], List[str], List[List[str]], List[List[str]], gr.Dropdown, gr.Dropdown
@@ -119,7 +123,7 @@ def refresh(
     return a, b, c, d, e, f
 
 
-def select_lora(evt: gr.SelectData, prompt: str) -> str:
+def _select_lora(evt: gr.SelectData, prompt: str) -> str:
     """
     Select LoRA model name from Dataframe object.
 
@@ -130,20 +134,16 @@ def select_lora(evt: gr.SelectData, prompt: str) -> str:
     :return: new prompt string
     :rtype: str
     """
-    global lora_selected
-    if lora_selected:  # avoid double select
-        lora_selected = False
-        return prompt
     selected_lora = evt.value
-    lora_selected = True
-    if evt.index[1] != 0:
+    exist_lora = parse_prompt(prompt)["lora"]
+    if evt.index[1] != 0 or selected_lora in exist_lora:
         return prompt
     if not prompt:
         return f"<{selected_lora}:1>"
     return f"{prompt};\n<{selected_lora}:1>"
 
 
-def token_name_change_evt(
+def _token_name_change_evt(
     token_name: str, vocab_fn: str
 ) -> Tuple[gr.Dropdown, gr.Tab, gr.Tab]:
     """
@@ -178,15 +178,17 @@ def run(
     batch_size: int,
     sequence_size: int,
     guidance_strength: float,
-    method: str,
+    method: Literal["BFN", "ODE"],
     temperature: float,
-    prompt: str,
-    scaffold: str,
-    template: str,
-    sar_control: str,
-    exclude_token: str,
-    quantise: str,
-    jited: str,
+    prompt: Optional[str],
+    scaffold: Optional[str],
+    template: Optional[str],
+    sar_control: Optional[str],
+    exclude_token: Optional[str],
+    quantise: Literal["on", "off"],
+    jited: Literal["on", "off"],
+    sorted_: Literal["on", "off"],
+    result_prep_fn: Optional[str],
 ) -> Tuple[Union[List, None], List[str], str, gr.TextArea, str]:
     """
     Run generation or inpainting.
@@ -207,6 +209,8 @@ def run(
     :param exclude_token: unwanted tokens
     :param quantise: `"on"` or `"off"`
     :param jited: `"on"` or `"off"`
+    :param sorted\\_: whether to sort the reulst; `"on"` or `"off"`
+    :param result_prep_fn: a string form result preprocessing function
     :type model_name: str
     :type token_name: str
     :type vocab_fn: str
@@ -216,13 +220,15 @@ def run(
     :type guidance_strength: float
     :type method: str
     :type temperature: float
-    :type prompt: str
-    :type scaffold: str
-    :type template: str
-    :type sar_control: str
-    :type exclude_token: str
+    :type prompt: str | None
+    :type scaffold: str | None
+    :type template: str | None
+    :type sar_control: str | None
+    :type exclude_token: str | None
     :type quantise: str
     :type jited: str
+    :type sorted\\_: str
+    :type result_prep_fn: str | None
     :return: list of images \n
              list of generated molecules \n
              Chemfig code \n
@@ -238,6 +244,8 @@ def run(
     lora_label_dict = dict([[i[0], i[2] != []] for i in models["lora"]])
     standalone_lmax_dict = dict([[i[0], i[3]] for i in models["standalone"]])
     lora_lmax_dict = dict([[i[0], i[3]] for i in models["lora"]])
+    # ------- build result preprocessing function -------
+    _result_prep_fn = build_result_prep_fn(result_prep_fn)
     # ------- build tokeniser -------
     if token_name == "SMILES & SAFE":
         vocab_keys = VOCAB_KEYS
@@ -245,27 +253,31 @@ def run(
         trans_fn = lambda x: [i for i in x if (MolFromSmiles(i) and i)]
         img_fn = lambda x: [Draw.MolToImage(MolFromSmiles(i), (500, 500)) for i in x]
         chemfig_fn = lambda x: [mol2chemfig(i, "-r", inline=True) for i in x]
-    if token_name == "FASTA":
+    elif token_name == "FASTA":
         vocab_keys = FASTA_VOCAB_KEYS
         tokeniser = fasta2vec
-        trans_fn = lambda x: x
+        trans_fn = lambda x: [i for i in x if i]
         img_fn = lambda _: None  # senseless to provide dumb 2D images
         chemfig_fn = lambda _: [""]  # senseless to provide very long Chemfig code
-    if token_name == "SELFIES":
+    elif token_name == "SELFIES":
         vocab_data = load_vocab(vocabs[vocab_fn])
         vocab_keys = vocab_data["vocab_keys"]
         vocab_dict = vocab_data["vocab_dict"]
         tokeniser = partial(selfies2vec, vocab_dict=vocab_dict)
-        trans_fn = lambda x: x
+        trans_fn = lambda x: [i for i in x if i]
         img_fn = lambda x: [
             Draw.MolToImage(MolFromSmiles(decoder(i)), (500, 500)) for i in x
         ]
         chemfig_fn = lambda x: [mol2chemfig(decoder(i), "-r", inline=True) for i in x]
+    else:
+        raise RuntimeError("Oops, maybe something wrong with Gradio.")
     _method = "bfn" if method == "BFN" else f"ode:{temperature}"
     # ------- build model -------
     prompt_info = parse_prompt(prompt)
     sar_flag = parse_sar_control(sar_control)
-    print("Prompt summary:", prompt_info)  # show prompt info
+    _info = deepcopy(prompt_info)
+    _info["semi-autoregression"] = deepcopy(sar_flag)
+    print("Prompt summary:", _info)  # prompt
     if not prompt_info["lora"]:
         if model_name in base_model_dict:
             lmax = sequence_size
@@ -286,11 +298,11 @@ def run(
                     mlp = MLP.from_checkpoint(
                         standalone_model_dict[model_name] / "mlp.pt"
                     )
-                    y = torch.tensor([prompt_info["objective"]], dtype=torch.float32)
+                    y = torch.tensor([prompt_info["objective"][0]], dtype=torch.float32)
                     y = mlp.forward(y)
             else:
                 y = None
-            _message.append(f"Sequence length is set to {lmax} from model metadata.")
+            _message.append(f"Sequence length set to {lmax} from model metadata.")
         bfn.semi_autoregressive = sar_flag[0]
         if quantise == "on":
             quantise_model_(bfn)
@@ -316,20 +328,20 @@ def run(
                 mlp = MLP.from_checkpoint(
                     lora_model_dict[prompt_info["lora"][0]] / "mlp.pt"
                 )
-                y = torch.tensor([prompt_info["objective"]], dtype=torch.float32)
+                y = torch.tensor([prompt_info["objective"][0]], dtype=torch.float32)
                 y = mlp.forward(y)
         else:
             y = None
         if prompt_info["lora_scaling"][0] != 1.0:
             adjust_lora_(bfn, prompt_info["lora_scaling"][0])
-        _message.append(f"Sequence length is set to {lmax} from model metadata.")
+        _message.append(f"Sequence length set to {lmax} from model metadata.")
         bfn.semi_autoregressive = sar_flag[0]
         if quantise == "on":
             quantise_model_(bfn)
         if jited == "on":
             bfn.compile()
     else:
-        lmax = max([lora_lmax_dict[i] for i in prompt_info["lora"]])
+        lmax = max(lora_lmax_dict[i] for i in prompt_info["lora"])
         if model_name in base_model_dict:
             base_model_dir = base_model_dict[model_name]
         else:
@@ -344,16 +356,25 @@ def run(
         if len(sar_flag) == 1:
             sar_flag = [sar_flag[0] for _ in range(len(weights))]
         bfn = EnsembleChemBFN(base_model_dir, lora_dir, mlps, weights)
-        y = [torch.tensor([i], dtype=torch.float32) for i in prompt_info["objective"]]
+        y = (
+            [torch.tensor([i], dtype=torch.float32) for i in prompt_info["objective"]]
+            if prompt_info["objective"]
+            else None
+        )
         if quantise == "on":
             bfn.quantise()
         if jited == "on":
             bfn.compile()
-        _message.append(f"Sequence length is set to {lmax} from model metadata.")
+        _message.append(f"Sequence length set to {lmax} from model metadata.")
+    result_prep_fn_ = lambda x: [_result_prep_fn(i) for i in x]
     # ------- inference -------
     allowed_tokens = parse_exclude_token(exclude_token, vocab_keys)
     if not allowed_tokens:
         allowed_tokens = "all"
+    if scaffold is None:
+        scaffold = ""
+    if template is None:
+        template = ""
     scaffold = scaffold.strip()
     template = template.strip()
     if scaffold:
@@ -369,8 +390,9 @@ def run(
             vocab_keys=vocab_keys,
             method=_method,
             allowed_tokens=allowed_tokens,
+            sort=sorted_ == "on",
         )
-        mols = trans_fn(mols)
+        mols = trans_fn(result_prep_fn_(mols))
         imgs = img_fn(mols)
         chemfigs = chemfig_fn(mols)
         if template:
@@ -388,8 +410,9 @@ def run(
             vocab_keys=vocab_keys,
             method=_method,
             allowed_tokens=allowed_tokens,
+            sort=sorted_ == "on",
         )
-        mols = trans_fn(mols)
+        mols = trans_fn(result_prep_fn_(mols))
         imgs = img_fn(mols)
         chemfigs = chemfig_fn(mols)
     else:
@@ -403,16 +426,19 @@ def run(
             vocab_keys=vocab_keys,
             method=_method,
             allowed_tokens=allowed_tokens,
+            sort=sorted_ == "on",
         )
-        mols = trans_fn(mols)
+        mols = trans_fn(result_prep_fn_(mols))
         imgs = img_fn(mols)
         chemfigs = chemfig_fn(mols)
-    n_mol = len(mols)
     with open(cache_dir / "results.csv", "w", encoding="utf-8", newline="") as rf:
         rf.write("\n".join(mols))
     _message.append(
-        f"{n_mol} smaples generated and saved to cache that can be downloaded."
+        f"{(n_mol := len(mols))} {'smaple' if n_mol in (0, 1) else 'samples'} "
+        "generated and saved to cache that can be downloaded."
     )
+    global _RESULT_COUNT
+    _RESULT_COUNT = n_mol
     return (
         imgs,
         mols,
@@ -422,11 +448,11 @@ def run(
     )
 
 
-with gr.Blocks(title="ChemBFN WebUI") as app:
-    gr.Markdown("### WebUI to generate and visualise molecules for ChemBFN method.")
+with gr.Blocks(title="ChemBFN WebUI", analytics_enabled=False) as app:
     with gr.Row():
         with gr.Column(scale=1):
             btn = gr.Button("RUN", variant="primary")
+            stop = gr.Button("\u23f9", variant="stop", visible=False)
             model_name = gr.Dropdown(
                 [i[0] for i in models["base"]] + [i[0] for i in models["standalone"]],
                 label="model",
@@ -471,14 +497,15 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
                 message = gr.TextArea(label="message", lines=2)
             with gr.Tab(label="result viewer"):
                 with gr.Tab(label="result"):
-                    btn_download = gr.File(label="download", visible=False)
+                    btn_download = gr.File(
+                        str(cache_dir / "results.csv"), label="download", visible=False
+                    )
                     result = gr.Dataframe(
                         headers=["molecule"],
-                        col_count=(1, "fixed"),
+                        column_count=(1, "fixed"),
                         label="",
-                        show_fullscreen_button=True,
+                        interactive=False,
                         show_row_numbers=True,
-                        show_copy_button=True,
                     )
                 with gr.Tab(
                     label="LATEX Chemfig", visible=token_name.value != "FASTA"
@@ -496,7 +523,7 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
                     vocab_table = gr.Dataframe(
                         list(vocabs.keys()),
                         headers=["name"],
-                        col_count=(1, "fixed"),
+                        column_count=(1, "fixed"),
                         label="",
                         interactive=False,
                         show_row_numbers=True,
@@ -505,7 +532,7 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
                     base_table = gr.Dataframe(
                         [i[0] for i in models["base"]],
                         headers=["name"],
-                        col_count=(1, "fixed"),
+                        column_count=(1, "fixed"),
                         label="",
                         interactive=False,
                         show_row_numbers=True,
@@ -514,7 +541,7 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
                     standalone_table = gr.Dataframe(
                         [[i[0], i[2]] for i in models["standalone"]],
                         headers=["name", "objective"],
-                        col_count=(2, "fixed"),
+                        column_count=(2, "fixed"),
                         label="",
                         interactive=False,
                         show_row_numbers=True,
@@ -523,7 +550,7 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
                     lora_tabel = gr.Dataframe(
                         [[i[0], i[2]] for i in models["lora"]],
                         headers=["name", "objective"],
-                        col_count=(2, "fixed"),
+                        column_count=(2, "fixed"),
                         label="",
                         interactive=False,
                         show_row_numbers=True,
@@ -540,10 +567,33 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
                     placeholder="key in unwanted tokens separated by comma.",
                     html_attributes=HTML_STYLE,
                 )
-                quantise = gr.Radio(["on", "off"], value="off", label="quantisation")
-                jited = gr.Radio(["on", "off"], value="off", label="JIT")
+                result_prep_fn = gr.Textbox(
+                    "lambda x: x",
+                    label="result preprocessing function",
+                    placeholder="lambda x: x",
+                    html_attributes=HTML_STYLE,
+                )
+                with gr.Row(scale=1):
+                    quantise = gr.Radio(
+                        ["on", "off"], value="off", label="quantisation"
+                    )
+                    jited = gr.Radio(["on", "off"], value="off", label="JIT")
+                    sorted_ = gr.Radio(
+                        ["on", "off"], value="off", label="sort result based on entropy"
+                    )
+    gr.HTML(sys_info(), elem_classes="custom_footer", elem_id="footer")
     # ------ user interaction events -------
-    btn.click(
+    gen = btn.click(
+        fn=lambda: (
+            gr.Button("RUN", variant="primary", visible=False),
+            gr.Button("\u23f9", variant="stop", visible=True),
+        ),
+        inputs=None,
+        outputs=[btn, stop],
+        api_name="switch_to_stop_mode",
+        api_description="Switch to STOP.",
+        api_visibility="private",
+    ).then(
         fn=run,
         inputs=[
             model_name,
@@ -562,11 +612,37 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
             exclude_token,
             quantise,
             jited,
+            sorted_,
+            result_prep_fn,
         ],
         outputs=[img, result, chemfig, message, btn_download],
+        api_name="run",
+        api_description="Run ChemBFN model.",
+    )
+    gen.then(
+        fn=lambda: (
+            gr.Button("RUN", variant="primary", visible=True),
+            gr.Button("\u23f9", variant="stop", visible=False),
+        ),
+        inputs=None,
+        outputs=[btn, stop],
+        api_name="switch_back_to_run_mode",
+        api_description="Swtch back to RUN.",
+        api_visibility="private",
+    )
+    stop.click(
+        fn=lambda: (
+            gr.Button("RUN", variant="primary", visible=True),
+            gr.Button("\u23f9", variant="stop", visible=False),
+        ),
+        inputs=None,
+        outputs=[btn, stop],
+        cancels=[gen],
+        api_name="stop",
+        api_description="Stop the model.",
     )
     btn_refresh.click(
-        fn=refresh,
+        fn=_refresh,
         inputs=[model_name, vocab_fn, token_name],
         outputs=[
             vocab_table,
@@ -576,11 +652,14 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
             model_name,
             vocab_fn,
         ],
+        api_name="refresh_model_list",
+        api_description="Refresh the model list.",
     )
     token_name.input(
-        fn=token_name_change_evt,
+        fn=_token_name_change_evt,
         inputs=[token_name, vocab_fn],
         outputs=[vocab_fn, code, gallery],
+        api_visibility="private",
     )
     method.input(
         fn=lambda x, y: gr.Slider(
@@ -593,12 +672,25 @@ with gr.Blocks(title="ChemBFN WebUI") as app:
         ),
         inputs=[method, temperature],
         outputs=temperature,
+        api_name="select_sampling_method",
+        api_description="Select sampling method between 'BFN' and 'ODE'.",
+        api_visibility="private",
+    )
+    lora_tabel.select(
+        fn=_select_lora,
+        inputs=prompt,
+        outputs=prompt,
+        api_name="select_lora",
+        api_description="Select LoRA model from the model list.",
+        api_visibility="private",
     )
-    lora_tabel.select(fn=select_lora, inputs=prompt, outputs=prompt)
     result.change(
-        fn=lambda x: gr.File(x, label="download", visible=True),
+        fn=lambda x: gr.File(x, label="download", visible=_RESULT_COUNT > 0),
         inputs=btn_download,
         outputs=btn_download,
+        api_name="change_download_state",
+        api_description="Hide or show the file downloading item.",
+        api_visibility="private",
     )
 
 
@@ -609,18 +701,29 @@ def main() -> None:
     :return:
     :rtype: None
     """
+    from rdkit import RDLogger
+
+    RDLogger.DisableLog("rdApp.*")  # type: ignore
     parser = argparse.ArgumentParser(
         description="A web-based visualisation tool for ChemBFN method.",
-        epilog=f"ChemBFN WebUI {__version__}, developed in Hiroshima University by chemists for chemists. "
+        epilog=f"ChemBFN WebUI {__version__}, "
+        "developed in Hiroshima University by chemists for chemists. "
         "Visit https://augus1999.github.io/bayesian-flow-network-for-chemistry/ for more details.",
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
     parser.add_argument(
-        "--public", default=False, help="open to public", action="store_true"
+        "-P", "--public", default=False, help="open to public", action="store_true"
     )
     parser.add_argument("-V", "--version", action="version", version=__version__)
     args = parser.parse_args()
-    app.launch(share=args.public, allowed_paths=[cache_dir.absolute().__str__()])
+    print(f"This is ChemBFN WebUI version {__version__}")
+    app.launch(
+        share=args.public,
+        footer_links=["api"],
+        allowed_paths=[cache_dir.absolute().__str__()],
+        favicon_path=favicon_dir.absolute().__str__(),
+        css=".custom_footer {text-align:center;bottom:0;}",
+    )
 
 
 if __name__ == "__main__":

chembfn_webui-2.1.3/chembfn_webui/cache/results.csv

@@ -0,0 +1 @@
+c1ccccc1