From f1a9859384a1b7d7b777b75d41ec03dbdb3f2020 Mon Sep 17 00:00:00 2001
From: Abi Raja <abimanyuraja@gmail.com>
Date: Thu, 14 Dec 2023 08:29:04 -0500
Subject: [PATCH] fix bug with history not being updated if image generation
 fails

---
 backend/routes/generate_code.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
index d34f861..b260cc0 100644
--- a/backend/routes/generate_code.py
+++ b/backend/routes/generate_code.py
@@ -254,6 +254,9 @@ async def stream_code(websocket: WebSocket):
     except Exception as e:
         traceback.print_exc()
         print("Image generation failed", e)
+        # Send set code even if image generation fails since that triggers
+        # the frontend to update history
+        await websocket.send_json({"type": "setCode", "value": completion})
         await websocket.send_json(
             {"type": "status", "value": "Image generation failed but code is complete."}
         )