Updated app.py

app.py

@@ -66,9 +66,13 @@ if file is not None:
     for i, idx in enumerate(indices[0]):
         product_path = filenames[idx]
         product_path = product_path.replace('\\', '/')
-        st.write(product_path)
-        product_image = Image.open(product_path)
-        
-        # Assign columns dynamically
-        with cols[i % 3]:
-            st.image(product_image, caption=f"Product {i + 1}")
+
+        # Check if the image exists before displaying
+        if os.path.exists(product_path):
+            product_image = Image.open(product_path)
+            # Assign columns dynamically
+            with cols[i % 3]:
+                st.image(product_image, caption=f"Product {i + 1}")
+        else:
+            # If the image does not exist, skip it
+            st.write(f"Image for Product {i + 1} not found.")