import streamlit as st
import cv2
import numpy as np
import datetime
import os
import time
from camera_input_live import camera_input_live

def save_image(image):
    # Generate a timestamped filename
    timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"captured_image_{timestamp}.png"

    # Convert the image to OpenCV format and save
    bytes_data = image.getvalue()
    cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR)
    cv2.imwrite(filename, cv2_img)

    return filename

def list_png_files():
    return [f for f in os.listdir('.') if f.endswith('.png')]

def main():
    st.title("Streamlit Camera Input Live Demo")
    st.header("Try holding a QR code in front of your webcam")

    # Sidebar for captured images
    st.sidebar.markdown("## Captured Images")
    captured_images = list_png_files()

    # Initialize session state
    if 'last_captured' not in st.session_state:
        st.session_state['last_captured'] = time.time()

    while True:
        image = camera_input_live()

        if image is not None:
            st.image(image)

            # Save image every 5 seconds
            if time.time() - st.session_state['last_captured'] > 5:
                filename = save_image(image)
                captured_images.append(filename)
                st.session_state['last_captured'] = time.time()

            # Update sidebar
            for img_file in captured_images:
                st.sidebar.markdown(f"- [{img_file}]({img_file})")

            # QR Code Detection
            detector = cv2.QRCodeDetector()
            bytes_data = image.getvalue()
            cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR)
            data, bbox, straight_qrcode = detector.detectAndDecode(cv2_img)

            if data:
                st.write("# Found QR code")
                st.write(data)
                with st.expander("Show details"):
                    st.write("BBox:", bbox)
                    st.write("Straight QR code:", straight_qrcode)

        time.sleep(0.1)  # Add a short delay to reduce CPU usage

if __name__ == "__main__":
    main()