Update app.py

app.py

@@ -1,8 +1,7 @@
 from aiohttp import web
 import json
 
-# In-memory storage for signaling sessions
-# Format: { "session_id": {"offer": {}, "answer": {}, "candidates": []} }
+# Storage for active handshake sessions
 sessions = {}
 
 # Cloudflare TURN Credentials
@@ -18,48 +17,39 @@ ICE_CONFIG = {
 }
 
 async def handle_post(request):
-    """Handles uploading Offers, Answers, and ICE Candidates"""
     data = await request.json()
     sid = data.get("id")
     msg_type = data.get("type")
-    payload = data.get("payload")
-
-    if not sid:
-        return web.json_response({"error": "No ID provided"}, status=400)
-
+    if not sid: return web.json_response({"error": "No ID"}, status=400)
+    
     if sid not in sessions:
         sessions[sid] = {"offer": None, "answer": None, "candidates": []}
-
-    if msg_type == "offer":
-        sessions[sid]["offer"] = payload
-    elif msg_type == "answer":
-        sessions[sid]["answer"] = payload
-    elif msg_type == "candidate":
-        sessions[sid]["candidates"].append(payload)
-
+    
+    if msg_type == "offer": sessions[sid]["offer"] = data["payload"]
+    elif msg_type == "answer": sessions[sid]["answer"] = data["payload"]
+    elif msg_type == "candidate": sessions[sid]["candidates"].append(data["payload"])
+    
     return web.json_response({"status": "stored"})
 
 async def handle_get(request):
-    """Handles retrieving session data and TURN config"""
     sid = request.query.get("id")
     if not sid or sid not in sessions:
-        return web.json_response({"error": "Session not found"}, status=404)
+        return web.json_response({"status": "waiting", "offer": None, "iceConfig": ICE_CONFIG})
     
-    response_data = sessions[sid].copy()
-    response_data["iceConfig"] = ICE_CONFIG
-    return web.json_response(response_data)
+    resp = sessions[sid].copy()
+    resp["iceConfig"] = ICE_CONFIG
+    return web.json_response(resp)
 
-# CORS Middleware to allow browsers to talk to this server
 async def cors_middleware(app, handler):
     async def middleware(request):
         if request.method == 'OPTIONS':
-            response = web.Response()
+            resp = web.Response()
         else:
-            response = await handler(request)
-        response.headers['Access-Control-Allow-Origin'] = '*'
-        response.headers['Access-Control-Allow-Methods'] = 'GET, POST, OPTIONS'
-        response.headers['Access-Control-Allow-Headers'] = 'Content-Type'
-        return response
+            resp = await handler(request)
+        resp.headers['Access-Control-Allow-Origin'] = '*'
+        resp.headers['Access-Control-Allow-Methods'] = 'GET, POST, OPTIONS'
+        resp.headers['Access-Control-Allow-Headers'] = 'Content-Type'
+        return resp
     return middleware
 
 app = web.Application(middlewares=[cors_middleware])
@@ -67,5 +57,5 @@ app.router.add_post('/signal', handle_post)
 app.router.add_get('/signal', handle_get)
 
 if __name__ == '__main__':
-    print("WebRTC Broker Server starting on http://0.0.0.0:3000")
-    web.run_app(app, port=7860)
+    print("Broker running at http://localhost:3000")
+    web.run_app(app, port=3000)