Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

instrument cancel and create

Browse Source
This commit is contained in:
Abi Raja 2024-04-18 12:59:32 -04:00
parent 07fc02a15d
commit aed7e3dacf
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frontend/src/App.tsx
View File

@ -184,6 +184,8 @@ function App({ navbarComponent }: Props) {
}; };
const cancelCodeGeneration = () => { const cancelCodeGeneration = () => {
addEvent("Cancel");
wsRef.current?.close?.(USER_CLOSE_WEB_SOCKET_CODE); wsRef.current?.close?.(USER_CLOSE_WEB_SOCKET_CODE);
// make sure stop can correct the state even if the websocket is already closed // make sure stop can correct the state even if the websocket is already closed
cancelCodeGenerationAndReset(); cancelCodeGenerationAndReset();
@ -289,6 +291,7 @@ function App({ navbarComponent }: Props) {
setReferenceImages(referenceImages); setReferenceImages(referenceImages);
setInputMode(inputMode); setInputMode(inputMode);
if (referenceImages.length > 0) { if (referenceImages.length > 0) {
addEvent("Create");
await doGenerateCode( await doGenerateCode(
{ {
generationType: "create", generationType: "create",