Update app.py

app.py

@@ -53,6 +53,10 @@ if 'cv_loaded' not in st.session_state:
     st.session_state['cv_loaded'] = False
 if 'active_tab' not in st.session_state:
     st.session_state['active_tab'] = "Build Titan 🌱"
+if 'cam0_index' not in st.session_state:
+    st.session_state['cam0_index'] = 0
+if 'cam1_index' not in st.session_state:
+    st.session_state['cam1_index'] = 1
 
 @dataclass
 class ModelConfig:
@@ -340,6 +344,20 @@ def get_available_video_devices():
             cap.release()
     return video_devices
 
+def capture_frame(device_index):
+    cap = cv2.VideoCapture(device_index)
+    if not cap.isOpened():
+        logger.error(f"Failed to open camera at index {device_index}")
+        return None
+    ret, frame = cap.read()
+    if ret:
+        # Convert BGR to RGB for PIL compatibility
+        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        cap.release()
+        return Image.fromarray(frame_rgb)
+    cap.release()
+    return None
+
 def mock_search(query: str) -> str:
     if "superhero" in query.lower():
         return "Latest trends: Gold-plated Batman statues, VR superhero battles."
@@ -547,34 +565,49 @@ with tab2:
     selected_cam0 = st.selectbox("Select Camera 0", video_devices, index=default_cam0_index, key="cam0_select")
     selected_cam1 = st.selectbox("Select Camera 1", video_devices, index=default_cam1_index, key="cam1_select")
     
+    # Store selected indices in session state
+    st.session_state['cam0_index'] = int(selected_cam0.split()[-1]) if selected_cam0 else 0
+    st.session_state['cam1_index'] = int(selected_cam1.split()[-1]) if selected_cam1 else (1 if len(video_devices) > 1 else 0)
+    
     slice_count = st.number_input("Image Slice Count 🎞️", min_value=1, max_value=20, value=10, help="How many snaps to dream of? (Automation’s on vacation! 😜)")
     video_length = st.number_input("Video Dream Length (seconds) 🎥", min_value=1, max_value=30, value=10, help="Imagine a vid this long—sadly, we’re stuck with pics for now! 😂")
     
     cols = st.columns(2)
     with cols[0]:
         st.subheader(f"Camera 0 ({selected_cam0}) 🎬" if selected_cam0 else "Camera 0 (None Detected)")
-        cam0_img = st.camera_input("Snap a Shot - Cam 0 📸", key="cam0", help="Click to capture a heroic moment! 🦸‍♂️")
-        if cam0_img:
-            filename = generate_filename(0)
-            with open(filename, "wb") as f:
-                f.write(cam0_img.getvalue())
-            st.image(Image.open(filename), caption=filename, use_container_width=True)
-            logger.info(f"Saved snapshot from Camera 0: {filename}")
-            st.session_state['captured_images'].append(filename)
-            update_gallery()
-        st.info("🚨 Multi-frame capture’s on strike! Snap one at a time—your Titan’s too cool for automation glitches! 😎")
+        if st.button("Snap a Shot - Cam 0 📸", key="cam0_button"):
+            if video_devices:
+                frame = capture_frame(st.session_state['cam0_index'])
+                if frame:
+                    filename = generate_filename(0)
+                    frame.save(filename)
+                    st.image(frame, caption=filename, use_container_width=True)
+                    logger.info(f"Saved snapshot from Camera 0: {filename}")
+                    st.session_state['captured_images'].append(filename)
+                    update_gallery()
+                else:
+                    st.error("Failed to capture from Camera 0!")
+            else:
+                st.error("No camera available to capture from!")
+        st.info("🚨 Multi-frame capture not implemented! Snap one at a time.")
+    
     with cols[1]:
         st.subheader(f"Camera 1 ({selected_cam1}) 🎥" if selected_cam1 else "Camera 1 (None Detected)")
-        cam1_img = st.camera_input("Snap a Shot - Cam 1 📸", key="cam1", help="Grab another epic frame! 🌟")
-        if cam1_img:
-            filename = generate_filename(1)
-            with open(filename, "wb") as f:
-                f.write(cam1_img.getvalue())
-            st.image(Image.open(filename), caption=filename, use_container_width=True)
-            logger.info(f"Saved snapshot from Camera 1: {filename}")
-            st.session_state['captured_images'].append(filename)
-            update_gallery()
-        st.info("🚨 Frame bursts? Nope, manual snaps only! One click, one masterpiece! 🎨")
+        if st.button("Snap a Shot - Cam 1 📸", key="cam1_button"):
+            if video_devices:
+                frame = capture_frame(st.session_state['cam1_index'])
+                if frame:
+                    filename = generate_filename(1)
+                    frame.save(filename)
+                    st.image(frame, caption=filename, use_container_width=True)
+                    logger.info(f"Saved snapshot from Camera 1: {filename}")
+                    st.session_state['captured_images'].append(filename)
+                    update_gallery()
+                else:
+                    st.error("Failed to capture from Camera 1!")
+            else:
+                st.error("No camera available to capture from!")
+        st.info("🚨 Multi-frame capture not implemented! Snap one at a time.")
 
 with tab3:
     st.header("Fine-Tune Titan (NLP) 🔧 (Teach Your Word Wizard Some Tricks!)")