awacke1 commited on
Commit
9a6f4b3
Β·
verified Β·
1 Parent(s): 82d8aa2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -8,6 +8,7 @@ from PIL import Image
8
  from io import BytesIO
9
  import base64
10
  import cv2
 
11
  import queue
12
 
13
  # 🌟πŸ”₯ Initialize session state like a galactic DJ spinning tracks!
@@ -35,7 +36,7 @@ class CamThread(threading.Thread):
35
  cam.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
36
  cam.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')) # Use MJPG for better USB compatibility
37
  if not cam.isOpened():
38
- st.error(f"🚨 Failed to open {self.preview_name}")
39
  return
40
 
41
  while self.running:
@@ -81,6 +82,7 @@ def auto_capture(frame_queues):
81
  # πŸŽ›οΈ Sidebar config like a spaceship control panel!
82
  with st.sidebar:
83
  st.header("πŸŽšοΈπŸ“Έ Snap Shack")
 
84
  if st.button("⏰ Start Auto-Snap"):
85
  st.session_state['auto_capture_running'] = True
86
  threading.Thread(target=auto_capture, args=(frame_queues,), daemon=True).start()
@@ -102,7 +104,6 @@ with st.sidebar:
102
  st.title("πŸ“Έ OpenCV Snap Craze")
103
 
104
  # πŸ“ΈπŸ“· Start camera threads
105
- frame_queues = [queue.Queue(maxsize=1), queue.Queue(maxsize=1)]
106
  threads = [
107
  CamThread("Camera 0", 0, frame_queues[0]),
108
  CamThread("Camera 1", 1, frame_queues[1])
@@ -166,7 +167,7 @@ if st.session_state['file_history']:
166
  else:
167
  st.write("πŸ•³οΈ Nothing snapped yet!")
168
 
169
- # Cleanup on app exit (not typically needed in Streamlit, but good practice)
170
  def cleanup():
171
  for thread in threads:
172
  thread.stop()
 
8
  from io import BytesIO
9
  import base64
10
  import cv2
11
+ import numpy as np
12
  import queue
13
 
14
  # 🌟πŸ”₯ Initialize session state like a galactic DJ spinning tracks!
 
36
  cam.set(cv2.CAP_PROP_FRAME_HEIGHT, 480)
37
  cam.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc('M', 'J', 'P', 'G')) # Use MJPG for better USB compatibility
38
  if not cam.isOpened():
39
+ st.error(f"🚨 Failed to open {self.preview_name}. Check connection or index.")
40
  return
41
 
42
  while self.running:
 
82
  # πŸŽ›οΈ Sidebar config like a spaceship control panel!
83
  with st.sidebar:
84
  st.header("πŸŽšοΈπŸ“Έ Snap Shack")
85
+ frame_queues = [queue.Queue(maxsize=1), queue.Queue(maxsize=1)] # Define frame_queues here
86
  if st.button("⏰ Start Auto-Snap"):
87
  st.session_state['auto_capture_running'] = True
88
  threading.Thread(target=auto_capture, args=(frame_queues,), daemon=True).start()
 
104
  st.title("πŸ“Έ OpenCV Snap Craze")
105
 
106
  # πŸ“ΈπŸ“· Start camera threads
 
107
  threads = [
108
  CamThread("Camera 0", 0, frame_queues[0]),
109
  CamThread("Camera 1", 1, frame_queues[1])
 
167
  else:
168
  st.write("πŸ•³οΈ Nothing snapped yet!")
169
 
170
+ # Cleanup on app exit
171
  def cleanup():
172
  for thread in threads:
173
  thread.stop()