Update pages/1_Live_Stream.py

pages/1_Live_Stream.py

@@ -22,6 +22,12 @@ elif mode == 'Pro':
 live_process_frame = ProcessFrame(thresholds=thresholds, flip_frame=True)
 pose = get_mediapipe_pose()
 
+# Input stream configuration
+stream_source = st.text_input("Enter RTSP/stream URL or leave blank for default camera:", "")
+
+if stream_source == "":
+    stream_source = 0  # Default to local camera
+
 # Temporary file to save processed video
 temp_dir = tempfile.TemporaryDirectory()
 output_video_file = os.path.join(temp_dir.name, 'output_live.avi')
@@ -30,17 +36,16 @@ output_video_file = os.path.join(temp_dir.name, 'output_live.avi')
 fourcc = cv2.VideoWriter_fourcc(*'XVID')
 video_writer = None
 
-# Start video capture
-cap = cv2.VideoCapture(0)
+# Process video
+cap = cv2.VideoCapture(stream_source)
 if not cap.isOpened():
-    st.error("Unable to access the camera")
+    st.error("Unable to access the video stream")
 else:
     stframe = st.empty()
 
     while True:
         ret, frame = cap.read()
         if not ret:
-            st.error("Failed to capture video")
             break
 
         frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
@@ -55,10 +60,6 @@ else:
         # Display in Streamlit
         stframe.image(processed_frame, channels="RGB")
 
-        # Stop button
-        if st.button("Stop"):
-            break
-
     cap.release()
     if video_writer:
         video_writer.release()