Update app.py

app.py

@@ -6,6 +6,7 @@ import threading
 import time
 from PIL import Image
 from io import BytesIO
+import base64
 
 # 🌟🔥 Initialize session state like a galactic DJ spinning tracks!
 if 'file_history' not in st.session_state:
@@ -20,8 +21,10 @@ 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):
+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(img_data)
     st.session_state['file_history'].append({
         "Timestamp": timestamp,
         "Type": file_type,
@@ -33,9 +36,7 @@ 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)
+        save_to_history("🖼️ Image", filename, cam_img.getvalue())
         threading.Timer(10, auto_capture).start()
 
 # 🎛️ Sidebar config like a spaceship control panel!
@@ -71,24 +72,20 @@ with cols[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())
+        save_to_history("🖼️ Image", filename, 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)
+        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")
     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())
+        save_to_history("🖼️ Image", filename, 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)
+        st.image(Image.open(BytesIO(cam1_img.getvalue())), caption="Camera 1 Snap", use_container_width=True)
 
 # 📂 Upload zone like a media drop party!
 st.header("📥🎉 Drop Zone")
@@ -96,9 +93,7 @@ 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}"
-        with open(file_path, "wb") as f:
-            f.write(uploaded_file.read())
-        save_to_history("🖼️ Image", file_path)
+        save_to_history("🖼️ Image", file_path, uploaded_file.getvalue())
 
 # 🖼️ Gallery like a media circus!
 st.header("🎪 Snap Show")
@@ -109,10 +104,13 @@ if st.session_state['file_history']:
         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)
+                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!")