From 9ed42672e731f9e8c4c1f2344fd05cd6507f4098 Mon Sep 17 00:00:00 2001
From: clean99 <xff9924@gmail.com>
Date: Sun, 26 Nov 2023 13:45:43 +0800
Subject: [PATCH] feat: update code generation interface with type

---
 frontend/src/generateCode.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/frontend/src/generateCode.ts b/frontend/src/generateCode.ts
index 8a05fe7..917b7db 100644
--- a/frontend/src/generateCode.ts
+++ b/frontend/src/generateCode.ts
@@ -7,8 +7,14 @@ const ERROR_MESSAGE =
 
 const STOP_MESSAGE = "Code generation stopped";
 
+export enum CodeType {
+  HTML = "HTML",
+  REACT = "react",
+}
+
 export interface CodeGenerationParams {
   generationType: "create" | "update";
+  codeType: CodeType;
   image: string;
   resultImage?: string;
   history?: string[];