Update app.py

app.py

@@ -4,15 +4,20 @@ from datetime import datetime
 import os
 import threading
 import time
-import base64
+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 'snapshot_data' not in st.session_state:
-    st.session_state['snapshot_data'] = None
+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):
@@ -25,17 +30,18 @@ def save_to_history(file_type, 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['snapshot_data']:
-        snapshot_data = st.session_state['snapshot_data']
+    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(base64.b64decode(snapshot_data.split(',')[1]))
+            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()
@@ -54,119 +60,45 @@ with st.sidebar:
         st.write("🕳️ Empty Stash!")
 
 # 🌍🎨 Main UI kicks off like a cosmic art show!
-st.title("📸 WebRTC Snap Craze")
+st.title("📸 Streamlit Snap Craze")
 
-# 📸📷 WebRTC camera snap zone!
+# 📸📷 Camera snap zone!
 st.header("📸🎥 Snap Zone")
-webrtc_html = """
-<div>
-    <select id="cameraSelect"></select>
-    <button onclick="startStream()">📷 Start Stream</button>
-    <button onclick="takeSnapshot()">📸 Snap It!</button>
-    <button onclick="stopStream()">⏹️ Stop Stream</button>
-    <video id="video" autoplay playsinline></video>
-    <canvas id="canvas" style="display:none;"></canvas>
-    <img id="snapshot" style="display:none;">
-</div>
-<script>
-    let stream = null;
-    let peerConnection = null;
-    const config = { iceServers: [{ urls: 'stun:stun.l.google.com:19302' }] };
-
-    // 📹🔍 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 + 1}`;
-            select.appendChild(option);
-        });
-    }
-
-    // 📸🎥 Start streaming like a live broadcast pro!
-    async function startStream() {
-        if (stream) stopStream();
-        const cameraId = document.getElementById('cameraSelect').value;
-        const constraints = { video: { deviceId: cameraId ? { exact: cameraId } : undefined } };
-        stream = await navigator.mediaDevices.getUserMedia(constraints);
-        const video = document.getElementById('video');
-        video.srcObject = stream;
-
-        peerConnection = new RTCPeerConnection(config);
-        stream.getTracks().forEach(track => peerConnection.addTrack(track, stream));
-        peerConnection.onicecandidate = event => {
-            if (event.candidate) console.log('ICE Candidate:', event.candidate);
-        };
-        const offer = await peerConnection.createOffer();
-        await peerConnection.setLocalDescription(offer);
-        // Simulate remote answer (self-loop for demo)
-        await peerConnection.setRemoteDescription(offer);
-    }
-
-    // 📸✂️ Snap a pic like a stealthy paparazzi!
-    function takeSnapshot() {
-        const video = document.getElementById('video');
-        const canvas = document.getElementById('canvas');
-        const snapshot = document.getElementById('snapshot');
-        canvas.width = video.videoWidth;
-        canvas.height = video.videoHeight;
-        canvas.getContext('2d').drawImage(video, 0, 0);
-        const dataUrl = canvas.toDataURL('image/jpeg');
-        snapshot.src = dataUrl;
-        snapshot.style.display = 'block';
-        // Send snapshot to Python via session state
-        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;
-        }
-        if (peerConnection) {
-            peerConnection.close();
-            peerConnection = null;
-        }
-        document.getElementById('video').srcObject = null;
-        document.getElementById('snapshot').style.display = 'none';
-    }
-
-    // 🎬 Kick off the camera party!
-    enumerateCameras();
-</script>
-"""
-st.markdown(webrtc_html, unsafe_allow_html=True)
-
-# 📸📥 Handle snapshots from JS
-def handle_snapshot():
-    if "snapshot" in st.session_state:
-        st.session_state['snapshot_data'] = st.session_state["snapshot"]
-        filename = f"snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+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(base64.b64decode(st.session_state['snapshot_data'].split(',')[1]))
+            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(filename, caption="Latest Snap", use_column_width=True)
-        del st.session_state["snapshot"]  # Clear after saving
-
-# Listen for JS messages
-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()
+        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")
@@ -178,7 +110,9 @@ if st.session_state['file_history']:
         for i, img in enumerate(images):
             with cols[i % 3]:
                 st.image(img['Path'], caption=img['Path'], use_column_width=True)
-                st.markdown(f'<a href="data:image/jpeg;base64,{base64.b64encode(open(img["Path"], "rb").read()).decode()}" download="{os.path.basename(img["Path"])}">📥 Snag It!</a>', unsafe_allow_html=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!")