import streamlit as st

# Initialize session state
if "page" not in st.session_state:
    st.session_state.page = "home"


def main():
    st.set_page_config(page_title="Fingerlings", layout="centered")
    
    # Home Page
    if st.session_state.page == "home":
        st.title("🖐️ Welcome to Fingerlings!")
        st.write("Choose an ASL learning mode:")
        mode = st.selectbox("Learning Mode", ["Finger Spelling", "Full-Hand Gestures"])
        
        if st.button("Start Lesson"):
            st.session_state.mode = mode
            st.session_state.page = "lesson"
            st.rerun()
    
    # Guided Lessons
    if st.session_state.page == "lesson":
        st.title("📖 Guided Lessons")
        st.write(f"Learning: {st.session_state.mode}")
        st.image("example_sign.jpg", caption="Example Sign")
        if st.button("Go to Recognition & Feedback"):
            st.session_state.page = "recognition"
            st.rerun()
    
    # Recognition & Feedback
    if st.session_state.page == "recognition":
        st.title("📷 ASL Recognition & Feedback")
        st.write("Show your hand sign in front of the camera.")
        st.camera_input("Take a Picture")
        st.write("Feedback: ✅ Correct / ❌ Try Again")
        if st.button("View Progress"):
            st.session_state.page = "progress"
            st.rerun()
    
    # Progress Tracking
    if st.session_state.page == "progress":
        st.title("📊 Progress Tracking")
        st.write("Your learning progress:")
        st.progress(0.6)  # Example progress at 60%
        st.button("Back to Home", on_click=lambda: st.session_state.update(page="home"))


if __name__ == "__main__":
    main()