Update app.py

app.py

@@ -45,17 +45,20 @@ if option == "Image":
                     st.error("Failed to process the image.")
 
 elif option == "Video":
-    st.header("🎥 Select and Process Video")
-    video_files = [f for f in os.listdir("videos/") if f.endswith(('.mp4', '.avi', '.mov'))]
-    if not video_files:
-        st.warning("No videos found in the 'videos/' folder.")
-    else:
-        selected_video = st.selectbox("Choose a video to process:", video_files)
-        video_path = os.path.join("videos/", selected_video)
-        st.video(video_path)
+    st.header("🎥 Upload and Process Video")
+    uploaded_video = st.file_uploader("Choose a video...", type=["mp4", "avi", "mov"])
+    
+    if uploaded_video is not None:
+        # Save the uploaded video to a temporary file
+        with open("temp_video.mp4", "wb") as f:
+            f.write(uploaded_video.read())
+        
+        # Display the uploaded video
+        st.video("temp_video.mp4")
+        
         if st.button("Process Video"):
             with st.spinner("Processing Video..."):
-                processed_path = process_video(video_path, "alfont_com_arial-1.ttf")
+                processed_path = process_video("temp_video.mp4", "alfont_com_arial-1.ttf")
                 if processed_path and os.path.exists(processed_path):
                     st.success("Video processed successfully!")
                     st.video(processed_path)
@@ -69,6 +72,7 @@ elif option == "Video":
                 else:
                     st.error("Failed to process the video.")
 
+
 elif option == "Live Camera":
     st.header("📷 Live Camera Feed")
     st.info("Live processing is active. Detected violations will be annotated on the video feed.")
@@ -81,14 +85,19 @@ elif option == "Live Camera":
             self.font_path = "alfont_com_arial-1.ttf"
 
         def transform(self, frame):
-            img = frame.to_ndarray(format="bgr24")
-            processed_img = process_frame(img, self.font_path)
+            img = frame.to_ndarray(format="bgr24")  # Convert the frame to a numpy array
+            processed_img = process_frame(img, self.font_path)  # Process the image using your custom function
             return processed_img
 
+    # Start the WebRTC streaming
     webrtc_ctx = webrtc_streamer(
         key="live-camera",
         rtc_configuration=RTC_CONFIGURATION,
         video_transformer_factory=VideoTransformer,
         media_stream_constraints={"video": True, "audio": False},
         async_transform=True
-    )
+    )
+
+    # Optional: Provide a stop button to end the live stream
+    if webrtc_ctx.video_receiver:
+        st.button("Stop Live Camera", on_click=webrtc_ctx.stop_stream)