Update app.py

app.py

@@ -6,6 +6,8 @@ import base64
 import boto3
 from PIL import Image
 from streamlit_tags import st_tags
+import firebase_admin
+from firebase_admin import credentials, auth
 
 # Load AWS credentials using correct HF Secrets
 AWS_ACCESS_KEY = os.getenv("AWS_ACCESS_KEY")
@@ -17,6 +19,11 @@ DYNAMODB_TABLE = os.getenv("DYNAMODB_TABLE", "image_metadata")
 # Load Firebase credentials
 FIREBASE_CONFIG = json.loads(os.getenv("FIREBASE_CONFIG", "{}"))
 
+# Initialize Firebase Admin SDK (Prevent multiple initialization)
+if not firebase_admin._apps:
+    cred = credentials.Certificate(FIREBASE_CONFIG)
+    firebase_admin.initialize_app(cred)
+
 # Initialize AWS Services (S3 & DynamoDB)
 s3 = boto3.client(
     "s3",
@@ -25,16 +32,50 @@ s3 = boto3.client(
 )
 
 dynamodb = boto3.resource(
-    "dynamodb",
+    "dynamodb", 
     region_name=AWS_REGION, 
-    aws_access_key_id=AWS_ACCESS_KEY,
+    aws_access_key_id=AWS_ACCESS_KEY, 
     aws_secret_access_key=AWS_SECRET_KEY,
 )
 metadata_table = dynamodb.Table(DYNAMODB_TABLE)
 
+# Streamlit Layout - Authentication Section
+st.sidebar.title("🔑 User Authentication")
+auth_option = st.sidebar.radio("Select an option", ["Login", "Sign Up", "Logout"])
+
+if auth_option == "Sign Up":
+    email = st.sidebar.text_input("Email")
+    password = st.sidebar.text_input("Password", type="password")
+    if st.sidebar.button("Sign Up"):
+        try:
+            user = auth.create_user(email=email, password=password)
+            st.sidebar.success("✅ User created successfully! Please log in.")
+        except Exception as e:
+            st.sidebar.error(f"Error: {e}")
+
+if auth_option == "Login":
+    email = st.sidebar.text_input("Email")
+    password = st.sidebar.text_input("Password", type="password")
+    if st.sidebar.button("Login"):
+        try:
+            user = auth.get_user_by_email(email)
+            st.session_state["user_id"] = user.uid
+            st.sidebar.success("✅ Logged in successfully!")
+        except Exception as e:
+            st.sidebar.error(f"Login failed: {e}")
+
+if auth_option == "Logout" and "user_id" in st.session_state:
+    del st.session_state["user_id"]
+    st.sidebar.success("✅ Logged out successfully!")
+
+# Ensure user is logged in before uploading
+if "user_id" not in st.session_state:
+    st.warning("⚠️ Please log in to upload images.")
+    st.stop()
+
 # Streamlit Layout - Three Panel Design
 st.title("🍽️ Food Image Review & Annotation")
-col1, col2, col3 = st.columns([1, 1, 1])  # Equal size columns
+col1, col2, col3 = st.columns([1, 1, 1])
 
 # Upload Image
 uploaded_file = st.file_uploader("Upload an image of your food", type=["jpg", "png", "jpeg"])
@@ -68,32 +109,4 @@ if "original_image" in st.session_state:
     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,
-        )
-
-        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!")
+        # Expanded food suggestions