Update app.py

app.py

@@ -363,11 +363,19 @@ async def handle_user_input(websocket,key, process, connected):
                                 
                                         return_code = await process.wait()
                                         if return_code == 0:
-                                            await send_message('Code executed successfully')
-                                            pass
+                                            event = {
+                                                "type": "terminal-data-end",
+                                               
+                                            }
+                                            await websocket.send(json.dumps(event))
+                                            
                                         else:
-                                            await send_message(f'error:Execution failed with return code {return_code}')
-                                            pass
+                                            event = {
+                                                "type": "terminal-data-end",
+                                               
+                                            }
+                                            await websocket.send(json.dumps(event))
+                                            
                                 
                                     except Exception as e:
                                         await error(websocket, str(e))