Update app.py

app.py

@@ -18,9 +18,10 @@ uploaded_images = st.file_uploader("Upload images", type=["jpg", "png"], accept_
 # Display thumbnail images alongside file names and sizes in the sidebar
 selected_images = []
 if uploaded_images:
-    for img in uploaded_images:
+    for idx, img in enumerate(uploaded_images):
         image = Image.open(img)
-        selected = st.sidebar.checkbox(f"{img.name} {img.size / 1024.0:.1f} KB", value=False, key=img.name)
+        checkbox_key = f"{img.name}_checkbox_{idx}"  # Unique key for each checkbox
+        selected = st.sidebar.checkbox(f"{img.name} {img.size / 1024.0:.1f} KB", value=False, key=checkbox_key)
         if selected:
             selected_images.append(image)
 
@@ -37,7 +38,8 @@ if st.button("Predict Emotions") and selected_images:
             col = col1 if i == 0 else col2
             col.image(selected_images[i], caption=f"Predicted emotion: {predicted_emotion}", use_column_width=True)
             col.write(f"Emotion Scores: {predicted_emotion}: {results[i][0]['score']:.4f}")
-            col.write(f"Original File Name: {uploaded_images[i].name}")  # Display original file name
+            # Use the index to get the corresponding filename
+            col.write(f"Original File Name: {uploaded_images[i].name}")  
 
         # Display the keys and values of all results
         st.write("Keys and Values of all results:")
@@ -61,5 +63,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: {selected_images[i].filename}")  # Display original file name
-
+            # Use the index to get the corresponding filename
+            st.write(f"Original File Name: {selected_images[i].filename}")