Update app.py

app.py

@@ -126,6 +126,8 @@ def main():
             start_index = (tabs - 1) * images_per_page
             end_index = min(start_index + images_per_page, length)
 
+            index = 0 
+
             for img_id in similar_images:
                 row = df[df['seller_img_id'] == img_id]
                 if not row.empty:
@@ -142,6 +144,8 @@ def main():
                     # Determine the column to place the image in
                     col_index = index % num_columns
                     st.image(resized_image, caption=f'Product ID: {product_id}\nImage ID: {img_id}', width=image_width)
+                    index += 1
+                    
                 else:
                     st.write(f"Could not find details for image with ID: {img_id}")