Update app.py

app.py

@@ -103,7 +103,8 @@ async def optimize_code(websocket: WebSocket):
         )
         
         for chunk in stream:
-            await websocket.send_text(chunk.choices[0].delta.content)
+            refined_code += chunk.choices[0].delta.content
+            await websocket.send_text(refined_code)
 
     except WebSocketDisconnect:
         print("Client disconnected")
@@ -131,7 +132,8 @@ async def generate_test_bench(websocket: WebSocket):
         )
         
         for chunk in stream:
-            await websocket.send_text(chunk.choices[0].delta.content)
+            refined_code += chunk.choices[0].delta.content
+            await websocket.send_text(refined_code)
 
     except WebSocketDisconnect:
         print("Client disconnected")