Update app.py

app.py

@@ -53,10 +53,6 @@ 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:
@@ -344,20 +340,6 @@ 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."
@@ -555,59 +537,46 @@ with tab2:
     video_devices = get_available_video_devices()
     st.subheader("Camera Settings ⚙️")
     if not video_devices:
-        st.warning("No video devices detected! 📷 Please connect a camera.")
+        st.warning("No video devices detected by OpenCV! 📷 Browser may still access cameras; ensure permissions are granted.")
     else:
         st.write(f"Detected Video Devices: {', '.join(video_devices)}")
     
     # Camera selection with defaults
     default_cam0_index = 0 if video_devices else -1
     default_cam1_index = 1 if len(video_devices) > 1 else (0 if video_devices else -1)
-    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")
+    selected_cam0 = st.selectbox("Select Camera 0", video_devices or ["Browser Default"], index=default_cam0_index if video_devices else 0, key="cam0_select")
+    selected_cam1 = st.selectbox("Select Camera 1", video_devices or ["Browser Default"], index=default_cam1_index if video_devices else 0, 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)
+    st.info("Note: Camera selection is informational. Actual device used depends on browser settings.")
     
     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)")
-        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.")
-    
+        st.subheader(f"Camera 0 ({selected_cam0}) 🎬")
+        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! 😎")
     with cols[1]:
-        st.subheader(f"Camera 1 ({selected_cam1}) 🎥" if selected_cam1 else "Camera 1 (None Detected)")
-        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.")
+        st.subheader(f"Camera 1 ({selected_cam1}) 🎥")
+        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! 🎨")
 
 with tab3:
     st.header("Fine-Tune Titan (NLP) 🔧 (Teach Your Word Wizard Some Tricks!)")