From 0cc2c78f5528c52eb18e15c9cfa543712c6d5ef2 Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Sun, 3 Dec 2023 14:04:40 -0500
Subject: [PATCH] move websocket close to fix memory leak

---
 backend/main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/backend/main.py b/backend/main.py
index d62e0b3..f91b25a 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -202,5 +202,5 @@ async def stream_code(websocket: WebSocket):
         await websocket.send_json(
             {"type": "status", "value": "Image generation failed but code is complete."}
         )
-    finally:
-        await websocket.close()
+
+    await websocket.close()