Update app.py

app.py

@@ -2,28 +2,56 @@ import streamlit as st
 import pandas as pd
 from datetime import datetime
 import os
-import base64
-from io import BytesIO
+import threading
+import time
 from PIL import Image
+from io import BytesIO
+import base64
+from imutils.video import VideoStream
+import cv2
 
 # 🌟🔥 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
 
-# 📜💾 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!
+# 📜💾 Save to history like a time-traveling scribe!
 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]))
+    # Convert RGB to BGR for OpenCV saving
+    img_bgr = cv2.cvtColor(img_data, cv2.COLOR_RGB2BGR)
+    cv2.imwrite(file_path, img_bgr)
     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(streams):
+    while st.session_state['auto_capture_running']:
+        for i, stream in enumerate(streams):
+            frame = stream.read()
+            if frame is not None:
+                filename = f"cam{i}_auto_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+                save_to_history("🖼️ Image", filename, frame)
+        time.sleep(10)
+
 # 🎛️ Sidebar config like a spaceship control panel!
 with st.sidebar:
     st.header("🎚️📸 Snap Shack")
+    streams = [
+        VideoStream(src=0).start(),  # Camera 0
+        VideoStream(src=1).start()   # Camera 1
+    ]
+    if st.button("⏰ Start Auto-Snap"):
+        st.session_state['auto_capture_running'] = True
+        threading.Thread(target=auto_capture, args=(streams,), daemon=True).start()
+    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"]
@@ -35,116 +63,26 @@ with st.sidebar:
         st.write("🕳️ Empty Stash!")
 
 # 🌍🎨 Main UI kicks off like a cosmic art show!
-st.title("📸 Live Camera Snap Craze")
-
-# 📸📷 JS live camera zone!
-st.header("📸🎥 Live Snap Zone")
-live_camera_html = """
-<div>
-    <select id="cameraSelect"></select>
-    <button id="startBtn" onclick="toggleStream()">📷 Start Live</button>
-    <button onclick="takeSnapshot()">📸 Snap Now</button>
-    <video id="video" autoplay playsinline style="width:100%;"></video>
-    <canvas id="canvas" style="display:none;"></canvas>
-</div>
-<script>
-    let stream = null;
-    const video = document.getElementById('video');
-    const canvas = document.getElementById('canvas');
-    let autoCaptureInterval = null;
-    let isStreaming = false;
-
-    // 📹🔍 Enumerate cameras like a tech detective!
-    async function enumerateCameras() {
-        const devices = await navigator.mediaDevices.enumerateDevices();
-        const videoDevices = devices.filter(device => device.kind === 'videoinput');
-        const select = document.getElementById('cameraSelect');
-        select.innerHTML = '';
-        videoDevices.forEach((device, index) => {
-            const option = document.createElement('option');
-            option.value = device.deviceId;
-            option.text = device.label || `Camera ${index}`;
-            select.appendChild(option);
-        });
-        if (videoDevices.length === 0) {
-            select.innerHTML = '<option>No cameras found</option>';
-        }
-    }
+st.title("📸 Imutils Snap Craze")
 
-    // 📸🎥 Toggle live stream like a broadcast pro!
-    async function toggleStream() {
-        const startBtn = document.getElementById('startBtn');
-        if (isStreaming) {
-            stopStream();
-            startBtn.textContent = '📷 Start Live';
-            isStreaming = false;
-        } else {
-            const cameraId = document.getElementById('cameraSelect').value;
-            if (!cameraId) {
-                alert('No camera selected or available!');
-                return;
-            }
-            const constraints = { video: { deviceId: { exact: cameraId } } };
-            try {
-                stream = await navigator.mediaDevices.getUserMedia(constraints);
-                video.srcObject = stream;
-                startBtn.textContent = '⏹️ Stop Live';
-                isStreaming = true;
-                autoCaptureInterval = setInterval(takeSnapshot, 10000); // Auto-save every 10s
-            } catch (e) {
-                alert('Failed to start camera: ' + e.message);
-            }
-        }
-    }
-
-    // 📸✂️ Snap a pic like a stealthy paparazzi!
-    function takeSnapshot() {
-        if (!stream) return;
-        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 }, '*');
-    }
-
-    // ⏹️📴 Stop the stream like a broadcast kill switch!
-    function stopStream() {
-        if (stream) {
-            stream.getTracks().forEach(track => track.stop());
-            stream = null;
-            video.srcObject = null;
-            clearInterval(autoCaptureInterval);
-        }
-    }
-
-    // 🎬 Kick off the camera party!
-    enumerateCameras();
-</script>
-"""
-st.markdown(live_camera_html, unsafe_allow_html=True)
-
-# 📸📥 Handle snapshots from JS
-def handle_snapshot():
-    if "snapshot" in st.session_state:
-        snapshot_data = st.session_state["snapshot"]
-        filename = f"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="Latest 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});
-        }
-    });
-    </script>
-    """,
-    height=0
-)
-handle_snapshot()
+# 📸📷 Camera snap zone!
+st.header("📸🎥 Snap Zone")
+cols = st.columns(2)
+placeholders = [cols[0].empty(), cols[1].empty()]
+for i, (placeholder, stream) in enumerate(zip(placeholders, streams)):
+    frame = stream.read()
+    if frame is not None:
+        placeholder.image(frame, caption=f"Live Cam {i}", use_container_width=True)
+    else:
+        placeholder.error(f"🚨 No feed from Cam {i}. Check connection or index.")
+    
+    if st.button(f"📸 Snap Cam {i}", key=f"snap{i}"):
+        frame = stream.read()
+        if frame is not None:
+            filename = f"cam{i}_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+            save_to_history("🖼️ Image", filename, frame)
+        else:
+            st.error(f"🚨 Failed to snap Cam {i}")
 
 # 📂 Upload zone like a media drop party!
 st.header("📥🎉 Drop Zone")
@@ -152,8 +90,10 @@ uploaded_files = st.file_uploader("📸 Toss Pics", accept_multiple_files=True,
 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)
+        img = Image.open(uploaded_file)
+        img_array = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)  # Convert for consistency
+        save_to_history("🖼️ Image", file_path, img_array)
+        st.image(img, caption=uploaded_file.name, use_container_width=True)
 
 # 🖼️ Gallery like a media circus!
 st.header("🎪 Snap Show")
@@ -180,4 +120,12 @@ if st.session_state['file_history']:
     df = pd.DataFrame(st.session_state['file_history'])
     st.dataframe(df)
 else:
-    st.write("🕳️ Nothing snapped yet!")
+    st.write("🕳️ Nothing snapped yet!")
+
+# Cleanup on app exit
+def cleanup():
+    for stream in streams:
+        stream.stop()
+
+import atexit
+atexit.register(cleanup)