Update app.py

app.py

@@ -98,31 +98,43 @@ st.checkbox("I agree to the terms and conditions", key="terms_accepted")
 
 # Upload Image
 uploaded_file = st.file_uploader("Upload an image of your food", type=["jpg", "png", "jpeg"])
+# Ensure an image is uploaded before displaying
 if uploaded_file:
-    image = Image.open(uploaded_file)
-    st.session_state["original_image"] = image
-
-# Resize image in Streamlit before sending to S3
-def resize_image(image, new_width=512):
-    aspect_ratio = image.width / image.height
-    new_height = int(new_width / aspect_ratio)
-    resized_image = image.resize((new_width, new_height))
-    return resized_image
+    original_img = Image.open(uploaded_file)
+    st.session_state["original_image"] = original_img
 
+# If an image has been uploaded, process and display it
 if "original_image" in st.session_state:
     original_img = st.session_state["original_image"]
-    processed_img = resize_image(original_img)
+    
+    # Resize while maintaining aspect ratio (longest side = 512)
+    def resize_image(image, max_size=512):
+        aspect_ratio = image.width / image.height
+        if image.width > image.height:
+            new_width = max_size
+            new_height = int(max_size / aspect_ratio)
+        else:
+            new_height = max_size
+            new_width = int(max_size * aspect_ratio)
+        return image.resize((new_width, new_height))
 
-# Adjusting Layout: Original Image (2 Columns) | Processed Image (2 Columns)
-col1, col2 = st.columns(2)  # Two columns for images
-
-with col1:
-    st.subheader("📷 Original Image")
-    st.image(original_img, caption=f"Original ({original_img.width}x{original_img.height} pixels)", use_column_width=True)
+    processed_img = resize_image(original_img)
 
-with col2:
-    st.subheader("🖼️ Processed Image")
-    st.image(processed_img, caption=f"Processed ({processed_img.width}x{processed_img.height} pixels)", use_column_width=True)
+    # UI Layout: Two Columns for Images
+    col1, col2 = st.columns(2)
+
+    with col1:
+        st.subheader("📷 Original Image")
+        st.image(original_img, caption=f"Original ({original_img.width}x{original_img.height} pixels)", use_column_width=True)
+
+    with col2:
+        st.subheader("🖼️ Processed Image")
+        st.image(processed_img, caption=f"Processed ({processed_img.width}x{processed_img.height} pixels)", use_column_width=True)
+        
+    # Store processed image in session state for annotations
+    st.session_state["processed_image"] = processed_img
+else:
+    st.warning("⚠️ Please upload an image first.")
 
 
 # Create a table-like structure for food annotation