import streamlit as st
import pandas as pd
from datetime import datetime
import os
import base64

# 🌟🔥 Initialize session state like a galactic DJ spinning tracks!
if 'file_history' not in st.session_state:
    st.session_state['file_history'] = []

# 📜💾 Save to history like a time-traveling scribe! | 📅✨ save_to_history("🖼️ Image", "pic.jpg", img_data) - Stamps a pic in the history books like a boss!
def save_to_history(file_type, file_path, img_data):
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    with open(file_path, "wb") as f:
        f.write(base64.b64decode(img_data.split(',')[1]))
    st.session_state['file_history'].append({
        "Timestamp": timestamp,
        "Type": file_type,
        "Path": file_path
    })

# 🎛️ Sidebar config like a spaceship control panel!
with st.sidebar:
    st.header("🎚️📸 Snap Shack")
    st.subheader("📜 Snap Stash")
    if st.session_state['file_history']:
        images = [f for f in st.session_state['file_history'] if f['Type'] == "🖼️ Image"]
        if images:
            st.write("🖼️ Images")
            for img in images:
                st.write(f"- {img['Path']} @ {img['Timestamp']}")
    else:
        st.write("🕳️ Empty Stash!")

# 🌍🎨 Main UI kicks off like a cosmic art show!
st.title("📸 Multi-Cam Snap Craze")

# 📸📷 JS camera zone with multiple streams!
st.header("📸🎥 Snap Zone")
multi_cam_html = """
<div id="videoContainer"></div>
<script>
    let streams = [];
    const container = document.getElementById('videoContainer');
    let autoCaptureIntervals = [];

    // 📹🔍 Enumerate cameras like a tech detective!
    async function enumerateAndStartCameras() {
        const devices = await navigator.mediaDevices.enumerateDevices();
        const videoDevices = devices.filter(device => device.kind === 'videoinput');
        videoDevices.forEach((device, index) => {
            const div = document.createElement('div');
            div.innerHTML = `
                <p>${device.label || 'Camera ' + index}</p>
                <video id="video${index}" autoplay playsinline style="width:100%;"></video>
                <canvas id="canvas${index}" style="display:none;"></canvas>
                <button onclick="takeSnapshot(${index})">📸 Snap Cam ${index}</button>
            `;
            container.appendChild(div);
            startStream(device.deviceId, index);
        });
    }

    // 📸🎥 Start streaming like a live broadcast pro!
    async function startStream(deviceId, index) {
        const constraints = { video: { deviceId: { exact: deviceId } } };
        const stream = await navigator.mediaDevices.getUserMedia(constraints);
        const video = document.getElementById(`video${index}`);
        video.srcObject = stream;
        streams.push(stream);
        // Auto-capture every 10 seconds
        autoCaptureIntervals[index] = setInterval(() => takeSnapshot(index), 10000);
    }

    // 📸✂️ Snap a pic like a stealthy paparazzi!
    function takeSnapshot(index) {
        const video = document.getElementById(`video${index}`);
        const canvas = document.getElementById(`canvas${index}`);
        canvas.width = video.videoWidth;
        canvas.height = video.videoHeight;
        canvas.getContext('2d').drawImage(video, 0, 0);
        const dataUrl = canvas.toDataURL('image/jpeg');
        window.parent.postMessage({ type: 'snapshot', data: dataUrl, camIndex: index }, '*');
    }

    // ⏹️📴 Stop all streams like a broadcast kill switch!
    function stopStreams() {
        streams.forEach(stream => stream.getTracks().forEach(track => track.stop()));
        streams = [];
        autoCaptureIntervals.forEach(interval => clearInterval(interval));
        autoCaptureIntervals = [];
    }

    // 🎬 Kick off the camera party!
    enumerateAndStartCameras();
    window.onunload = stopStreams;
</script>
"""
st.markdown(multi_cam_html, unsafe_allow_html=True)

# 📸📥 Handle snapshots from JS
def handle_snapshot():
    if "snapshot" in st.session_state:
        snapshot_data = st.session_state["snapshot"]
        cam_index = st.session_state.get("camIndex", 0)
        filename = f"cam{cam_index}_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
        save_to_history("🖼️ Image", filename, snapshot_data)
        st.image(Image.open(BytesIO(base64.b64decode(snapshot_data.split(',')[1]))), caption=f"Cam {cam_index} Snap", use_container_width=True)
        del st.session_state["snapshot"]

st.components.v1.html(
    """
    <script>
    window.addEventListener('message', function(event) {
        if (event.data.type === 'snapshot') {
            window.streamlitAPI.setComponentValue({snapshot: event.data.data, camIndex: event.data.camIndex});
        }
    });
    </script>
    """,
    height=0
)
handle_snapshot()

# 📂 Upload zone like a media drop party!
st.header("📥🎉 Drop Zone")
uploaded_files = st.file_uploader("📸 Toss Pics", accept_multiple_files=True, type=['jpg', 'png'])
if uploaded_files:
    for uploaded_file in uploaded_files:
        file_path = f"uploaded_{uploaded_file.name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
        save_to_history("🖼️ Image", file_path, base64.b64encode(uploaded_file.getvalue()).decode())
        st.image(Image.open(uploaded_file), caption=uploaded_file.name, use_container_width=True)

# 🖼️ Gallery like a media circus!
st.header("🎪 Snap Show")
if st.session_state['file_history']:
    images = [f for f in st.session_state['file_history'] if f['Type'] == "🖼️ Image"]
    if images:
        st.subheader("🖼️ Pic Parade")
        cols = st.columns(3)
        for i, img in enumerate(images):
            with cols[i % 3]:
                if os.path.exists(img['Path']):
                    st.image(img['Path'], caption=img['Path'], use_container_width=True)
                    with open(img['Path'], "rb") as f:
                        img_data = f.read()
                    st.markdown(f'<a href="data:image/jpeg;base64,{base64.b64encode(img_data).decode()}" download="{os.path.basename(img["Path"])}">📥 Snag It!</a>', unsafe_allow_html=True)
                else:
                    st.warning(f"🚨 Missing file: {img['Path']}")
else:
    st.write("🚫 No snaps yet!")

# 📜 History log like a time machine!
st.header("⏳ Snap Saga")
if st.session_state['file_history']:
    df = pd.DataFrame(st.session_state['file_history'])
    st.dataframe(df)
else:
    st.write("🕳️ Nothing snapped yet!")