added model validation

app.py

@@ -14,6 +14,8 @@ PROJECT_URL = "https://www.nature.com/articles/s41598-022-12180-x"
 
 DEFAULT_PATH_DOCKER = "/home/user/app"
 
+ENABLED_MODELS = ["ada"]
+
 def load_and_filter_data(protein_id, ligand_smiles):
     
     # generate random short id, make short
@@ -23,7 +25,12 @@ def load_and_filter_data(protein_id, ligand_smiles):
     
     # check that ligand_smiles is not empty
     if not ligand_smiles or ligand_smiles.strip() == "":
-        error_msg = f"!SMILES string is required"
+        error_msg = f"!SMILES string is required 💥"
+        raise gr.Error(error_msg, duration=5)
+    
+    
+    if protein_id not in ENABLED_MODELS:
+        error_msg = f"!Invalid 💥 target protein ID, the available options are: {ENABLED_MODELS}. To do inference other proteins, you can run the model locally an train the model for each target protein."
         raise gr.Error(error_msg, duration=5)
     
     # Split the input SMILES string by ':' to get a list
@@ -99,7 +106,7 @@ def create_interface():
             with gr.Column():
                 gr.Markdown("#### Target Protein")
                 protein_id = gr.Dropdown(
-                    choices=["ada"],
+                    choices=ENABLED_MODELS,
                     label="Target Protein ID",
                     info="Select the target protein from the dropdown menu.",
                     value="ada"