Fix device state (#428)

whitphx · Sep 21, 2021 · e7e1649 · e7e1649
1 parent 0469697
commit e7e1649
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/streamlit_webrtc/frontend/src/WebRtcStreamer.tsx b/streamlit_webrtc/frontend/src/WebRtcStreamer.tsx
@@ -29,12 +29,14 @@ interface WebRtcStreamerInnerProps {
   onComponentValueChange: (newComponentValue: ComponentValue) => void;
 }
 const WebRtcStreamerInner: React.VFC<WebRtcStreamerInnerProps> = (props) => {
-  const [videoInput, setVideoInput] = useState<MediaDeviceInfo | null>(null);
-  const [audioInput, setAudioInput] = useState<MediaDeviceInfo | null>(null);
+  const [devices, setDevices] = useState<{
+    video: MediaDeviceInfo | null;
+    audio: MediaDeviceInfo | null;
+  }>({ video: null, audio: null });
   const { state, start, stop } = useWebRtc(
     props,
-    videoInput,
-    audioInput,
+    devices.video,
+    devices.audio,
     props.onComponentValueChange
   );
 
@@ -52,9 +54,7 @@ const WebRtcStreamerInner: React.VFC<WebRtcStreamerInnerProps> = (props) => {
 
   const handleDeviceSelect = useCallback(
     (video: MediaDeviceInfo | null, audio: MediaDeviceInfo | null) => {
-      setVideoInput(video);
-      setAudioInput(audio);
-      // this.setState({ videoInput: video, audioInput: audio });
+      setDevices({ video, audio });
     },
     []
   );
@@ -100,10 +100,7 @@ const WebRtcStreamerInner: React.VFC<WebRtcStreamerInnerProps> = (props) => {
             videoEnabled={videoEnabled}
             audioEnabled={audioEnabled}
             onSelect={handleDeviceSelect}
-            value={{
-              video: videoInput,
-              audio: audioInput,
-            }}
+            value={devices}
           />
         )}
       </Box>