adding live camera output

deployment/edge/ansible/files/docker-compose.template.yml

@@ -45,9 +45,9 @@ services:
     build:
       context: edge_streamlit
       dockerfile: Dockerfile
-    # devices:
-    #   - /dev/video0:/dev/video0
-    #   - /dev/video2:/dev/video2
+    devices:
+      - /dev/video0:/dev/video0
+      - /dev/video2:/dev/video2
     ports:
       - 8502:8501
     profiles: [edge]

edge_streamlit/Dockerfile

@@ -20,6 +20,7 @@ COPY requirements.txt .
 RUN pip install -r requirements.txt
 
 COPY app.py .
+COPY utils.py .
 COPY prediction_boxes.py .
 
 EXPOSE 8501

edge_streamlit/app.py

@@ -6,6 +6,8 @@
 from prediction_boxes import filtering_items_that_have_predictions, plot_predictions
 from PIL import Image
 from io import BytesIO
+from utils import display_camera_checkboxes, list_cameras
+import cv2
 
 # Page configuration
 st.set_page_config(page_title="VIO-edge", page_icon="🔦", layout="wide")
@@ -28,6 +30,8 @@ def main():
         st.session_state.trigger = False
     if "item_id" not in st.session_state:
         st.session_state.item_id = None
+    if "recording" not in st.session_state:
+        st.session_state.recording = None
 
     col1, col2, col3 = st.columns(3)
 
@@ -105,6 +109,36 @@ def main():
             unsafe_allow_html=True,
         )
 
+    # Sidebar parameters
+    st.sidebar.title("Configuration")
+    available_cameras = list_cameras()
+    selected_cameras = display_camera_checkboxes(available_cameras)
+    st.session_state.selected_cameras = selected_cameras
+
+    if st.sidebar.button("Start/Stop Recording"):
+        st.session_state.recording = not st.session_state.recording
+
+    # Video capture logic
+    if st.session_state.recording and selected_cameras:
+        caps = {index: cv2.VideoCapture(index) for index in selected_cameras}
+        columns = st.columns(len(selected_cameras))
+        frames_video = {
+            index: columns[i].empty() for i, index in enumerate(selected_cameras)
+        }
+
+        while st.session_state.recording:
+            for index in selected_cameras:
+                ret, frame = caps[index].read()
+                if st.session_state.trigger:
+                    st.session_state.image = frame
+
+                if ret:
+                    frames_video[index].image(frame, channels="BGR")
+                else:
+                    frames_video[index].text(
+                        f"Camera {index} - Failed to capture video"
+                    )
+
 
 if __name__ == "__main__":
     main()

edge_streamlit/utils.py

@@ -0,0 +1,45 @@
+import logging
+from typing import List
+
+import cv2
+import streamlit as st
+import os
+
+logger = logging.getLogger(__name__)
+logger.addHandler(logging.StreamHandler())
+logger.setLevel(logging.DEBUG)
+
+os.environ["OPENCV_LOG_LEVEL"] = "OFF"
+os.environ["OPENCV_FFMPEG_LOGLEVEL"] = "-8"
+
+
+def list_cameras() -> List[str]:
+    """
+    Liste toutes les caméras disponibles sur le système.
+    """
+    index = 0
+    available_cameras = []
+    while True:
+        cap = cv2.VideoCapture(index)
+        if not cap.read()[0]:
+            break
+        available_cameras.append(f"Cam {index}")
+        cap.release()
+        index += 1
+    return available_cameras
+
+
+def display_camera_checkboxes(available_cameras: List[str]) -> List[int]:
+    """
+    Affiche les caméras disponibles sous forme de cases à cocher.
+    Retourne les indices des caméras sélectionnées.
+    """
+    selected_cameras = []
+    if len(available_cameras) == 0:
+        st.warning("No camera detected.")
+    else:
+        for i, camera_name in enumerate(available_cameras):
+            if st.sidebar.checkbox(camera_name, key=f"camera_{i}"):
+                selected_cameras.append(i)
+        logger.debug(f"Selected cameras are {selected_cameras}")
+    return selected_cameras