Add: Switch from gradio interface to demo for adding re-index button

src/app.py

@@ -8,7 +8,7 @@ from datasets import *
 dataset = load_dataset("huggan/CelebA-faces")
 candidate_subset = dataset["train"].select(range(1000)) # This is a small CBIR app! :D
 
-def emb_dataset(dataset):
+def index_dataset(dataset):
     # This function might need to be split up, to reduce start-up time of app
     # It could also use batches to increase speed
     # If indexes are saved in files, this is all not really necessary
@@ -29,7 +29,7 @@ def emb_dataset(dataset):
     print(dataset_with_embeddings)
     return dataset_with_embeddings
 
-dataset_with_embeddings = emb_dataset(candidate_subset)
+dataset_with_embeddings = index_dataset(candidate_subset)
 
 # Main function, to find similar images
 # TODO: implement different distance measures
@@ -81,3 +81,20 @@ iface = gr.Interface(
 
 # Launch the Gradio interface
 iface.launch()
+
+with gr.Blocks() as demo:
+    image_input = gr.Image(type="pil", label="Please upload an image")
+    checkboxes_descr = gr.CheckboxGroup(["Color Descriptor", "LBP", "CLIP"], label="Please choose an descriptor"))
+
+    btn = gr.Button(value="Submit")
+    gallery_output = gr.Gallery()
+
+    btn.click(get_neighbors,
+              inputs=[image_input, checkboxes_descr],
+              outputs=[gallery_output]
+              )
+    btn_index = gr.Button(value="Re-index Dataset")
+    btn.click(index_dataset)
+
+if __name__ == "__main__":
+    demo.launch()