Update app.py

app.py

@@ -126,11 +126,11 @@ def main():
             start_index = (tabs - 1) * images_per_page
             end_index = min(start_index + images_per_page, length)
 
-            # Loop through each row in the CSV data for the selected page
+            # Loop through each similar image
             for index in range(start_index, end_index):
-                img_id = df.loc[index, 'seller_img_id']
-                img_path = df.loc[index, 'img_path']
-                product_id = df.loc[index, 'product_id']
+                img_id = similar_images[index]
+                img_path = df.loc[df['seller_img_id'] == img_id, 'img_path'].values[0]
+                product_id = df.loc[df['seller_img_id'] == img_id, 'product_id'].values[0]
 
                 # Construct the full path to the image
                 full_img_path = img_path
@@ -142,6 +142,7 @@ 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)
+        
 
 if __name__ == '__main__':
     main()