filename_textbox

app.py

@@ -118,11 +118,15 @@ def image_retrieval_from_text(text):
 
 
 def image_retrieval_from_image(state, selected_locally):
+    if state is None or len(state) == 0:
+        return [], []
     selected = state[int(selected_locally)]
     return features_to_gallery(embeddings[selected])
 
 
 def show_folder(state, selected_locally):
+    if state is None or len(state) == 0:
+        return [], []
     selected = state[int(selected_locally)]
     target_folder = folders[selected]
     indices = []
@@ -146,22 +150,27 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
         ).style(container=False)
         text_query_button = gr.Button("Search").style(full_width=False)
 
-    gallery = gr.Gallery(label="Images", show_label=False, elem_id="gallery"
-        ).style(columns=4, container=False)
-
     with gr.Row():
         show_folder_button = gr.Button("Show folder")
         image_query_button = gr.Button("Show similar")
         selected = gr.Number(0, show_label=False, visible=False)
 
+    gallery = gr.Gallery(label="Images", show_label=False, elem_id="gallery"
+        ).style(columns=4, container=False)
+
+    with gr.Row():
+        filename_textbox = gr.Textbox("", show_label=False).style(container=False)
+
     text_query_button.click(image_retrieval_from_text, [text], [gallery, state])
     image_query_button.click(image_retrieval_from_image, [state, selected], [gallery, state])
     show_folder_button.click(show_folder, [state, selected], [gallery, state])
 
-    def get_select_index(evt: gr.SelectData):
-        return evt.index
+    def get_select_index(evt: gr.SelectData, state):
+        selected_locally = evt.index
+        selected = state[int(selected_locally)]
+        return selected_locally, filenames[selected]
 
-    gallery.select(get_select_index, None, selected)
+    gallery.select(get_select_index, [state], [selected, filename_textbox])
 
 
 if __name__ == "__main__":