x

app.py

@@ -36,13 +36,19 @@ col1.subheader(descr)
 img = images[index]
 images[index] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=20,fill='blue')
 
-col1.image(img, use_column_width=True)
 caps = list(range(10))
 
 if show:
+    if index == idx:
+        cap = f'{index} (GROUNDTRUTH)'
     caps[idx] = f'{idx} (GROUNDTRUTH)'
+    images[idx] = ImageOps.expand(Image.open(io.BytesIO(requests.get(images[index], stream=True).content)),border=50,fill='green')
 if hide:
+    if index == idx:
+        cap = f'{index}'
     caps[idx] = f'{idx}'
+    images[idx] = Image.open(io.BytesIO(requests.get(images[index], stream=True).content))
 # shown = True if len(caps[idx]) > 2 else False
 # if shown 
+col1.image(img, use_column_width=True, caption=cap)
 col2.image(images, width=175, caption=caps)