Update app.py

app.py

@@ -58,30 +58,31 @@ def searchEmbeddings(id, mod1, mod2):
     image10 = get_image(dataset_image1, dataset_image2, dataset_image_mask, processid_to_index, I[0][9])
 
     # 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])
+    # 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
+    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], indx
 
-def getTax(indx):
-    s = species[indx]
-    g = genus[indx]
-    f = family[indx]
-    str = "Species: " + s + "\nGenus: " + g + "\nFamily: " + f
-    return str
+# def getTax(indx):
+#     s = species[indx]
+#     g = genus[indx]
+#     f = family[indx]
+#     str = "Species: " + s + "\nGenus: " + g + "\nFamily: " + f
+#     return str
 
 with gr.Blocks(title="Bioscan-Clip") as demo:
     # open general files
@@ -107,12 +108,12 @@ with gr.Blocks(title="Bioscan-Clip") as demo:
         id_to_dna_emb_dict = pickle.load(f)
 
     # open taxonomy files
-    with open("family.pickle", "rb") as f:
-        family = [item.decode("utf-8") for item in pickle.load(f)]
-    with open("genus.pickle", "rb") as f:
-        genus= [item.decode("utf-8") for item in pickle.load(f)]
-    with open("species.pickle", "rb") as f:
-        species = [item.decode("utf-8") for item in pickle.load(f)]
+    # with open("family.pickle", "rb") as f:
+    #     family = [item.decode("utf-8") for item in pickle.load(f)]
+    # with open("genus.pickle", "rb") as f:
+    #     genus= [item.decode("utf-8") for item in pickle.load(f)]
+    # with open("species.pickle", "rb") as f:
+    #     species = [item.decode("utf-8") for item in pickle.load(f)]
 
     with gr.Column():
         process_id = gr.Textbox(label="ID:", info="Enter a sample ID to search for")
@@ -168,8 +169,8 @@ with gr.Blocks(title="Bioscan-Clip") as demo:
 
     id_btn.click(fn=getRandID, inputs=[], outputs=[rand_id, rand_id_indx])
     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, 
-                              tax0, tax1, tax2, tax3, tax4, tax5, tax6, tax7, tax8, tax9, tax10])
+                     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"],