import cv2
import numpy as np
import streamlit as st
from Code import apply_effects

# Load cascade classifiers and images
glassesCasc = cv2.CascadeClassifier('Train/third-party/frontalEyes35x16.xml')
noseCasc = cv2.CascadeClassifier('Train/third-party/Nose18x15.xml')
glasses = cv2.imread('Train/glasses.png', cv2.IMREAD_UNCHANGED)
mustache = cv2.imread('Train/mustache.png', cv2.IMREAD_UNCHANGED)

def main():
    st.title("Snapchat Filter App")
    st.write("Upload an image or use your webcam to apply face effects!")

    option = st.selectbox("Choose an option", ("Upload Image", "Use Webcam"))

    if option == "Upload Image":
        uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
        if uploaded_image is not None:
            image = cv2.imdecode(np.fromstring(uploaded_image.read(), np.uint8), 1)
            image_with_effects = apply_effects(image)
            st.image(image_with_effects, channels="BGR", use_column_width=True)

    else:  # Use Webcam
        cap = cv2.VideoCapture(0)
        st.write("Webcam is active.")
        frame_placeholder = st.empty()

        while cap.isOpened():
            ret, frame = cap.read()
            if not ret:
                break
            image_with_effects = apply_effects(frame)
            frame_placeholder.image(image_with_effects, channels="BGR", use_column_width=True)

if __name__ == "__main__":
    main()