Update UI

src/app.py

@@ -92,18 +92,53 @@ def get_neighbors(query_image, selected_descriptor, top_k=5):
         return []
 
 
+def load_cbir_dataset(datasetname, size):
+    pass
+
 # Define the Gradio Interface
 
 with gr.Blocks() as demo:
-    image_input = gr.Image(type="pil", label="Please upload an image")
+    gr.Markdown("""
+    # Welcome to this CBIR app
+
+    This is a CBIR app focused on the retrieval of similar faces.
+
+    ## Find similar images
+    Here you can upload an image, that is compared with existing image in our dataset.
+    """)
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Please upload your image")
+        gallery_output = gr.Gallery()
+
     checkboxes_descr = gr.CheckboxGroup(["Color Descriptor", "LBP", "CLIP"], label="Please choose an descriptor")
 
     btn = gr.Button(value="Submit")
-    gallery_output = gr.Gallery()
+    gr.Markdown("""
+    ## Settings
+    Here you can adjust how your image is compared
+    """)
+
+    with gr.Row():
+        checkboxes_descr = gr.CheckboxGroup(["Color Descriptor", "LBP", "CLIP"], label="Please choose an descriptor")
+        dataset_dropdown = gr.Dropdown(
+        ["huggan/CelebA-faces", "EIT/cbir-eit"],
+        value=["huggan/CelebA-faces"]
+    )
+        btn_index = gr.Button(value="Switch Dataset")
+    btn_index.click(load_cbir_dataset, inputs=[dataset_dropdown, 100], outputs=[dataset_with_embeddings])
     btn.click(get_neighbors, inputs=[image_input, checkboxes_descr], outputs=[gallery_output])
 
-    btn_index = gr.Button(value="Re-index Dataset")
-    btn_index.click(index_dataset)
+    # gr.Markdown(
+    # """
+    # ## Upload your own data to the CBIR
+    # WARNING! Please be aware, that what you are uploading here, will be public to anyone.
+    # Don't upload images for which you don't have the rights to do so.
+    # """)
+
+    # file_output = gr.File()
+    # upload_button = gr.UploadButton("Click to upload a file", file_types=["image"], file_count="multiple")
+    # upload_button.upload(upload_file, upload_button, file_output)
+
 
 if __name__ == "__main__":
     demo.launch()