Update app.py

app.py

@@ -15,9 +15,6 @@ import pandas as pd
 import streamlit_tags as st_tags
 from dotenv import load_dotenv
 
-# Add debug mode detection here
-debug_mode = "debug" in st.query_params
-
 # Load environment variables from .env file if it exists
 load_dotenv()
 
@@ -193,10 +190,6 @@ def upload_to_s3(image, user_id, folder=""):
         quality = 95 if folder == "raw-uploads" else 85
         image.save(buffer, format="JPEG", quality=quality, optimize=True)
         buffer.seek(0)
-
-        # Add debug logging
-        if debug_mode:
-            st.sidebar.info(f"S3 Upload Path: {s3_path}")
         
         # Upload to S3
         s3.upload_fileobj(buffer, S3_BUCKET_NAME, s3_path)
@@ -233,20 +226,6 @@ 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 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)
@@ -347,13 +326,6 @@ if auth_option == "Logout" and "user_id" in st.session_state:
     del st.session_state["user_id"]
     st.sidebar.success("✅ Logged out successfully!")
 
-# Add Debug Mode Indicator to sidebar
-if debug_mode:
-    st.sidebar.warning("⚠️ DEBUG MODE ACTIVE")
-    test_mode = st.sidebar.checkbox("Mark submissions as tests")
-    if test_mode:
-        st.sidebar.info("Test mode enabled - all submissions will be prefixed with 'TEST_'")
-
 # Ensure user is logged in before uploading
 if "user_id" not in st.session_state and not st.session_state.get("demo_mode", False):
     st.warning("⚠️ Please log in to upload images.")
@@ -440,152 +412,6 @@ if "original_image" in st.session_state:
     if size_reduction > 5:  # Only show if there's a meaningful reduction
         st.success(f"✅ Image size reduced by {size_reduction:.1f}% for faster uploads and processing")
 
-# Add S3 Upload Test Panel for Debug Mode with proper subfolder structure
-if debug_mode and "processed_image" in st.session_state:
-    with st.expander("🔍 DEBUG: S3 Upload Tests", expanded=True):
-        st.warning("⚠️ Debug Mode: Test S3 functionality with correct folder structure")
-        
-        s3_test_col1, s3_test_col2 = st.columns(2)
-        
-        with s3_test_col1:
-            st.subheader("Raw Image Upload")
-            test_raw_button = st.button("Test Raw Upload")
-            
-            if test_raw_button and "original_image" in st.session_state:
-                # Test upload to raw-uploads folder
-                raw_s3_path = upload_to_s3(st.session_state["original_image"], 
-                                          st.session_state["user_id"],
-                                          folder="raw-uploads")
-                if raw_s3_path:
-                    st.success(f"✅ Raw upload successful!")
-                    st.code(f"s3://{S3_BUCKET_NAME}/{raw_s3_path}", language="text")
-                    st.session_state["test_raw_s3_path"] = raw_s3_path
-                else:
-                    st.error("❌ Raw upload failed!")
-        
-        with s3_test_col2:
-            st.subheader("Processed Image Upload")
-            test_processed_button = st.button("Test Processed Upload")
-            
-            if test_processed_button and "processed_image" in st.session_state:
-                # Test upload to processed-512x512 folder
-                processed_s3_path = upload_to_s3(st.session_state["processed_image"], 
-                                               st.session_state["user_id"],
-                                               folder="processed-512x512")
-                if processed_s3_path:
-                    st.success(f"✅ Processed upload successful!")
-                    st.code(f"s3://{S3_BUCKET_NAME}/{processed_s3_path}", language="text")
-                    st.session_state["test_processed_s3_path"] = processed_s3_path
-                else:
-                    st.error("❌ Processed upload failed!")
-
-# Add DynamoDB Test Panel for Multiple Food Items
-if debug_mode and "processed_image" in st.session_state:
-    with st.expander("🔍 DEBUG: Complete Food Annotation Test", expanded=True):
-        st.warning("⚠️ Debug Mode: Test Complete Flow with Multiple Food Items")
-        
-        # 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"] = []
-            
-        # 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
-            )
-            
-            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"]:
         st.subheader("📋 Added Food Items")
@@ -668,14 +494,6 @@ if debug_mode and "processed_image" in st.session_state:
     
     # Submit all foods button - outside the form
     if st.button("📤 Submit All Food Items", disabled=len(st.session_state["food_items"]) == 0):
-    
-        # Modify submission to handle test data
-        if debug_mode and test_mode:
-            for food_item in st.session_state["food_items"]:
-                # Prefix the food name to identify test data
-                food_item["food_name"] = f"TEST_{food_item['food_name']}"
-            st.info("ℹ️ Test mode: Data will be marked as test data")
-        
         if not st.session_state["food_items"]:
             st.error("❌ Please add at least one food item before submitting")
         else:
@@ -759,54 +577,4 @@ if st.sidebar.button("Token Rewards System"):
 if st.sidebar.button("Terms of Service"):
     with open("TERMS_OF_SERVICE.md", "r") as f:
         terms = f.read()
-    st.sidebar.markdown(terms)
-
-# Add Cleanup Tools for Debug Mode
-if debug_mode and st.sidebar.checkbox("Show Cleanup Tools"):
-    st.sidebar.markdown("---")
-    st.sidebar.subheader("🧹 Debug Cleanup Tools")
-    
-    with st.sidebar.expander("⚠️ Cleanup Test Data"):
-        st.sidebar.warning("This will delete all test records from DynamoDB and S3")
-        if st.sidebar.button("Delete All Test Records"):
-            try:
-                # Scan for test records in DynamoDB
-                from boto3.dynamodb.conditions import Attr
-                
-                response = metadata_table.scan(
-                    FilterExpression=Attr('food_name').begins_with('TEST_')
-                )
-                
-                total_records = 0
-                deleted_records = 0
-                s3_paths = []
-                
-                # Delete each record and collect S3 paths
-                for item in response.get('Items', []):
-                    total_records += 1
-                    try:
-                        metadata_table.delete_item(Key={'image_id': item['image_id']})
-                        deleted_records += 1
-                        s3_paths.append(item['s3_path'])
-                    except Exception as e:
-                        st.sidebar.error(f"Failed to delete record {item['image_id']}: {e}")
-                
-                # Delete S3 objects
-                for s3_path in set(s3_paths):  # Use set to remove duplicates
-                    try:
-                        s3.delete_object(Bucket=S3_BUCKET_NAME, Key=s3_path)
-                    except Exception as e:
-                        st.sidebar.error(f"Failed to delete S3 object {s3_path}: {e}")
-                
-                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}")
-
-
-# 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()
+    st.sidebar.markdown(terms)