import streamlit as st
from PIL import Image

st.title("📸 Auto Weight Logger with Retake Option")

# Session state to track if image is captured
if 'captured' not in st.session_state:
    st.session_state.captured = False
if 'image' not in st.session_state:
    st.session_state.image = None

# Function to reset camera
def retake():
    st.session_state.captured = False
    st.session_state.image = None

# Only show camera if image is not captured
if not st.session_state.captured:
    image = st.camera_input("Take a picture")
    if image:
        st.session_state.image = image
        st.session_state.captured = True

# If image is captured, show options
if st.session_state.captured and st.session_state.image:
    st.image(st.session_state.image, caption="Captured Image", use_column_width=True)
    st.button("Retake", on_click=retake)
    st.success("✅ Image captured successfully.")
    # You can add OCR or Salesforce upload button here