Update src/about.py

src/about.py

@@ -103,4 +103,19 @@ benchmark_specific_metrics = {
     'Benchmark 2': ['Metric B1', 'Metric B2', 'Metric B3'],
     'Benchmark 3': ['Metric C1', 'Metric C2', 'Metric C3'],
     'affinity': ['affinity_mse_train', 'affinity_mse_val']
-}
+}
+
+display_labels = [
+    'BLAST', 'HMMER', 'K-SEP', 'APAAC', 'PFAM', 'AAC', 'PROTVEC',
+    'GENE2VEC', 'LEARNED-VEC', 'MUT2VEC', 'TCGA-EMBEDDING', 'CPC-PROT', 
+    'SEQVEC', 'BERT-BFD', 'BERT-PFAM', 'ESMB1', 'ALBERT', 'XLNET', 'UNIREP', 'T5'
+]
+
+group_colors = [
+    'green', 'green', 'green', 'green', 'green', 'green', 'blue',
+    'blue', 'blue', 'blue', 'blue', 'blue', 'red', 'red',
+    'red', 'red', 'red', 'red', 'red', 'red'
+]
+
+# Create a color dictionary from the labels and colors
+color_dict = dict(zip(display_labels, group_colors))