init

main.py

@@ -1,85 +1,47 @@
-# main.py
-from fastapi import FastAPI, Request
+# main.py (Simplified for Pure WebSockets)
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.middleware.cors import CORSMiddleware
-from aiortc import RTCPeerConnection, RTCSessionDescription, RTCConfiguration, RTCIceServer
+# 🛑 REMOVE: from aiortc import RTCPeerConnection, RTCSessionDescription, ...
 import json
 import re
+# ... (rest of imports)
 
 app = FastAPI()
+# ... (CORS setup remains) ...
 
-# Enable CORS
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],  
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
+# 🛑 REMOVE: ice_servers and rtc_config
 
-peer_connections = {}
+# 🚀 NEW: Pure WebSocket endpoint for RPC
+@app.websocket("/ws") 
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+    print("✅ WebSocket connected for RPC")
 
-# ✅ Add STUN + TURN servers here
-ice_servers = [
-    RTCIceServer(urls="stun:stun.l.google.com:19302"),
-    RTCIceServer(
-        urls="turn:relay1.expressturn.com:3478",
-        username="efl-projects",
-        credential="turn123"
-    )
-]
-
-rtc_config = RTCConfiguration(iceServers=ice_servers)
-
-
-@app.post("/signal")
-async def signal(request: Request):
-    body = await request.json()
-    offer = body["offer"]
-
-    # ✅ Peer connection with TURN support
-    pc = RTCPeerConnection(rtc_config)
-    peer_connections[id(pc)] = pc  # prevent GC
-
-    @pc.on("datachannel")
-    def on_datachannel(channel):
-        print("✅ Data channel opened")
-
-        @channel.on("message")
-        def on_message(message):
+    try:
+        while True:
+            # Receive RPC message from the client
+            message = await websocket.receive_text()
             print("📩 Received:", message)
-
-            rpc = json.loads(message)
-            result = handle_rpc(rpc)
-
-            response = json.dumps({
-                "jsonrpc": "2.0",
-                "id": rpc["id"],
-                "result": result,
-            })
-
-            print("📤 Sending:", response)
-            channel.send(response)
-
-    # Apply remote SDP
-    await pc.setRemoteDescription(
-        RTCSessionDescription(
-            sdp=offer["sdp"],
-            type=offer["type"]
-        )
-    )
-
-    # Create & return answer
-    answer = await pc.createAnswer()
-    await pc.setLocalDescription(answer)
-
-    print("✅ Answer generated and sent")
-
-    return {
-        "answer": {
-            "sdp": pc.localDescription.sdp,
-            "type": pc.localDescription.type
-        }
-    }
-
+            
+            try:
+                rpc = json.loads(message)
+                result = handle_rpc(rpc)
+
+                response = json.dumps({
+                    "jsonrpc": "2.0",
+                    "id": rpc["id"],
+                    "result": result,
+                })
+                
+                # Send the RPC result back over the WebSocket
+                await websocket.send_text(response)
+
+            except json.JSONDecodeError:
+                print(f"Error decoding RPC message: {message}")
+
+    except WebSocketDisconnect:
+        print("🔌 WebSocket disconnected.")
+    # No RTCPeerConnection cleanup needed!
 
 # ✅ Your RPC logic stays the same
 def handle_rpc(rpc):