Update app.py

app.py

@@ -19,44 +19,44 @@ def get_image_base64(image_path):
     with open(image_path, "rb") as image_file:
         return base64.b64encode(image_file.read()).decode()
 
-def list_png_files():
-    return [f for f in os.listdir('.') if f.endswith('.png')]
-
 def main():
     st.title("Streamlit Camera Input Live Demo")
     st.header("Try holding a QR code in front of your webcam")
 
-    if 'captured_images' not in st.session_state:
-        st.session_state['captured_images'] = list_png_files()
+    # Slider for snapshot interval
+    snapshot_interval = st.sidebar.slider("Snapshot Interval (seconds)", 1, 10, 5)
+
+    # Display container for live image
+    image_placeholder = st.empty()
 
+    # Initialize session state for captured images and last captured time
+    if 'captured_images' not in st.session_state:
+        st.session_state['captured_images'] = []
     if 'last_captured' not in st.session_state:
         st.session_state['last_captured'] = time.time()
-        
-
-    while True:
-        try:
-            image = camera_input_live()
-            
-            if image is not None:
-                st.image(image)
-    
-                if time.time() - st.session_state['last_captured'] > 5:
-                    filename = save_image(image)
-                    st.session_state['captured_images'].append(filename)
-                    st.session_state['last_captured'] = time.time()
-    
-                sidebar_html = "<div style='display:flex;flex-direction:column;'>"
-                for img_file in st.session_state['captured_images']:
-                    image_base64 = get_image_base64(img_file)
-                    sidebar_html += f"<img src='data:image/png;base64,{image_base64}' style='width:100px;'><br>"
-                sidebar_html += "</div>"
-                st.sidebar.markdown("## Captured Images")
-                st.sidebar.markdown(sidebar_html, unsafe_allow_html=True)
-    
-                time.sleep(0.1)  # Add a short delay to reduce CPU usage
-
-        except:
-            st.write('.')
+
+    # Capture and display live image
+    image = camera_input_live()
+    if image is not None:
+        image_placeholder.image(image)
+
+        # Save image based on interval
+        if time.time() - st.session_state['last_captured'] > snapshot_interval:
+            filename = save_image(image)
+            st.session_state['captured_images'].append(filename)
+            st.session_state['last_captured'] = time.time()
+
+    # Update sidebar with captured images
+    sidebar_html = "<div style='display:flex;flex-direction:column;'>"
+    for img_file in st.session_state['captured_images']:
+        image_base64 = get_image_base64(img_file)
+        sidebar_html += f"<img src='data:image/png;base64,{image_base64}' style='width:100px;'><br>"
+    sidebar_html += "</div>"
+    st.sidebar.markdown("## Captured Images")
+    st.sidebar.markdown(sidebar_html, unsafe_allow_html=True)
+
+    # Trigger a rerun of the app to update the live feed
+    st.experimental_rerun()
 
 if __name__ == "__main__":
     main()