Update app.py

app.py

@@ -124,9 +124,28 @@ with gr.Blocks(theme=gr.themes.Glass(primary_hue=gr.themes.colors.stone)) as dem
                 # output_img_thumb = gr.Textbox(label="Image names")
         with gr.Row():
             with gr.Column():
+                 
                 image_name = gr.Textbox(label="""
                 Note: To check the image provide single name and to compare provide both the image names separated by a comma(,)
                 """)
+                image_name_lst = []
+                image_id_lst = []
+                image_name_str = ''
+                def get_select_index(evt: gr.SelectData):
+                    image_name = evt.value['image']['path']
+                    image_name_lst = image_name.split('/')
+                    if len(image_id_lst) >= 2 :
+                        image_id_lst[0] = image_id_lst[1]
+                        image_id_lst[1] = image_name_lst[4]
+                    else:
+                        image_id_lst.append(image_name_lst[4])
+                    if len(image_id_lst) == 1 :
+                        image_name_str = image_id_lst[0]
+                        return image_name_str
+                    else:
+                        image_name_str = image_id_lst[0] + ',' + image_id_lst[1]
+                        return image_name_str
+                img_gallery.select(get_select_index, None, image_name)
                 show_button = gr.Button(value="Show Images")
                 output_img = gr.Image(type="filepath",label="Selected Graph")