Update app.py

app.py

@@ -4,168 +4,96 @@ import os
 import io
 import base64
 import boto3
-import time
 from PIL import Image
-import firebase_admin
-from firebase_admin import credentials, auth, firestore
+from streamlit_tags import st_tags
 
-# 🚀 Load Secrets from Hugging Face Secrets
-
-# Load AWS credentials from secrets
-AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY_ID")
-AWS_SECRET_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
+# Load AWS credentials using correct HF Secrets
+AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY")
+AWS_SECRET_KEY = os.getenv("AWS_SECRET_KEY")
 AWS_REGION = os.getenv("AWS_REGION", "us-east-1")
 S3_BUCKET_NAME = os.getenv("S3_BUCKET_NAME", "food-image-crowdsourcing")
-DYNAMODB_TABLE = os.getenv("DYNAMODB_TABLE", "image_metadata")  # ✅ Corrected Table Name
+DYNAMODB_TABLE = os.getenv("DYNAMODB_TABLE", "image_metadata")
 
 # Load Firebase credentials
-FIREBASE_CONFIG = os.getenv("FIREBASE_CONFIG")
-
-# ✅ Initialize Firebase Admin SDK
-if FIREBASE_CONFIG:
-    try:
-        firebase_config_dict = json.loads(FIREBASE_CONFIG)
-        if not firebase_admin._apps:
-            cred = credentials.Certificate(firebase_config_dict)
-            firebase_admin.initialize_app(cred)
-    except Exception as e:
-        st.error(f"⚠️ Firebase initialization error: {e}")
-else:
-    st.error("⚠️ Firebase configuration is missing! Please check HF Secrets.")
-
-# ✅ Initialize AWS Services (S3 & DynamoDB)
-try:
-    s3 = boto3.client(
-        "s3",
-        aws_access_key_id=AWS_ACCESS_KEY,
-        aws_secret_access_key=AWS_SECRET_KEY,
-        region_name=AWS_REGION
-    )
-
-    dynamodb = boto3.resource(
-        "dynamodb",
-        region_name=AWS_REGION, 
-        aws_access_key_id=AWS_ACCESS_KEY,
-        aws_secret_access_key=AWS_SECRET_KEY,
-    )
-    metadata_table = dynamodb.Table(DYNAMODB_TABLE)  # ✅ Corrected Table Name
-except Exception as e:
-    st.error(f"⚠️ AWS initialization error: {e}")
-
-# 🔹 Initialize Session State for Tokens
-if "tokens" not in st.session_state:
-    st.session_state.tokens = 0
-if "user_email" not in st.session_state:
-    st.session_state.user_email = None
-
-# 🔹 UI - Page Title
-st.title("📷 Food Crowdsourcing Research App")
-st.write("Help our research by uploading food images! Earn tokens for future app use.")
-
-# ✅ **User Authentication Section**
-st.sidebar.header("🔑 User Authentication")
-
-auth_choice = st.sidebar.radio("Select an option", ["Login", "Sign Up", "Logout"])
-
-if auth_choice == "Sign Up":
-    with st.sidebar.form("signup_form"):
-        email = st.text_input("Email")
-        password = st.text_input("Password", type="password")
-        submit_button = st.form_submit_button("Sign Up")
-
-    if submit_button:
-        try:
-            user = auth.create_user(email=email, password=password)
-            st.success(f"✅ Account created! Now login with {email}.")
-        except Exception as e:
-            st.error(f"⚠️ {str(e)}")
-
-if auth_choice == "Login":
-    with st.sidebar.form("login_form"):
-        email = st.text_input("Email")
-        password = st.text_input("Password", type="password")
-        submit_button = st.form_submit_button("Login")
-
-    if submit_button:
-        try:
-            user = auth.get_user_by_email(email)
-            st.session_state.user_email = email
-            st.success(f"✅ Welcome back, {email}!")
-        except Exception as e:
-            st.error(f"⚠️ {str(e)}")
-
-if auth_choice == "Logout":
-    if st.session_state.user_email:
-        st.session_state.user_email = None
-        st.success("✅ Logged out successfully!")
+FIREBASE_CONFIG = json.loads(os.getenv("FIREBASE_CONFIG", "{}"))
 
