Update app.py

app.py

@@ -8,19 +8,19 @@ from PIL import Image
 from io import BytesIO
 import base64
 
+# 🌟🔥 Initialize global state for threading safety!
+AUTO_CAPTURE_RUNNING = False
+SELECTED_CAM_IMG = None
+
 # 🌟🔥 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!
+# 📜💾 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:
@@ -33,10 +33,16 @@ def save_to_history(file_type, file_path, img_data):
 
 # 📸⏰ 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']
+    global AUTO_CAPTURE_RUNNING, SELECTED_CAM_IMG
+    if AUTO_CAPTURE_RUNNING and SELECTED_CAM_IMG:
         filename = f"auto_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-        save_to_history("🖼️ Image", filename, cam_img.getvalue())
+        with open(filename, "wb") as f:
+            f.write(SELECTED_CAM_IMG)
+        st.session_state['file_history'].append({
+            "Timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "Type": "🖼️ Image",
+            "Path": filename
+        })
         threading.Timer(10, auto_capture).start()
 
 # 🎛️ Sidebar config like a spaceship control panel!
@@ -44,10 +50,12 @@ 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
+        global AUTO_CAPTURE_RUNNING
+        AUTO_CAPTURE_RUNNING = True
         auto_capture()
     if st.button("⏹️ Stop Auto-Snap"):
-        st.session_state['auto_capture_running'] = False
+        global AUTO_CAPTURE_RUNNING
+        AUTO_CAPTURE_RUNNING = False
 
     # 📂 Sidebar file outline with emoji flair!
     st.subheader("📜 Snap Stash")
@@ -74,7 +82,8 @@ with cols[0]:
             os.remove(st.session_state['cam0_file'])
         save_to_history("🖼️ Image", filename, cam0_img.getvalue())
         st.session_state['cam0_file'] = filename
-        st.session_state['selected_cam_img'] = cam0_img
+        global SELECTED_CAM_IMG
+        SELECTED_CAM_IMG = cam0_img.getvalue()
         st.image(Image.open(BytesIO(cam0_img.getvalue())), caption="Camera 0 Snap", use_container_width=True)
 with cols[1]:
     cam1_img = st.camera_input("📷 Camera 1", key="cam1")
@@ -84,7 +93,8 @@ with cols[1]:
             os.remove(st.session_state['cam1_file'])
         save_to_history("🖼️ Image", filename, cam1_img.getvalue())
         st.session_state['cam1_file'] = filename
-        st.session_state['selected_cam_img'] = cam1_img
+        global SELECTED_CAM_IMG
+        SELECTED_CAM_IMG = cam1_img.getvalue()
         st.image(Image.open(BytesIO(cam1_img.getvalue())), caption="Camera 1 Snap", use_container_width=True)
 
 # 📂 Upload zone like a media drop party!
@@ -92,8 +102,9 @@ 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}"
+        file_path = f"uploaded_{uploaded_file.name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
         save_to_history("🖼️ Image", file_path, uploaded_file.getvalue())
+        st.image(Image.open(BytesIO(uploaded_file.getvalue())), caption=uploaded_file.name, use_container_width=True)
 
 # 🖼️ Gallery like a media circus!
 st.header("🎪 Snap Show")