Update app.py

app.py

@@ -113,56 +113,52 @@ if "original_image" in st.session_state:
     original_img = st.session_state["original_image"]
     processed_img = resize_image(original_img)
 
-    # 🖼️ Panel 1: Original Image
-    with col1:
-        st.subheader("📷 Original Image")
-        st.image(original_img, caption=f"Uploaded ({original_img.width}x{original_img.height} pixels)", use_column_width=True)
+# Adjusting Layout: Original Image (2 Columns) | Processed Image (2 Columns)
+col1, col2 = st.columns(2)  # Two columns for images
 
-    # 🖼️ Panel 2: Resized Image (512x512 Maintaining Aspect Ratio)
-    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)
+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)
 
-# 📝 Annotation Table
-st.subheader("📝 Add Annotations")
+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)
 
-if "annotations" not in st.session_state:
-    st.session_state["annotations"] = []
 
-# Create dataframe for annotations
-annotation_df = pd.DataFrame(st.session_state["annotations"], columns=["Food Item", "Unit", "Quantity", "Ingredients"])
+# Create a table-like structure for food annotation
+st.subheader("📝 Add Annotations")
 
-# Food input with intellisense
-food_item = st.selectbox("Select food item or type custom:", FOOD_SUGGESTIONS + ["Other"])
-if food_item == "Other":
-    food_item = st.text_input("Enter food item manually:")
+# Define columns for table structure
+col_food, col_qty, col_unit, col_ing = st.columns([2, 1, 1, 2])  # Define column widths
 
-unit = st.selectbox("Select unit:", UNIT_OPTIONS)
-quantity = st.number_input("Enter quantity:", min_value=0.1, step=0.1)
+# Food Item Dropdown (with Intellisense)
+food_item = col_food.selectbox("Food Item", options=suggested_foods, index=None, placeholder="Select or enter food")
 
-# Dynamic ingredient input
-ingredients = st.text_area("Enter ingredients (one per line):")
-ingredient_list = [i.strip() for i in ingredients.split("\n") if i.strip()]
+# Quantity Input
+quantity = col_qty.number_input("Qty", min_value=0, step=1)
 
-if st.button("Add Food Item"):
-    if food_item and unit and quantity:
-        st.session_state["annotations"].append([food_item, unit, quantity, ingredient_list])
-        st.success(f"✅ Added {food_item} successfully!")
+# Unit Dropdown (Grams, Ounces, etc.)
+unit = col_unit.selectbox("Unit", ["g", "oz", "tsp", "tbsp", "cup", "slice", "piece"])
 
-# Display annotation table
-st.dataframe(pd.DataFrame(st.session_state["annotations"], columns=["Food Item", "Unit", "Quantity", "Ingredients"]))
+# Ingredients Section with Expandable Input
+with col_ing:
+    ingredients = []
+    if st.button("+ Add Ingredients"):
+        ingredient = st.text_input("Enter ingredient")
+        if ingredient:
+            ingredients.append(ingredient)
 
-# Save Annotations to DynamoDB
+# Display added annotations in a structured format
 if st.button("Save Annotations"):
-    metadata_table.update_item(
-        Key={"image_id": uploaded_file.name},
-        UpdateExpression="SET user_id = :u, annotations = :a, processing_status = :p, s3_url = :s, tokens_earned = :t",
-        ExpressionAttributeValues={
-            ":u": st.session_state["user_id"],
-            ":a": st.session_state["annotations"],
-            ":p": "processed",
-            ":s": f"s3://{S3_BUCKET_NAME}/raw-uploads/{uploaded_file.name}",
-            ":t": len(st.session_state["annotations"])
-        },
-    )
-    st.success("✅ Annotations saved successfully!")
+    new_entry = {
+        "name": food_item,
+        "quantity": quantity,
+        "unit": unit,
+        "ingredients": ingredients,
+    }
+    st.session_state["annotations"].append(new_entry)
+    st.success("✅ Annotation saved!")
+
+# Show existing annotations
+st.write("### Current Annotations")
+st.table(st.session_state["annotations"])