-# 🔹 **Ensure User is Logged In**
-if not st.session_state.user_email:
-    st.warning("⚠️ Please log in to upload images.")
-    st.stop()
-
-# ✅ **Image Upload Section**
-uploaded_file = st.file_uploader("📸 Upload a food image", type=["jpg", "png", "jpeg"])
-
-# ✅ **Terms & Conditions**
-st.markdown("### **Terms & Conditions**")
-st.write(
-    "By uploading an image, you agree to transfer full copyright to the research team for AI training purposes. "
-    "You are responsible for ensuring you own the image and it does not violate any copyright laws. "
-    "We do not guarantee when tokens will be redeemable. Keep track of your user ID."
+# Initialize AWS Services (S3 & DynamoDB)
+s3 = boto3.client(
+    "s3",
+    aws_access_key_id=AWS_ACCESS_KEY,
+    aws_secret_access_key=AWS_SECRET_KEY,
 )
-st.checkbox("I agree to the terms and conditions", key="terms_accepted")
-
-# ✅ **Process and Upload Image**
-if uploaded_file and st.session_state.terms_accepted:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-
-    # Show Original Size
-    st.write(f"📏 Original size: {image.size}")
-
-    # Convert PNG to JPEG if needed
-    if image.format == "PNG":
-        image = image.convert("RGB")
-
-    # Resize image to max 1024x1024 px
-    MAX_SIZE = (1024, 1024)
-    image.thumbnail(MAX_SIZE)
 
-    # Convert Image to Bytes
-    img_bytes = io.BytesIO()
-    image.save(img_bytes, format="JPEG", quality=80)  # Reduce quality to save space
-    img_bytes.seek(0)
-
-    # 🔹 Generate Unique File Name
-    file_name = f"raw-uploads/{st.session_state.user_email}_{int(time.time())}.jpg"
+dynamodb = boto3.resource(
+    "dynamodb",
+    region_name=AWS_REGION, 
+    aws_access_key_id=AWS_ACCESS_KEY,
+    aws_secret_access_key=AWS_SECRET_KEY,
+)
+metadata_table = dynamodb.Table(DYNAMODB_TABLE)
 
-    # ✅ **Upload to S3**
-    try:
-        s3.put_object(
-            Bucket=S3_BUCKET_NAME,
-            Key=file_name,
-            Body=img_bytes.getvalue(),
-            ContentType="image/jpeg",
-        )
-        st.success("✅ Image uploaded successfully to S3!")
+# Streamlit Layout - Three Panel Design
+st.title("🍽️ Food Image Review & Annotation")
+col1, col2, col3 = st.columns([1, 1, 1])  # Equal size columns
 
-        # ✅ **Store Metadata in DynamoDB**
-        metadata_table.put_item(
-            Item={
-                "user_email": st.session_state.user_email,
-                "file_name": file_name,
-                "upload_time": int(time.time()),
-            }
+# Upload Image
+uploaded_file = st.file_uploader("Upload an image of your food", type=["jpg", "png", "jpeg"])
+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):
+    aspect_ratio = image.width / image.height
+    new_width = 512
+    new_height = int(new_width / aspect_ratio)
+    resized_image = image.resize((new_width, new_height))
+    return resized_image
+
+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="Uploaded", use_container_width=True)
+
+    # 🖼️ Panel 2: Resized Image (512x512 Maintaining Aspect Ratio)
+    with col2:
+        st.subheader("🖼️ Processed Image")
+        st.image(processed_img, caption="512x512 Maintained Aspect", use_container_width=True)
+
+    # ✏️ Panel 3: Food Annotations with Intellisense
+    with col3:
+        st.subheader("📝 Add Annotations")
+        
+        # Expanded food suggestions
+        suggested_foods = [
+            "Pizza", "Pasta", "Tacos", "Sushi", "Ramen", "Kimchi", "Bratwurst", "Jambalaya", 
+            "Chicken", "Rice", "Steak", "Bread", "Cheese", "Salmon", "Avocado", "Eggs", 
+            "Carrots", "Tomatoes", "Cucumber", "Yogurt", "Peanuts", "Lettuce", "Pierogies", "Mongolian Beef", "Spaghetti Bolognese",
+            "Bigos", "Stuffed Cabbage", "Zurek", "Schnitzel", "Tomato Soup", "Potato Pancakes", "Blintzes", "Broccoli Chicken"
+        ]
+
+        # User selects food items from suggestions
+        food_items = st_tags(
+            label="Enter food items",
+            text="Add items...",
+            value=[],
+            suggestions=suggested_foods,
+            maxtags=10,
         )
 
-        # 🎉 **Reward Tokens**
-        st.session_state.tokens += 1
-        st.write("🎉 You earned **1 token** for uploading!")
-        st.success(f"🔢 Total tokens: {st.session_state.tokens}")
-
-    except Exception as e:
-        st.error(f"⚠️ Failed to upload image: {str(e)}")
-
-# ✅ **Future: Annotation for Additional Token**
-# If annotation is provided, reward another token
+        if st.button("Save Annotations"):
+            metadata_table.update_item(
+                Key={"image_id": uploaded_file.name},
+                UpdateExpression="SET annotations = :a, processing_status = :p, s3_url = :s, tokens_earned = :t",
+                ExpressionAttributeValues={
+                    ":a": food_items,
+                    ":p": "processed",
+                    ":s": f"s3://{S3_BUCKET_NAME}/raw-uploads/{uploaded_file.name}",
+                    ":t": 1 if food_items else 0
+                },
+            )
+            st.success("✅ Annotations saved successfully!")