adding the organism to the label

app.py

@@ -24,7 +24,7 @@ def limit_n_results(n):
 def update_dropdown_menu(search_res):
     choices = []
     for row in search_res:
-        choice = ",".join([row["pdb_name"], row["chain_id"]])
+        choice = ".".join([row["pdb_name"], row["chain_id"]])
         choices.append(choice)
 
     return gr.Dropdown.update(
@@ -47,15 +47,15 @@ def format_search_result(raw_result):
     prot = raw_result["pdb_name"]
     chain = raw_result["chain_id"]
     value = raw_result["score"]
-    gene_name = get_gene_name(pdb_id=prot, chain_id=chain)
-    key = f"PDB: {prot} | Chain: {chain}"
+    gene_name, species = get_gene_name(pdb_id=prot, chain_id=chain)
+    key = f"PDB: {prot}.{chain}"
     if gene_name != "Unknown":
-        key += f" | Gene: {gene_name}"
+        key += f" | Gene: {gene_name} | Organism: {species}"
     return key, value
 
 
 def switch_viz(new_choice):
-    choice_parts = new_choice.split(",")
+    choice_parts = new_choice.split(".")
     pdb_id, chain = choice_parts[0], choice_parts[1]
     title_update = gr.Markdown.update(visible=True)
     protein_name = get_protein_name(pdb_id)

protein_viz.py

@@ -32,8 +32,10 @@ def render_html(pdb_id, chain):
 
 def get_gene_name(pdb_id, chain_id):
     entity_id = get_polymer_entity_id(chain_id, pdb_id)
-    gene_name = get_gene_name_from_polymer_entity(pdb_id=pdb_id, entity_id=entity_id)
-    return gene_name
+    gene_name, species = get_gene_name_from_polymer_entity(
+        pdb_id=pdb_id, entity_id=entity_id
+    )
+    return gene_name, species
 
 
 def get_polymer_entity_id(chain_id, pdb_id):
@@ -59,13 +61,13 @@ def get_gene_name_from_polymer_entity(pdb_id, entity_id):
         if response.ok:
             res_data = response.json()
             try:
-                gene_name = res_data["rcsb_entity_source_organism"][0][
-                    "rcsb_gene_name"
-                ][0]["value"]
-                return gene_name
+                source_organism = res_data["rcsb_entity_source_organism"][0]
+                gene_name = source_organism["rcsb_gene_name"][0]["value"]
+                species = source_organism["ncbi_scientific_name"]
+                return gene_name, species
             except KeyError:
                 pass
-    return "Unknown"
+    return "Unknown", "Unknown"
 
 
 def get_protein_name(pdb_id: str):