fix: convert json in string

main.py

@@ -214,7 +214,7 @@ async def create_upload_file(file: UploadFile):
     return {"filename": file.filename}
 
 @api_app.websocket("/ws/{room_id}/{user_id}")
-async def websocket_endpoint(websocket: WebSocket,  room_id: str):
+async def websocket_endpoint_room(websocket: WebSocket, room_id: str, user_id: str):
     global QA
 
     await socket_manager.add_user_to_room(room_id, websocket)
@@ -237,9 +237,39 @@ async def websocket_endpoint(websocket: WebSocket,  room_id: str):
         await socket_manager.remove_user_from_room(room_id, websocket)
 
         message = {
-            "message": f"User {user_id} disconnected from room - {room_id}"
+            "message": f"Student {user_id} disconnected from room - {room_id}"
         }
 
         await socket_manager.broadcast_to_room(room_id, json.dumps(message))
     except RuntimeError as error:
         print(error)
+
+@api_app.websocket("/ws/{user_id}")
+async def websocket_endpoint_student(websocket: WebSocket, user_id: str):
+    global QA
+
+    message = {
+        "message": f"Student {user_id} connected"
+    }
+
+    await socket_manager.add_user_to_room(user_id, websocket)
+    await socket_manager.broadcast_to_room(user_id, json.dumps(message))
+
+    try:
+        while True:
+            data = await websocket.receive_text()
+
+            prompt_response_dict = sendPromptChain(QA, data)
+
+            await socket_manager.broadcast_to_room(user_id, json.dumps(prompt_response_dict))
+
+    except WebSocketDisconnect:
+        await socket_manager.remove_user_from_room(user_id, websocket)
+
+        message = {
+            "message": f"Student {user_id} disconnected"
+        }
+
+        await socket_manager.broadcast_to_room(user_id, json.dumps(message))
+    except RuntimeError as error:
+        print(error)