import streamlit as st
import pandas as pd
from datetime import datetime
import os
import threading
import time
from PIL import Image
from io import BytesIO

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

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

# 📸⏰ Auto-capture every 10 secs like a sneaky shutterbug!
def auto_capture():
    if st.session_state['auto_capture_running'] and st.session_state['selected_cam_img']:
        cam_img = st.session_state['selected_cam_img']
        filename = f"auto_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
        with open(filename, "wb") as f:
            f.write(cam_img.getvalue())
        save_to_history("🖼️ Image", filename)
        threading.Timer(10, auto_capture).start()

# 🎛️ Sidebar config like a spaceship control panel!
with st.sidebar:
    st.header("🎚️📸 Snap Shack")
    camera_choice = st.selectbox("📷 Pick a Cam", ["Camera 0", "Camera 1"])
    if st.button("⏰ Start Auto-Snap"):
        st.session_state['auto_capture_running'] = True
        auto_capture()
    if st.button("⏹️ Stop Auto-Snap"):
        st.session_state['auto_capture_running'] = False

    # 📂 Sidebar file outline with emoji flair!
    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("📸 Streamlit Snap Craze")

# 📸📷 Camera snap zone!
st.header("📸🎥 Snap Zone")
cols = st.columns(2)
with cols[0]:
    cam0_img = st.camera_input("📷 Camera 0", key="cam0")
    if cam0_img and camera_choice == "Camera 0":
        filename = f"cam0_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
        if st.session_state['cam0_file'] and os.path.exists(st.session_state['cam0_file']):
            os.remove(st.session_state['cam0_file'])
        with open(filename, "wb") as f:
            f.write(cam0_img.getvalue())
        st.session_state['cam0_file'] = filename
        st.session_state['selected_cam_img'] = cam0_img
        save_to_history("🖼️ Image", filename)
        st.image(Image.open(filename), caption="Camera 0 Snap", use_column_width=True)
with cols[1]:
    cam1_img = st.camera_input("📷 Camera 1", key="cam1")
    if cam1_img and camera_choice == "Camera 1":
        filename = f"cam1_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
        if st.session_state['cam1_file'] and os.path.exists(st.session_state['cam1_file']):
            os.remove(st.session_state['cam1_file'])
        with open(filename, "wb") as f:
            f.write(cam1_img.getvalue())
        st.session_state['cam1_file'] = filename
        st.session_state['selected_cam_img'] = cam1_img
        save_to_history("🖼️ Image", filename)
        st.image(Image.open(filename), caption="Camera 1 Snap", use_column_width=True)

# 📂 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}"
        with open(file_path, "wb") as f:
            f.write(uploaded_file.read())
        save_to_history("🖼️ Image", file_path)

# 🖼️ 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]:
                st.image(img['Path'], caption=img['Path'], use_column_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.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!")