Update app.py

app.py

@@ -1,11 +1,15 @@
 import streamlit as st
-from processor import process_video, process_image
+from processor import process_frame, process_image, process_video
 import os
 from PIL import Image
 import tempfile
 import cv2
+from streamlit_webrtc import VideoTransformerBase, webrtc_streamer, WebRtcMode, ClientSettings
+import av
+import asyncio
 
-st.set_page_config(page_title="Traffic Violation Detection", layout="wide")
+# Configure Streamlit page
+st.set_page_config(page_title="🚦 Traffic Violation Detection", layout="wide")
 
 st.title("🚦 Traffic Violation Detection App")
 
@@ -86,5 +90,32 @@ elif option == "Video":
 
 elif option == "Live Camera":
     st.header("📷 Live Camera Processing")
-    st.warning("Live camera processing is currently not supported in this app due to Streamlit limitations.")
-    st.info("Consider running the live camera processing separately using your existing script.")
+    st.warning("Live camera processing is in progress. Please allow camera access.")
+
+    # Define settings for WebRTC
+    WEBRTC_CLIENT_SETTINGS = ClientSettings(
+        rtc_configuration={"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]},
+        media_stream_constraints={"video": True, "audio": False},
+    )
+
+    class VideoTransformer(VideoTransformerBase):
+        def __init__(self):
+            self.font_path = "fonts/alfont_com_arial-1.ttf"  # Update the path as needed
+
+        def transform(self, frame):
+            img = frame.to_ndarray(format="bgr24")
+            processed_img = process_frame(img, self.font_path)
+            if processed_img is not None:
+                return processed_img
+            return img
+
+    webrtc_ctx = webrtc_streamer(
+        key="live-camera",
+        mode=WebRtcMode.SENDRECV,
+        client_settings=WEBRTC_CLIENT_SETTINGS,
+        video_transformer_factory=VideoTransformer,
+        async_transform=True,
+    )
+
+    st.info("Live camera feed is being processed. Detected violations will be annotated on the video stream.")
+