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[];