From ac47e9d622d087134a0f38d8dd510a4bcf083a13 Mon Sep 17 00:00:00 2001
From: clean99 <xff9924@gmail.com>
Date: Sun, 26 Nov 2023 14:51:39 +0800
Subject: [PATCH] feat: update do generate code

---
 frontend/src/App.tsx                        | 6 +++---
 frontend/src/components/ReactCodeEditor.tsx | 4 ++--
 frontend/src/generateCode.ts                | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx
index d2b9fc8..fdd7ad6 100644
--- a/frontend/src/App.tsx
+++ b/frontend/src/App.tsx
@@ -96,7 +96,7 @@ function App() {
     setAppState(AppState.CODE_READY);
   };
 
-  function doGenerateCode(params: CodeGenerationParams) {
+  function doGenerateCode(params: CodeGenerationParams, setCode: (value: React.SetStateAction<string>) => void = setGeneratedCode) {
     setExecutionConsole([]);
     setAppState(AppState.CODING);
 
@@ -106,8 +106,8 @@ function App() {
     generateCode(
       wsRef,
       updatedParams,
-      (token) => setGeneratedCode((prev) => prev + token),
-      (code) => setGeneratedCode(code),
+      (token) => setCode((prev) => prev + token),
+      (code) => setCode(code),
       (line) => setExecutionConsole((prev) => [...prev, line]),
       () => setAppState(AppState.CODE_READY)
     );
diff --git a/frontend/src/components/ReactCodeEditor.tsx b/frontend/src/components/ReactCodeEditor.tsx
index 8f43d97..829ce5c 100644
--- a/frontend/src/components/ReactCodeEditor.tsx
+++ b/frontend/src/components/ReactCodeEditor.tsx
@@ -19,7 +19,7 @@ import { Textarea } from "./ui/textarea";
 
 
 interface IProps {
-    doGenerateCode: (params: CodeGenerationParams) => void;
+    doGenerateCode: (params: CodeGenerationParams, setCode: (value: React.SetStateAction<string>) => void) => void;
     referenceImage: string;
 }
 
@@ -44,7 +44,7 @@ export const ReactCodeEditor: React.FC<IProps> = ({ doGenerateCode, referenceIma
             codeType: CodeType.REACT,
             image: referenceImage,
             history: updatedHistory,
-        });
+        }, setGeneratedReactCode);
 
         setReactHistory(updatedHistory);
         setGeneratedReactCode("");
diff --git a/frontend/src/generateCode.ts b/frontend/src/generateCode.ts
index 917b7db..aab10aa 100644
--- a/frontend/src/generateCode.ts
+++ b/frontend/src/generateCode.ts
@@ -8,7 +8,7 @@ const ERROR_MESSAGE =
 const STOP_MESSAGE = "Code generation stopped";
 
 export enum CodeType {
-  HTML = "HTML",
+  HTML = "html",
   REACT = "react",
 }