File size: 4,475 Bytes
f30bc3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import time
import cv2
import numpy as np
import streamlit as st
from streamlit_webrtc import webrtc_streamer, WebRtcMode, VideoTransformerBase
# Set wide layout and page title
st.set_page_config(page_title="Live Stream Broadcast", layout="wide")
# -----------------------------------------------------------------------------
# Custom CSS: Hide the timeline parts of the video controls (works in WebKit‑based browsers)
st.markdown(
"""
<style>
video::-webkit-media-controls-timeline {
display: none !important;
}
video::-webkit-media-controls-current-time-display {
display: none !important;
}
video::-webkit-media-controls-time-remaining-display {
display: none !important;
}
</style>
""",
unsafe_allow_html=True,
)
# -----------------------------------------------------------------------------
st.title("Live Streaming Space")
# -----------------------------------------------------------------------------
# A simple video transformer that “buffers” frames for 3 seconds.
# (In a more complex app you might encode these frames into a video segment file.)
class BufferingTransformer(VideoTransformerBase):
def __init__(self):
self.buffer = [] # List to hold frames
self.buffer_duration = 3.0 # Buffer duration in seconds
self.last_flush_time = time.time()
def transform(self, frame):
# Convert the frame (a VideoFrame object) into a NumPy array (BGR format)
img = frame.to_ndarray(format="bgr24")
self.buffer.append(img)
now = time.time()
if now - self.last_flush_time >= self.buffer_duration:
# Here you could (for example) encode/store the buffered frames.
# For this demo we just clear the buffer every 3 seconds.
self.buffer = []
self.last_flush_time = now
# Return the image unchanged for display
return img
# -----------------------------------------------------------------------------
# Sidebar: Password input for broadcasting. Only if you enter the correct password
# ("test123") do you get access to the broadcast controls.
password = st.sidebar.text_input("Enter broadcasting password", type="password")
if password == "test123":
st.sidebar.success("Authenticated for broadcasting!")
broadcast_mode = True
else:
broadcast_mode = False
# -----------------------------------------------------------------------------
if broadcast_mode:
st.sidebar.header("Broadcast Settings")
# (In a real app you might enumerate the actual connected cameras.
# Here we simply provide a dummy list of choices.)
camera_options = ["Camera 0", "Camera 1", "Camera 2"]
camera_choice = st.sidebar.selectbox("Select Camera", camera_options)
camera_index = int(camera_choice.split(" ")[-1])
st.write("### You are in **BROADCASTING** mode")
st.write("Your live stream is being sent to all viewers.")
# Start the WebRTC streamer in SENDONLY mode (broadcasting).
# The same key ("live_stream") is used so that viewers join the same “room.”
webrtc_ctx = webrtc_streamer(
key="live_stream",
mode=WebRtcMode.SENDONLY,
video_device_index=camera_index,
media_stream_constraints={"video": True, "audio": False},
video_transformer_factory=BufferingTransformer,
video_html_attrs={
"controls": True,
"style": {
"width": "100%",
"border": "2px solid #ccc",
"border-radius": "10px",
},
},
rtc_configuration={"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]},
)
else:
st.write("### Viewing Broadcast")
st.info("If you have the broadcasting password, enter it in the sidebar to broadcast your own stream.")
# In viewing mode, join the same “room” in RECVONLY mode so you can see the active stream.
webrtc_ctx = webrtc_streamer(
key="live_stream",
mode=WebRtcMode.RECVONLY,
media_stream_constraints={"video": True, "audio": False},
video_transformer_factory=BufferingTransformer,
video_html_attrs={
"controls": True,
"style": {
"width": "100%",
"border": "2px solid #ccc",
"border-radius": "10px",
},
},
rtc_configuration={"iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]},
)
|