Update app.py

app.py

@@ -19,18 +19,17 @@ uploaded_images = st.file_uploader("Upload images", type=["jpg", "png"], accept_
 if uploaded_images:
     for img in uploaded_images:
         image = Image.open(img)
-        col1, col2 = st.columns([0.1, 1])
-        with col1:
-            st.image(image, caption=f"{img.name} {img.size / 1024.0:.1f} KB", width=40)
-        with col2:
-            st.sidebar.checkbox(f"{img.name} {img.size / 1024.0:.1f} KB", value=False, key=img.name)
+        st.sidebar.image(image, caption=f"{img.name} {img.size / 1024.0:.1f} KB", width=40)
+        st.sidebar.checkbox(f"Select {img.name}")
 
 # Collect selected images based on checkbox input
 selected_images = []
+selected_image_names = []
 for img in uploaded_images:
-    selected = st.sidebar.checkbox(f"{img.name} {img.size / 1024.0:.1f} KB", value=False, key=img.name)
+    selected = st.sidebar.checkbox(f"Select {img.name}", value=False, key=img.name)
     if selected:
         selected_images.append(Image.open(img))
+        selected_image_names.append(img.name)
 
 if st.button("Predict Emotions") and selected_images:
     if len(selected_images) == 2:
@@ -45,7 +44,7 @@ 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
+            col.write(f"Original File Name: {selected_image_names[i]}")  # Display original file name
 
         # Display the keys and values of all results
         st.write("Keys and Values of all results:")
@@ -69,4 +68,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: {uploaded_images[i].name}")  # Display original file name
+            st.write(f"Original File Name: {selected_image_names[i]}")  # Display original file name