Update app.py

app.py

@@ -233,9 +233,20 @@ def save_metadata(user_id, s3_path, food_name, portion_size, portion_unit, cooki
             's3_path': s3_path,
             'tokens_awarded': tokens_awarded
         }
-        # Add debug logging
+
+        # Add debug logging with more details
         if debug_mode:
             st.sidebar.info(f"DynamoDB Save: {image_id} - {food_name}")
+            if "test_debug_details" not in st.session_state:
+                st.session_state["test_debug_details"] = []
+            
+            # Store details for debugging
+            st.session_state["test_debug_details"].append({
+                "time": datetime.now().isoformat(),
+                "image_id": image_id,
+                "food": food_name,
+                "s3_path": s3_path
+            })
         
         # Save to DynamoDB
         metadata_table.put_item(Item=item)
@@ -468,49 +479,112 @@ if debug_mode and "processed_image" in st.session_state:
                 else:
                     st.error("❌ Processed upload failed!")
 
-# Add DynamoDB Test Panel for Debug Mode
+# Add DynamoDB Test Panel for Multiple Food Items
 if debug_mode and "processed_image" in st.session_state:
-    with st.expander("🔍 DEBUG: DynamoDB Test", expanded=True):
-        st.warning("⚠️ Debug Mode: Test DynamoDB functionality")
-        test_db_col1, test_db_col2 = st.columns(2)
-        with test_db_col1:
-            test_food = st.text_input("Test Food Name", "TEST_ITEM")
-            test_portion = st.number_input("Test Portion", value=1.0)
-        with test_db_col2:
-            test_unit = st.selectbox("Test Unit", UNIT_OPTIONS)
-            test_cooking = st.selectbox("Test Cooking", [""] + COOKING_METHODS)
-        
-        test_ingredients = st_tags.st_tags(
-            label="Test Ingredients",
-            text="Press enter to add",
-            value=["test_ingredient"],
-            maxtags=3
-        )
+    with st.expander("🔍 DEBUG: Complete Food Annotation Test", expanded=True):
+        st.warning("⚠️ Debug Mode: Test Complete Flow with Multiple Food Items")
         
-        test_db_button = st.button("Test DynamoDB Save")
-        if test_db_button:
-            # Use the processed path from testing if available
-            s3_path = st.session_state.get("test_processed_s3_path", "test/processed-512x512/test_image.jpg")
+        # Create a test session state if it doesn't exist
+        if "test_food_items" not in st.session_state:
+            st.session_state["test_food_items"] = []
             
-            # Convert to Decimal for DynamoDB
-            portion_size_decimal = Decimal(str(test_portion))
-            
-            # Call save_metadata with test data
-            success = save_metadata(
-                st.session_state["user_id"], 
-                s3_path,
-                f"TEST_{test_food}",
-                portion_size_decimal,
-                test_unit,
-                test_cooking,
-                test_ingredients,
-                1  # Token value for test
+        # Form for adding test food items
+        with st.form(key="test_food_form"):
+            test_food = st.text_input("Food Name", "Test Pizza")
+            test_portion = st.number_input("Portion Size", value=1.0)
+            test_unit = st.selectbox("Unit", UNIT_OPTIONS)
+            test_cooking = st.selectbox("Cooking Method", COOKING_METHODS)
+            test_ingredients = st_tags.st_tags(
+                label="Ingredients",
+                text="Press enter to add",
+                value=["Test Ingredient"],
+                maxtags=5
             )
             
-            if success:
-                st.success("✅ DynamoDB save successful!")
-            else:
-                st.error("❌ DynamoDB save failed!")
+            test_add_button = st.form_submit_button("Add Test Food Item")
+            
+            if test_add_button:
+                st.session_state["test_food_items"].append({
+                    "food_name": f"TEST_{test_food}",
+                    "portion_size": test_portion,
+                    "portion_unit": test_unit,
+                    "cooking_method": test_cooking,
+                    "ingredients": test_ingredients
+                })
+                st.success(f"Added test food item: {test_food}")
+                st.rerun()
+        
+        # Display test food items
+        if st.session_state["test_food_items"]:
+            st.subheader("Test Food Items")
+            for i, item in enumerate(st.session_state["test_food_items"]):
+                with st.expander(f"Item #{i+1}: {item['food_name']}"):
+                    st.write(f"**Portion:** {item['portion_size']} {item['portion_unit']}")
+                    st.write(f"**Cooking Method:** {item['cooking_method']}")
+                    st.write(f"**Ingredients:** {', '.join(item['ingredients'])}")
+                    if st.button(f"Remove Test Item #{i+1}", key=f"remove_test_{i}"):
+                        st.session_state["test_food_items"].pop(i)
+                        st.rerun()
+            
+            # Button to run the full test
+            test_full_button = st.button("Run Complete Annotation Test")
+            if test_full_button and st.session_state["test_food_items"]:
+                with st.spinner("Testing complete annotation flow..."):
+                    # Upload both original and processed images
+                    raw_s3_path = upload_to_s3(original_img, st.session_state["user_id"], folder="raw-uploads")
+                    processed_s3_path = upload_to_s3(processed_img, st.session_state["user_id"], folder="processed-512x512")
+                    
+                    all_saved = True
+                    saved_items = 0
+                    
+                    if raw_s3_path and processed_s3_path:
+                        # Set up a results area
+                        results_area = st.empty()
+                        
+                        # Save each food item
+                        for item in st.session_state["test_food_items"]:
+                            try:
+                                # Convert to Decimal
+                                portion_size_decimal = Decimal(str(item["portion_size"]))
+                                
+                                # Save to DynamoDB
+                                success = save_metadata(
+                                    st.session_state["user_id"],
+                                    processed_s3_path,
+                                    item["food_name"],
+                                    portion_size_decimal,
+                                    item["portion_unit"],
+                                    item["cooking_method"],
+                                    item["ingredients"],
+                                    1  # Test token value
+                                )
+                                
+                                if success:
+                                    saved_items += 1
+                                else:
+                                    all_saved = False
+                                    break
+                            except Exception as e:
+                                results_area.error(f"Error saving {item['food_name']}: {e}")
+                                all_saved = False
+                                break
+                        
+                        if all_saved:
+                            st.success(f"✅ Successfully saved {saved_items} food items!")
+                            st.json({
+                                "user_id": st.session_state["user_id"],
+                                "s3_paths": {
+                                    "raw": raw_s3_path,
+                                    "processed": processed_s3_path
+                                },
+                                "food_items": st.session_state["test_food_items"]
+                            })
+                        else:
+                            st.error("❌ Failed to save all food items")
+                    else:
+                        st.error("❌ Failed to upload images to S3")
+        else:
+            st.info("Add test food items above to run a complete annotation test")
     
     # Display existing food annotations if any
     if st.session_state["food_items"]:
@@ -726,4 +800,13 @@ if debug_mode and st.sidebar.checkbox("Show Cleanup Tools"):
                 
                 st.sidebar.success(f"Deleted {deleted_records}/{total_records} test records and their S3 objects")
             except Exception as e:
-                st.sidebar.error(f"Cleanup error: {e}")
+                st.sidebar.error(f"Cleanup error: {e}")
+
+
+# Add Debug Info Panel at the end of your app
+if debug_mode and "test_debug_details" in st.session_state and st.session_state["test_debug_details"]:
+    with st.expander("🔍 DEBUG: Last Test Results"):
+        st.json(st.session_state["test_debug_details"])
+        if st.button("Clear Debug History"):
+            st.session_state["test_debug_details"] = []
+            st.rerun()