Webrtc to fastrtc

ui.py

@@ -95,46 +95,72 @@ def build_ui():
 
                     // Set up WebRTC connection to Render signaling server
                     async function setupWebRTC() {
-                        if (rtcConnection) {
-                            rtcConnection.close();
-                        }
-                        
-                        // Create new RTCPeerConnection
-                        rtcConnection = new RTCPeerConnection();
-                        
-                        // Add audio track to connection
-                        mediaStream.getAudioTracks().forEach(track => {
-                            rtcConnection.addTrack(track, mediaStream);
-                        });
-                        
-                        // Create data channel for signaling
-                        const dataChannel = rtcConnection.createDataChannel('audio');
-                        
-                        // Create and set local description
-                        const offer = await rtcConnection.createOffer();
-                        await rtcConnection.setLocalDescription(offer);
-                        
-                        // Connect to signaling server and exchange SDP
-                        const signalingUrl = window.RENDER_SIGNALING_URL;
-                        const response = await fetch(signalingUrl, {
-                            method: 'POST',
-                            headers: { 'Content-Type': 'application/json' },
-                            body: JSON.stringify({ sdp: rtcConnection.localDescription })
-                        });
-                        
-                        const data = await response.json();
-                        await rtcConnection.setRemoteDescription(new RTCSessionDescription(data.sdp));
-                        
-                        // Handle ICE candidates
-                        rtcConnection.onicecandidate = event => {
-                            if (event.candidate) {
-                                fetch(signalingUrl, {
-                                    method: 'POST',
-                                    headers: { 'Content-Type': 'application/json' },
-                                    body: JSON.stringify({ candidate: event.candidate })
-                                });
+                        try {
+                            if (rtcConnection) {
+                                rtcConnection.close();
                             }
-                        };
+                            
+                            // Use FastRTC's connection approach
+                            const pc = new RTCPeerConnection({
+                                iceServers: [{ urls: 'stun:stun.l.google.com:19302' }]
+                            });
+                            
+                            // Add audio track
+                            mediaStream.getAudioTracks().forEach(track => {
+                                pc.addTrack(track, mediaStream);
+                            });
+                            
+                            // Connect to FastRTC signaling via WebSocket
+                            const signalWs = new WebSocket(window.RENDER_SIGNALING_URL.replace('wss://', 'wss://'));
+                            
+                            // Handle signaling messages
+                            signalWs.onmessage = async (event) => {
+                                const message = JSON.parse(event.data);
+                                
+                                if (message.type === 'offer') {
+                                    await pc.setRemoteDescription(new RTCSessionDescription(message));
+                                    const answer = await pc.createAnswer();
+                                    await pc.setLocalDescription(answer);
+                                    signalWs.send(JSON.stringify(pc.localDescription));
+                                } else if (message.type === 'candidate') {
+                                    if (message.candidate) {
+                                        await pc.addIceCandidate(new RTCIceCandidate(message));
+                                    }
+                                }
+                            };
+                            
+                            // Send ICE candidates
+                            pc.onicecandidate = (event) => {
+                                if (event.candidate) {
+                                    signalWs.send(JSON.stringify({
+                                        type: 'candidate',
+                                        candidate: event.candidate
+                                    }));
+                                }
+                            };
+                            
+                            // Keep connection reference
+                            rtcConnection = pc;
+                            
+                            // Wait for connection to be established
+                            await new Promise((resolve, reject) => {
+                                const timeout = setTimeout(() => reject(new Error("WebRTC connection timeout")), 10000);
+                                pc.onconnectionstatechange = () => {
+                                    if (pc.connectionState === 'connected') {
+                                        clearTimeout(timeout);
+                                        resolve();
+                                    } else if (pc.connectionState === 'failed' || pc.connectionState === 'disconnected') {
+                                        clearTimeout(timeout);
+                                        reject(new Error("WebRTC connection failed"));
+                                    }
+                                };
+                            });
+                            
+                            updateStatus('connected');
+                        } catch (err) {
+                            console.error('WebRTC setup error:', err);
+                            updateStatus('error', 'WebRTC setup failed: ' + err.message);
+                        }
                     }
                     
                     // Set up WebSocket connection to HF Space for conversation updates