fix: memory error

main.py

@@ -78,7 +78,7 @@ def sendPromptChain(QA, user_prompt):
 
     return prompt_response_dict;
 
-socket_manager = WebSocketManager()
+# socket_manager = WebSocketManager()
 
 app = FastAPI(title="homepage-app")
 api_app = FastAPI(title="api app")
@@ -213,63 +213,63 @@ async def create_upload_file(file: UploadFile):
 
     return {"filename": file.filename}
 
-@api_app.websocket("/ws/{user_id}")
-async def websocket_endpoint_student(websocket: WebSocket, user_id: str):
-    global QA
+# @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"
-    }
+#     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))
+#     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()
+#     try:
+#         while True:
+#             data = await websocket.receive_text()
 
-            prompt_response_dict = sendPromptChain(QA, data)
+#             prompt_response_dict = sendPromptChain(QA, data)
 
-            await socket_manager.broadcast_to_room(user_id, json.dumps(prompt_response_dict))
+#             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)
+#     except WebSocketDisconnect:
+#         await socket_manager.remove_user_from_room(user_id, websocket)
 
-        message = {
-            "message": f"Student {user_id} disconnected"
-        }
+#         message = {
+#             "message": f"Student {user_id} disconnected"
+#         }
 
-        await socket_manager.broadcast_to_room(user_id, json.dumps(message))
-    except RuntimeError as error:
-        print(error)
+#         await socket_manager.broadcast_to_room(user_id, json.dumps(message))
+#     except RuntimeError as error:
+#         print(error)
 
-@api_app.websocket("/ws/{room_id}/{user_id}")
-async def websocket_endpoint_room(websocket: WebSocket, room_id: str, user_id: str):
-    global QA
+# @api_app.websocket("/ws/{room_id}/{user_id}")
+# async def websocket_endpoint_room(websocket: WebSocket, room_id: str, user_id: str):
+#     global QA
 
-    message = {
-        "message": f"Student {user_id} connected to the classroom"
-    }
+#     message = {
+#         "message": f"Student {user_id} connected to the classroom"
+#     }
 
-    await socket_manager.add_user_to_room(room_id, websocket)
-    await socket_manager.broadcast_to_room(room_id, json.dumps(message))
+#     await socket_manager.add_user_to_room(room_id, websocket)
+#     await socket_manager.broadcast_to_room(room_id, json.dumps(message))
 
-    try:
-        while True:
-            data = await websocket.receive_text()
+#     try:
+#         while True:
+#             data = await websocket.receive_text()
 
-            prompt_response_dict = sendPromptChain(QA, data)
+#             prompt_response_dict = sendPromptChain(QA, data)
 
-            await socket_manager.broadcast_to_room(room_id, json.dumps(prompt_response_dict))
+#             await socket_manager.broadcast_to_room(room_id, json.dumps(prompt_response_dict))
 
-    except WebSocketDisconnect:
-        await socket_manager.remove_user_from_room(room_id, websocket)
+#     except WebSocketDisconnect:
+#         await socket_manager.remove_user_from_room(room_id, websocket)
 
-        message = {
-            "message": f"Student {user_id} disconnected from room - {room_id}"
-        }
+#         message = {
+#             "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)
+#         await socket_manager.broadcast_to_room(room_id, json.dumps(message))
+#     except RuntimeError as error:
+#         print(error)