Update main.py

main.py

@@ -2,16 +2,16 @@ from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
 from fastapi.staticfiles import StaticFiles
-from starlette.requests import Request
+from fastapi import Request
 
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 
-# In-memory mapping of rooms -> connected websockets
+# Keep track of rooms and their connections
 rooms = {}
 
 @app.get("/")
-async def get(request: Request):
+async def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 @app.websocket("/ws/{room_id}")
@@ -20,14 +20,13 @@ async def websocket_endpoint(websocket: WebSocket, room_id: str):
     if room_id not in rooms:
         rooms[room_id] = []
     rooms[room_id].append(websocket)
-    
     try:
         while True:
             data = await websocket.receive_text()
-            # Relay data to the other peer
-            for peer in rooms[room_id]:
-                if peer != websocket:
-                    await peer.send_text(data)
+            # Broadcast to other clients in room
+            for conn in rooms[room_id]:
+                if conn != websocket:
+                    await conn.send_text(data)
     except WebSocketDisconnect:
         rooms[room_id].remove(websocket)
         if not rooms[room_id]: