Update app.py

app.py

@@ -47,6 +47,7 @@ def searchEmbeddings(id, mod1, mod2):
         id = indx_to_id_dict[indx]
         id_list.append(id)
 
+    # get images
     image0 = get_image(dataset_image1, dataset_image2, dataset_image_mask, processid_to_index, original_indx)
     image1 = get_image(dataset_image1, dataset_image2, dataset_image_mask, processid_to_index, I[0][0])
     image2 = get_image(dataset_image1, dataset_image2, dataset_image_mask, processid_to_index, I[0][1])
@@ -59,20 +60,31 @@ def searchEmbeddings(id, mod1, mod2):
     image9 = get_image(dataset_image1, dataset_image2, dataset_image_mask, processid_to_index, I[0][8])
     image10 = get_image(dataset_image1, dataset_image2, dataset_image_mask, processid_to_index, I[0][9])
         
-    return id_list, image0, image1, image2, image3, image4, image5, image6, image7, image8, image9, image10
+    # get taxonomic information
+    s0 = getTax(original_indx)
+    s1 = getTax(I[0][0])
+    s2 = getTax(I[0][1])
+    s3 = getTax(I[0][2])
+    s4 = getTax(I[0][3])
+    s5 = getTax(I[0][4])
+    s6 = getTax(I[0][5])
+    s7 = getTax(I[0][6])
+    s8 = getTax(I[0][7])
+    s9 = getTax(I[0][8])
+    s10 = getTax(I[0][9])
+        
+    return id_list, image0, image1, image2, image3, image4, image5, image6, image7, image8, image9, image10, s0, s1, s2, s3, s4, s5, s6, s7, s8, s9, s10
 
 def getRandID():
     indx = random.randrange(0, 325667)
     return indx_to_id_dict[indx]
 
-
-# css = """
-# #image0 {
-#       display: block;
-#       margin-left: auto;
-#       margin-right: auto;
-# }
-# """
+def getTax(indx):
+    s = species[indx]
+    g = genus[indx]
+    f = family[indx]
+    str = "Species: " + s + "\nGenus: " + g + "\nFamily: " + f
+    return str
 
 with gr.Blocks() as demo:
     # open general files
@@ -97,6 +109,11 @@ with gr.Blocks() as demo:
     with open("id_to_dna_emb_dict.pickle", "rb") as f: 
         id_to_dna_emb_dict = pickle.load(f)
 
+    # open taxonomy files
+    family = [item.decode("utf-8") for item in dataset_hdf5_all_key["family"][:]]
+    genus = [item.decode("utf-8") for item in dataset_hdf5_all_key["genus"][:]]
+    species = [item.decode("utf-8") for item in dataset_hdf5_all_key["species"][:]]
+
     with gr.Column():
         process_id = gr.Textbox(label="ID:", info="Enter a sample ID to search for")
         process_id_list = gr.Textbox(label="Closest 10 matches:" )
@@ -105,28 +122,54 @@ with gr.Blocks() as demo:
         search_btn = gr.Button("Search")
 
         with gr.Row():
-            image0 = gr.Image(label="Original", height=550, elem_id="image0")
+            with gr.Column():
+                image0 = gr.Image(label="Original", height=550)
+                tax0 = gr.Textbox(label="Taxonomy")
             with gr.Column():
                 rand_id = gr.Textbox(label="Random ID:")
+                rand_id_indx = gr.Textbox(label="Index:")
                 id_btn = gr.Button("Get Random ID")
 
         with gr.Row():
-            image1 = gr.Image(label=1)
-            image2 = gr.Image(label=2)
-            image3 = gr.Image(label=3)
-            image4 = gr.Image(label=4)
-            image5 = gr.Image(label=5)
-            
+            with gr.Column():
+                image1 = gr.Image(label=1)
+                tax1 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image2 = gr.Image(label=2)
+                tax2 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image3 = gr.Image(label=3)
+                tax3 = gr.Textbox(label="Taxonomy")
+
+        with gr.Row():
+            with gr.Column():
+                image4 = gr.Image(label=4)
+                tax4 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image5 = gr.Image(label=5)
+                tax5 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image6 = gr.Image(label=6)
+                tax6 = gr.Textbox(label="Taxonomy")
+
         with gr.Row():   
-            image6 = gr.Image(label=6)
-            image7 = gr.Image(label=7)
-            image8 = gr.Image(label=8)
-            image9 = gr.Image(label=9)
-            image10 = gr.Image(label=10)
+            with gr.Column():
+                image7 = gr.Image(label=7)
+                tax7 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image8 = gr.Image(label=8)
+                tax8 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image9 = gr.Image(label=9)
+                tax9 = gr.Textbox(label="Taxonomy")
+            with gr.Column():
+                image10 = gr.Image(label=10)
+                tax10 = gr.Textbox(label="Taxonomy")
 
     id_btn.click(fn=getRandID, inputs=[], outputs=rand_id)
     search_btn.click(fn=searchEmbeddings, inputs=[process_id, mod1, mod2], 
-                     outputs=[process_id_list, image0, image1, image2, image3, image4, image5, image6, image7, image8, image9, image10])
+                     outputs=[process_id_list, image0, image1, image2, image3, image4, image5, image6, image7, image8, image9, image10, 
+                              tax0, tax1, tax2, tax3, tax4, tax5, tax6, tax7, tax8, tax9, tax10])
     examples = gr.Examples(
         examples=[["ABOTH966-22", "DNA", "DNA"],
                   ["CRTOB8472-22", "DNA", "Image"],