refine key and query types

app.py

@@ -49,7 +49,7 @@ def chooseDNAIndex(indexType):
         return dna_index_LSH
 
 
-def searchEmbeddings(id, mod1, mod2, indexType):
+def searchEmbeddings(id, key_type, query_type, index_type):
     # variable and index initialization
     dim = 768
     count = 0
@@ -58,15 +58,15 @@ def searchEmbeddings(id, mod1, mod2, indexType):
     index = faiss.IndexFlatIP(dim)
 
     # get index
-    if mod2 == "Image":
-        index = chooseImageIndex(indexType)
-    elif mod2 == "DNA":
-        index = chooseDNAIndex(indexType)
+    if query_type == "Image":
+        index = chooseImageIndex(index_type)
+    elif query_type == "DNA":
+        index = chooseDNAIndex(index_type)
 
     # search for query
-    if mod1 == "Image":
+    if key_type == "Image":
         query = id_to_image_emb_dict[id]
-    elif mod1 == "DNA":
+    elif key_type == "DNA":
         query = id_to_dna_emb_dict[id]
     query = query.astype(np.float32)
     D, I = index.search(query, num_neighbors)
@@ -118,8 +118,8 @@ with gr.Blocks() as demo:
                 rand_id_indx = gr.Textbox(label="Index:")
                 id_btn = gr.Button("Get Random ID")
             with gr.Column():
-                key_type = gr.Radio(choices=["DNA", "Image"], label="Search From:")
-                query_type = gr.Radio(choices=["DNA", "Image"], label="Search To:")
+                key_type = gr.Radio(choices=["Image", "DNA"], label="Search From:", value="Image")
+                query_type = gr.Radio(choices=["Image", "DNA"], label="Search To:", value="Image")
 
         index_type = gr.Radio(
             choices=["FlatIP(default)", "FlatL2", "HNSWFlat", "IVFFlat", "LSH"], label="Index:", value="FlatIP(default)"