Add new METL models in dropdown

app.py

@@ -14,6 +14,78 @@ metl_config = AutoConfig.from_pretrained('gitter-lab/METL', trust_remote_code=Tr
 metl = AutoModel.from_pretrained('gitter-lab/METL', trust_remote_code=True, cache_dir='./cache')
 pdb_path = None
 
+# UUID lookup for models -- used for the dropdown list
+uuid_lookup = {
+    "Source / Global / 1D — [METL-G-20M-1D] (D72M9aEp)": "D72M9aEp",
+    "Source / Global / 3D — [METL-G-20M-3D] (Nr9zCKpR)": "Nr9zCKpR",
+    "Source / Global / 1D — [METL-G-50M-1D] (auKdzzwX)": "auKdzzwX",
+    "Source / Global / 3D — [METL-G-50M-3D] (6PSAzdfv)": "6PSAzdfv",
+    "Source / Local / GFP / 1D — [METL-L-2M-1D-GFP] (8gMPQJy4)": "8gMPQJy4",
+    "Source / Local / GFP / 3D — [METL-L-2M-3D-GFP] (Hr4GNHws)": "Hr4GNHws",
+    "Source / Local / DLG4 / 1D — [METL-L-2M-1D-DLG4_2022] (8iFoiYw2)": "8iFoiYw2",
+    "Source / Local / DLG4 / 3D — [METL-L-2M-3D-DLG4_2022] (kt5DdWTa)": "kt5DdWTa",
+    "Source / Local / GB1 / 1D — [METL-L-2M-1D-GB1] (DMfkjVzT)": "DMfkjVzT",
+    "Source / Local / GB1 / 3D — [METL-L-2M-3D-GB1] (epegcFiH)": "epegcFiH",
+    "Source / Local / GRB2 / 1D — [METL-L-2M-1D-GRB2] (kS3rUS7h)": "kS3rUS7h",
+    "Source / Local / GRB2 / 3D — [METL-L-2M-3D-GRB2] (X7w83g6S)": "X7w83g6S",
+    "Source / Local / Pab1 / 1D — [METL-L-2M-1D-Pab1] (UKebCQGz)": "UKebCQGz",
+    "Source / Local / Pab1 / 3D — [METL-L-2M-3D-Pab1] (2rr8V4th)": "2rr8V4th",
+    "Source / Local / PTEN / 1D — [METL-L-2M-1D-PTEN] (CEMSx7ZC)": "CEMSx7ZC",
+    "Source / Local / PTEN / 3D — [METL-L-2M-3D-PTEN] (PjxR5LW7)": "PjxR5LW7",
+    "Source / Local / TEM-1 / 1D — [METL-L-2M-1D-TEM-1] (PREhfC22)": "PREhfC22",
+    "Source / Local / TEM-1 / 3D — [METL-L-2M-3D-TEM-1] (9ASvszux)": "9ASvszux",
+    "Source / Local / Ube4b / 1D — [METL-L-2M-1D-Ube4b] (HscFFkAb)": "HscFFkAb",
+    "Source / Local / Ube4b / 3D — [METL-L-2M-3D-Ube4b] (H48oiNZN)": "H48oiNZN",
+    "METL-BIND / Source / Local / GB1 / 3D — [METL-BIND-2M-3D-GB1-STANDARD] (K6mw24Rg)": "K6mw24Rg",
+    "METL-BIND / Source / Local / GB1 / 3D — [METL-BIND-2M-3D-GB1-BINDING] (Bo5wn2SG)": "Bo5wn2SG",
+    "Finetuned / Global / GFP / 1D — (PeT2D92j)": "PeT2D92j",
+    "Finetuned / Global / GFP / 3D — (6JBzHpkQ)": "6JBzHpkQ",
+    "Finetuned / Global / DLG4-Abundance / 1D — (4Rh3WCbG)": "4Rh3WCbG",
+    "Finetuned / Global / DLG4-Abundance / 3D — (RBtqxzvu)": "RBtqxzvu",
+    "Finetuned / Global / DLG4-Binding / 1D — (4xbuC5y7)": "4xbuC5y7",
+    "Finetuned / Global / DLG4-Binding / 3D — (BuvxgE2x)": "BuvxgE2x",
+    "Finetuned / Global / GB1 / 1D — (dAndZfJ4)": "dAndZfJ4",
+    "Finetuned / Global / GB1 / 3D — (9vSB3DRM)": "9vSB3DRM",
+    "Finetuned / Global / GRB2-Abundance / 1D — (HenDpDWe)": "HenDpDWe",
+    "Finetuned / Global / GRB2-Abundance / 3D — (dDoCCvfr)": "dDoCCvfr",
+    "Finetuned / Global / GRB2-Binding / 1D — (cvnycE5Q)": "cvnycE5Q",
+    "Finetuned / Global / GRB2-Binding / 3D — (jYesS9Ki)": "jYesS9Ki",
+    "Finetuned / Global / Pab1 / 1D — (ho54gxzv)": "ho54gxzv",
+    "Finetuned / Global / Pab1 / 3D — (jhbL2FeB)": "jhbL2FeB",
+    "Finetuned / Global / PTEN-Abundance / 1D — (UEuMtmfx)": "UEuMtmfx",
+    "Finetuned / Global / PTEN-Abundance / 3D — (eJPPQYEW)": "eJPPQYEW",
+    "Finetuned / Global / PTEN-Activity / 1D — (U3X8mSeT)": "U3X8mSeT",
+    "Finetuned / Global / PTEN-Activity / 3D — (4gqYnW6V)": "4gqYnW6V",
+    "Finetuned / Global / TEM-1 / 1D — (ELL4GGQq)": "ELL4GGQq",
+    "Finetuned / Global / TEM-1 / 3D — (K6BjsWXm)": "K6BjsWXm",
+    "Finetuned / Global / Ube4b / 1D — (BAWw23vW)": "BAWw23vW",
+    "Finetuned / Global / Ube4b / 3D — (G9piq6WH)": "G9piq6WH",
+    "Finetuned / Local / GFP / 1D — (HaUuRwfE)": "HaUuRwfE",
+    "Finetuned / Local / GFP / 3D — (LWEY95Yb)": "LWEY95Yb",
+    "Finetuned / Local / DLG4-Abundance / 1D — (RMFA6dnX)": "RMFA6dnX",
+    "Finetuned / Local / DLG4-Abundance / 3D — (V3uTtXVe)": "V3uTtXVe",
+    "Finetuned / Local / DLG4-Binding / 1D — (YdzBYWHs)": "YdzBYWHs",
+    "Finetuned / Local / DLG4-Binding / 3D — (iu6ZahPw)": "iu6ZahPw",
+    "Finetuned / Local / GB1 / 1D — (Pgcseywk)": "Pgcseywk",
+    "Finetuned / Local / GB1 / 3D — (UvMMdsq4)": "UvMMdsq4",
+    "Finetuned / Local / GRB2-Abundance / 1D — (VNpi9Zjt)": "VNpi9Zjt",
+    "Finetuned / Local / GRB2-Abundance / 3D — (PqBMjXkA)": "PqBMjXkA",
+    "Finetuned / Local / GRB2-Binding / 1D — (Z59BhUaE)": "Z59BhUaE",
+    "Finetuned / Local / GRB2-Binding / 3D — (VwcRN6UB)": "VwcRN6UB",
+    "Finetuned / Local / Pab1 / 1D — (TdjCzoQQ)": "TdjCzoQQ",
+    "Finetuned / Local / Pab1 / 3D — (5SjoLx3y)": "5SjoLx3y",
+    "Finetuned / Local / PTEN-Abundance / 1D — (oUScGeHo)": "oUScGeHo",
+    "Finetuned / Local / PTEN-Abundance / 3D — (DhuasDEr)": "DhuasDEr",
+    "Finetuned / Local / PTEN-Activity / 1D — (m9UsG7dq)": "m9UsG7dq",
+    "Finetuned / Local / PTEN-Activity / 3D — (8Vi7ENcC)": "8Vi7ENcC",
+    "Finetuned / Local / TEM-1 / 1D — (64ncFxBR)": "64ncFxBR",
+    "Finetuned / Local / TEM-1 / 3D — (PncvgiJU)": "PncvgiJU",
+    "Finetuned / Local / Ube4b / 1D — (e9uhhnAv)": "e9uhhnAv",
+    "Finetuned / Local / Ube4b / 3D — (NfbZL7jK)": "NfbZL7jK",
+    "GFP DESIGN / Finetuned / Local / GFP / 1D — (YoQkzoLD)": "YoQkzoLD",
+    "GFP DESIGN / Finetuned / Local / GFP / 3D — (PEkeRuxb)": "PEkeRuxb"
+}
+
 RADIO_CSS = """
 #indexing>div {
     flex-direction: column;
@@ -275,20 +347,26 @@ def empty_pdb_path(button):
     pdb_path = None
     return ""
 
-def load_model(model_id, _):
+def load_model(model_input, _):
     global metl
 
-    if not isinstance(model_id, str):
-        return get_color(RED, "Select Model"), gr.Button(interactive=False)
+    # Case 1: input is one of the dropdown labels → lookup UUID
+    if model_input in uuid_lookup:
+        uuid = uuid_lookup[model_input]
+        metl.load_from_uuid(uuid)
+        return get_color(GREEN, f"{model_input} loaded (from dropdown)"), gr.Button(interactive=True)
 
-    if model_id.lower() in metl.config.IDENT_UUID_MAP:
-        metl.load_from_ident(model_id)
-    elif model_id in metl.config.UUID_URL_MAP:
-        metl.load_from_uuid(model_id)
-    else:
-        return get_color(RED, "Model Load Failed"), gr.Button(interactive=False)
+    # Case 2: input is a raw UUID
+    elif model_input in metl.config.UUID_URL_MAP:
+        metl.load_from_uuid(model_input)
+        return get_color(GREEN, f"{model_input} loaded (direct UUID)"), gr.Button(interactive=True)
+
+    # Case 3: input is a raw IDENT
+    elif model_input in metl.config.IDENT_UUID_MAP:
+        metl.load_from_ident(model_input)
+        return get_color(GREEN, f"{model_input} loaded (direct IDENT)"), gr.Button(interactive=True)
 
-    return get_color(GREEN, f"{model_id} loaded"), gr.Button(interactive=True)
+    return get_color(RED, "Model not recognized"), gr.Button(interactive=False)
 
 def update_pdb(variant_modal, indexing):
     if len(variant_modal['text']) == 0 and len(variant_modal['files']) == 0:
@@ -322,7 +400,7 @@ def hide_variants(checkbox_values):
 def populate_example():
     global pdb_path
     
-    model = "metl-l-2m-3d-gb1"
+    model = "Source / Local / GB1 / 3D — [METL-L-2M-3D-GB1] (epegcFiH)"
     wt = "MQYKLILNGKTLKGETTTEAVDAATAEKVFKQYANDNGVDGEWTYDDATKTFTVTE"
     variants = '["T17P,T54F", "V28L,F51A"]'
     # "T17P,V28L,F51A,T54F"
@@ -419,12 +497,20 @@ def predict(input_multi_modal, variant_multi_modal, variant_index_type):
     return out_str, iframe, checkbox
 
 with gr.Blocks(css=RADIO_CSS) as demo:    
-    with gr.Row(equal_height=True, elem_id="modelPDBRow"):
-        with gr.Column(elem_id="modelInputCol"):
-            metl_model_id = gr.Dropdown(label="METL model IDENT or UUID", choices=list(metl.config.IDENT_UUID_MAP.keys()), allow_custom_value=False)
-            metl_model_status = gr.HTML(get_color(RED, "Select Model"), elem_id="modelStatus")
-        with gr.Column():
-                upload_pdb = gr.File(label="PDB File upload", elem_id="pdbUpload", file_types=[".pdb", ".txt"])
+    # with gr.Row(equal_height=True, elem_id="modelPDBRow"):
+    #     with gr.Column(elem_id="modelInputCol"):
+    #         # metl_model_id = gr.Dropdown(label="METL model IDENT or UUID", choices=list(metl.config.IDENT_UUID_MAP.keys()), allow_custom_value=False)
+    #         metl_model_id = gr.Dropdown(label="METL model", choices=['Source / Global / 1D — [METL-G-20M-1D] (D72M9aEp)', 'Source / Global / 3D — [METL-G-20M-3D] (Nr9zCKpR)', 'Source / Global / 1D — [METL-G-50M-1D] (auKdzzwX)', 'Source / Global / 3D — [METL-G-50M-3D] (6PSAzdfv)', 'Source / Local / GFP / 1D — [METL-L-2M-1D-GFP] (8gMPQJy4)', 'Source / Local / GFP / 3D — [METL-L-2M-3D-GFP] (Hr4GNHws)', 'Source / Local / DLG4 / 1D — [METL-L-2M-1D-DLG4_2022] (8iFoiYw2)', 'Source / Local / DLG4 / 3D — [METL-L-2M-3D-DLG4_2022] (kt5DdWTa)', 'Source / Local / GB1 / 1D — [METL-L-2M-1D-GB1] (DMfkjVzT)', 'Source / Local / GB1 / 3D — [METL-L-2M-3D-GB1] (epegcFiH)', 'Source / Local / GRB2 / 1D — [METL-L-2M-1D-GRB2] (kS3rUS7h)', 'Source / Local / GRB2 / 3D — [METL-L-2M-3D-GRB2] (X7w83g6S)', 'Source / Local / Pab1 / 1D — [METL-L-2M-1D-Pab1] (UKebCQGz)', 'Source / Local / Pab1 / 3D — [METL-L-2M-3D-Pab1] (2rr8V4th)', 'Source / Local / PTEN / 1D — [METL-L-2M-1D-PTEN] (CEMSx7ZC)', 'Source / Local / PTEN / 3D — [METL-L-2M-3D-PTEN] (PjxR5LW7)', 'Source / Local / TEM-1 / 1D — [METL-L-2M-1D-TEM-1] (PREhfC22)', 'Source / Local / TEM-1 / 3D — [METL-L-2M-3D-TEM-1] (9ASvszux)', 'Source / Local / Ube4b / 1D — [METL-L-2M-1D-Ube4b] (HscFFkAb)', 'Source / Local / Ube4b / 3D — [METL-L-2M-3D-Ube4b] (H48oiNZN)', 'METL-BIND / Source / Local / GB1 / 3D — [METL-BIND-2M-3D-GB1-STANDARD] (K6mw24Rg)', 'METL-BIND / Source / Local / GB1 / 3D — [METL-BIND-2M-3D-GB1-BINDING] (Bo5wn2SG)', 'Finetuned / Global / GFP / 1D — (PeT2D92j)', 'Finetuned / Global / GFP / 3D — (6JBzHpkQ)', 'Finetuned / Global / DLG4-Abundance / 1D — (4Rh3WCbG)', 'Finetuned / Global / DLG4-Abundance / 3D — (RBtqxzvu)', 'Finetuned / Global / DLG4-Binding / 1D — (4xbuC5y7)', 'Finetuned / Global / DLG4-Binding / 3D — (BuvxgE2x)', 'Finetuned / Global / GB1 / 1D — (dAndZfJ4)', 'Finetuned / Global / GB1 / 3D — (9vSB3DRM)', 'Finetuned / Global / GRB2-Abundance / 1D — (HenDpDWe)', 'Finetuned / Global / GRB2-Abundance / 3D — (dDoCCvfr)', 'Finetuned / Global / GRB2-Binding / 1D — (cvnycE5Q)', 'Finetuned / Global / GRB2-Binding / 3D — (jYesS9Ki)', 'Finetuned / Global / Pab1 / 1D — (ho54gxzv)', 'Finetuned / Global / Pab1 / 3D — (jhbL2FeB)', 'Finetuned / Global / PTEN-Abundance / 1D — (UEuMtmfx)', 'Finetuned / Global / PTEN-Abundance / 3D — (eJPPQYEW)', 'Finetuned / Global / PTEN-Activity / 1D — (U3X8mSeT)', 'Finetuned / Global / PTEN-Activity / 3D — (4gqYnW6V)', 'Finetuned / Global / TEM-1 / 1D — (ELL4GGQq)', 'Finetuned / Global / TEM-1 / 3D — (K6BjsWXm)', 'Finetuned / Global / Ube4b / 1D — (BAWw23vW)', 'Finetuned / Global / Ube4b / 3D — (G9piq6WH)', 'Finetuned / Local / GFP / 1D — (HaUuRwfE)', 'Finetuned / Local / GFP / 3D — (LWEY95Yb)', 'Finetuned / Local / DLG4-Abundance / 1D — (RMFA6dnX)', 'Finetuned / Local / DLG4-Abundance / 3D — (V3uTtXVe)', 'Finetuned / Local / DLG4-Binding / 1D — (YdzBYWHs)', 'Finetuned / Local / DLG4-Binding / 3D — (iu6ZahPw)', 'Finetuned / Local / GB1 / 1D — (Pgcseywk)', 'Finetuned / Local / GB1 / 3D — (UvMMdsq4)', 'Finetuned / Local / GRB2-Abundance / 1D — (VNpi9Zjt)', 'Finetuned / Local / GRB2-Abundance / 3D — (PqBMjXkA)', 'Finetuned / Local / GRB2-Binding / 1D — (Z59BhUaE)', 'Finetuned / Local / GRB2-Binding / 3D — (VwcRN6UB)', 'Finetuned / Local / Pab1 / 1D — (TdjCzoQQ)', 'Finetuned / Local / Pab1 / 3D — (5SjoLx3y)', 'Finetuned / Local / PTEN-Abundance / 1D — (oUScGeHo)', 'Finetuned / Local / PTEN-Abundance / 3D — (DhuasDEr)', 'Finetuned / Local / PTEN-Activity / 1D — (m9UsG7dq)', 'Finetuned / Local / PTEN-Activity / 3D — (8Vi7ENcC)', 'Finetuned / Local / TEM-1 / 1D — (64ncFxBR)', 'Finetuned / Local / TEM-1 / 3D — (PncvgiJU)', 'Finetuned / Local / Ube4b / 1D — (e9uhhnAv)', 'Finetuned / Local / Ube4b / 3D — (NfbZL7jK)', 'GFP DESIGN / Finetuned / Local / GFP / 1D — (YoQkzoLD)', 'GFP DESIGN / Finetuned / Local / GFP / 3D — (PEkeRuxb)'], allow_custom_value=False)
+    #         metl_model_status = gr.HTML(get_color(RED, "Select Model"), elem_id="modelStatus")
+    #     with gr.Column():
+    #             upload_pdb = gr.File(label="PDB File upload", elem_id="pdbUpload", file_types=[".pdb", ".txt"])
+
+    with gr.Column():
+        metl_model_id = gr.Dropdown(label="METL model", choices=list(uuid_lookup.keys()), allow_custom_value=False)
+        metl_model_status = gr.HTML(get_color(RED, "Select Model"), elem_id="modelStatus")
+
+    with gr.Column():
+        upload_pdb = gr.File(label="PDB File upload", elem_id="pdbUpload", file_types=[".pdb", ".txt"])
 
     with gr.Column():
         metl_seq_input = gr.MultimodalTextbox(label="Input Protein Sequence", interactive=True, elem_classes="multiModalText", elem_id="wildTypeSequence")