uniprot ID added to display

app.py

@@ -37,7 +37,10 @@ def limit_n_results(n):
 def aggregate_search_results(raw_results: List[dict], max_res: int) -> Dict[str, dict]:
     aggregated_by_gene = collections.defaultdict(list)
     for raw_result in raw_results:
-        entry = select_keys(raw_result, ["pdb_name", "chain_id", "score", "organism"])
+        entry = select_keys(
+            raw_result,
+            keys=["pdb_name", "chain_id", "score", "organism", "uniprot_id"]
+        )
         genes = raw_result["genes"]
         if genes is not None:
             gene_names = genes.split(" ")
@@ -58,7 +61,8 @@ def format_search_results(agg_search_results):
         entry = entries[0]
         organism = entry["organism"]
         score = entry["score"]
-        key = f"Gene: {gene} | Organism: {organism}"
+        uniprot_id = entry["uniprot_id"]
+        key = f"Gene: {gene} | Organism: {organism} | Uniprot ID: {uniprot_id}"
         formatted_search_results[key] = score
     return formatted_search_results