Spaces:
Runtime error
Runtime error
import streamlit as st | |
import cv2 | |
import numpy as np | |
import datetime | |
import os | |
import time | |
import base64 | |
from camera_input_live import camera_input_live | |
def save_image(image): | |
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") | |
filename = f"captured_image_{timestamp}.png" | |
bytes_data = image.getvalue() | |
cv2_img = cv2.imdecode(np.frombuffer(bytes_data, np.uint8), cv2.IMREAD_COLOR) | |
cv2.imwrite(filename, cv2_img) | |
return filename | |
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() | |
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('.') | |
if __name__ == "__main__": | |
main() | |