Update app.py

app.py

@@ -16,27 +16,13 @@ st.write(f"{x} squared is {x * x}")
 uploaded_images = st.file_uploader("Upload images", type=["jpg", "png"], accept_multiple_files=True)
 
 # Display thumbnail images alongside file names and sizes in the sidebar
-@st.cache
-def load_image(img):
-    return Image.open(img)
-
-if uploaded_images:
-    for idx, img in enumerate(uploaded_images):
-        image = load_image(img)
-        # Generate unique keys for image display and checkbox
-        image_key = f"{img.name}_image_{idx}"
-        checkbox_key = f"{img.name}_checkbox_{idx}"
-        st.sidebar.image(image, caption=f"{img.name} {img.size / 1024.0:.1f} KB", width=40, key=image_key)
-        st.sidebar.checkbox(f"Select {img.name}", value=False, key=checkbox_key)
-
-# Collect selected images based on checkbox input
 selected_images = []
-for idx, img in enumerate(uploaded_images):
-    # Generate a unique key for each checkbox
-    checkbox_key = f"{img.name}_checkbox_{idx}"
-    selected = st.sidebar.checkbox(f"Select {img.name}", value=False, key=checkbox_key)
-    if selected:
-        selected_images.append(load_image(img))
+if uploaded_images:
+    for img in uploaded_images:
+        image = Image.open(img)
+        selected = st.sidebar.checkbox(f"{img.name} {img.size / 1024.0:.1f} KB", value=False, key=img.name)
+        if selected:
+            selected_images.append(image)
 
 if st.button("Predict Emotions") and selected_images:
     if len(selected_images) == 2:
@@ -75,4 +61,5 @@ if st.button("Predict Emotions") and selected_images:
             st.image(selected_images[i], caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
             st.write(f"Emotion Scores for #{i+1} Image")
             st.write(f"{predicted_emotion}: {result[0]['score']:.4f}")
-            st.write(f"Original File Name: {uploaded_images[i].name}")  # Display original file name
+            st.write(f"Original File Name: {selected_images[i].filename}")  # Display original file name
+