Update app.py

app.py

@@ -1,44 +1,20 @@
-import cv2
 import streamlit as st
+import cv2
 import numpy as np
-from PIL import Image
-
-st.title("Live Webcam Stream - Original and Flipped")
-
-# Try accessing the webcam (with different indexes)
-cap = cv2.VideoCapture(0)  # First try index 0
-if not cap.isOpened():
-    st.warning("Webcam not found. Trying with index 1.")
-    cap = cv2.VideoCapture(1)  # Try index 1 if index 0 fails
-    if not cap.isOpened():
-        st.error("Failed to access the webcam. Please check your device or permissions.")
-    else:
-        st.write("Webcam accessed using index 1.")
-
-# If camera is available, proceed with the processing
-if cap.isOpened():
-    # Create two columns to display the original and flipped video streams
-    col1, col2 = st.columns(2)
-    original_placeholder = col1.empty()
-    flipped_placeholder = col2.empty()
+from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
 
-    # Capture and display the webcam feed frame by frame
-    frame = cap.read()[1]  # Capture the first frame
-    if frame is not None:
-        # Convert original frame to RGB format
-        original_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        original_img = Image.fromarray(original_frame)
+class RotatedVideoTransformer(VideoTransformerBase):
+    def transform(self, frame):
+        img = frame.to_ndarray(format="bgr24")  # Convert frame to NumPy array
+        rotated_img = cv2.rotate(img, cv2.ROTATE_90_COUNTERCLOCKWISE)  # Rotate 90 degrees
+        return rotated_img  # Return the rotated frame
 
-        # Flip the frame horizontally
-        flipped_frame = cv2.flip(original_frame, 1)
-        flipped_img = Image.fromarray(flipped_frame)
+st.title("Live Webcam Streaming with Rotation")
 
-        # Display both original and flipped frames in their respective columns
-        original_placeholder.image(original_img, caption="Original Video Stream", use_column_width=True)
-        flipped_placeholder.image(flipped_img, caption="Flipped Video Stream", use_column_width=True)
+# Original Stream
+st.subheader("Original Video Stream")
+webrtc_streamer(key="original")
 
-    # Stop streaming if the user presses the button
-    stop_button = st.button("Stop Streaming")
-    if stop_button:
-        cap.release()
-        st.write("Stream stopped.")
+# Rotated Stream
+st.subheader("Rotated Video Stream")
+webrtc_streamer(key="rotated", video_transformer_factory=RotatedVideoTransformer)