Update app.py

app.py

@@ -16,14 +16,17 @@ 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
-selected_images = []
 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)
-            st.sidebar.image(image, caption=f"{img.name} {img.size / 1024.0:.1f} KB", width=40)
+        st.sidebar.image(image, caption=f"{img.name} {img.size / 1024.0:.1f} KB", width=40)
+
+# Collect selected images based on checkbox input
+selected_images = []
+for img in uploaded_images:
+    selected = st.sidebar.checkbox(f"{img.name} {img.size / 1024.0:.1f} KB", value=False, key=img.name)
+    if selected:
+        selected_images.append(Image.open(img))
 
 if st.button("Predict Emotions") and selected_images:
     if len(selected_images) == 2:
@@ -62,4 +65,4 @@ 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: {selected_images[i].filename}")  # Display original file name
+            st.write(f"Original File Name: {uploaded_images[i].name}")  # Display original file name