chembfn-webui 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

chembfn_webui/bin/app.py

@@ -42,7 +42,8 @@ from lib.version import __version__
 
 vocabs = find_vocab()
 models = find_model()
-lora_selected = False  # lora select flag
+_lora_selected = False  # lora select flag
+_run_in_public = False  # public flag
 cache_dir = Path(__file__).parent.parent / "cache"
 
 HTML_STYLE = gr.InputHTMLAttributes(
@@ -130,12 +131,12 @@ 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
+    global _lora_selected
+    if _lora_selected and not _run_in_public:  # avoid double select
+        _lora_selected = False
         return prompt
     selected_lora = evt.value
-    lora_selected = True
+    _lora_selected = True
     if evt.index[1] != 0:
         return prompt
     if not prompt:
@@ -344,7 +345,11 @@ 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":
@@ -616,10 +621,12 @@ def main() -> None:
         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()
+    global _run_in_public
+    _run_in_public = args.public
     app.launch(share=args.public, allowed_paths=[cache_dir.absolute().__str__()])
 
 

chembfn_webui/cache/results.csv

@@ -1 +0,0 @@
-CN1CCOCC1c1cccc(C(=O)Nc2ccnn2C(C)C)c1

chembfn_webui/lib/version.py

@@ -4,5 +4,5 @@
 Version info.
 """
 
-__version__ = "1.0.0"
+__version__ = "1.1.0"
 __author__ = "Nianze A. TAO"

{chembfn_webui-1.0.0.dist-info → chembfn_webui-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chembfn_webui
-Version: 1.0.0
+Version: 1.1.0
 Summary: WebUI for ChemBFN
 Home-page: https://github.com/Augus1999/ChemBFN-WebUI
 Author: Nianze A. Tao
@@ -83,6 +83,21 @@ 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
+> >}
+> >```
+
 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.

{chembfn_webui-1.0.0.dist-info → chembfn_webui-1.1.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 chembfn_webui/__init__.py,sha256=AXUdd_PrlfVO56losFUP7A8XrqCDPylwRbTpe_WG3Uc,87
-chembfn_webui/bin/app.py,sha256=fqIP5O5aojwZmw2-eHBcuFm4YuQBIEuwOMTvDzW7jcA,21632
+chembfn_webui/bin/app.py,sha256=Oguy3G8jaIuNT7RZWgtGweMKzCLipagMGH7E7y7C27A,21849
 chembfn_webui/cache/cache_file_here.txt,sha256=hi60T_q6Cf5WPtXuwe4CqjiWpaUqrczsmGMhKIUL--M,28
-chembfn_webui/cache/results.csv,sha256=xdSOWM1GGGJEFS1Y4sfCjA-9-66AECEo6El59_yW1hw,37
+chembfn_webui/cache/results.csv,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chembfn_webui/lib/utilities.py,sha256=ALPw-Evjd9DdsU_RQA6Zp2Gc6XnRR7Y_5fZrqG9azWo,7460
-chembfn_webui/lib/version.py,sha256=Mbvn1j2C-hWCKICax3XLfU-P0Q3j0oPQ4GQXoDFJ3fs,138
+chembfn_webui/lib/version.py,sha256=9gQIR8VWU0vhANDgg8w-friQemzdDuGMpih04LtIwCY,138
 chembfn_webui/model/base_model/place_base_model_here.txt,sha256=oa8_ILaAlWpTXICVDi-Y46_OahV7wB6Che6gbiEIh-c,39
 chembfn_webui/model/lora/place_lora_folder_here.txt,sha256=YYOo0Cj278DyRcgVrCLa1f2Q-cqgNeMnelaLiA3Fuic,69
 chembfn_webui/model/standalone_model/place_standalone_model_folder_here.txt,sha256=Dp42UscfI0Zp3SnvRv5vOfWiJZnxdY7rG3jo0kf86VM,80
 chembfn_webui/model/vocab/place_vocabulary_file_here.txt,sha256=fLOINvZP2022oE7RsmfDjgyaw2yMi7glmdu_cTwmo88,28
-chembfn_webui-1.0.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-chembfn_webui-1.0.0.dist-info/METADATA,sha256=iRpNEXoJZRG42fMjyHwsmcUJ5jV0MrGgWUMkZrqUb3s,5897
-chembfn_webui-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chembfn_webui-1.0.0.dist-info/entry_points.txt,sha256=fp8WTPybvwpeYKrUhTi456wwZbmCMJXN1TeFGpR1SlY,55
-chembfn_webui-1.0.0.dist-info/top_level.txt,sha256=VdWt3Z7jhbB0pQO_mkRawnU5s75SBT9BV8fGaAIJTDI,14
-chembfn_webui-1.0.0.dist-info/RECORD,,
+chembfn_webui-1.1.0.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+chembfn_webui-1.1.0.dist-info/METADATA,sha256=zLnFwjbkQsvqAm2fg5f17A7N-4X2sIFvJ4n4tcr2Sa0,6412
+chembfn_webui-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chembfn_webui-1.1.0.dist-info/entry_points.txt,sha256=fp8WTPybvwpeYKrUhTi456wwZbmCMJXN1TeFGpR1SlY,55
+chembfn_webui-1.1.0.dist-info/top_level.txt,sha256=VdWt3Z7jhbB0pQO_mkRawnU5s75SBT9BV8fGaAIJTDI,14
+chembfn_webui-1.1.0.dist-info/RECORD,,