diff --git a/frontend/src/App.tsx b/frontend/src/App.tsx
index fd1576e..83d5d87 100644
--- a/frontend/src/App.tsx
+++ b/frontend/src/App.tsx
@@ -269,8 +269,7 @@ function App() {
       doGenerateCode(
         {
           generationType: "create",
-          image: referenceImages[0],
-          inputMode,
+          images: referenceImages,
         },
         currentVersion
       );
@@ -303,8 +302,7 @@ function App() {
       doGenerateCode(
         {
           generationType: "update",
-          inputMode,
-          image: referenceImages[0],
+          images: referenceImages,
           resultImage: resultImage,
           history: updatedHistory,
           isImportedFromCode,
@@ -315,8 +313,7 @@ function App() {
       doGenerateCode(
         {
           generationType: "update",
-          inputMode,
-          image: referenceImages[0],
+          images: referenceImages,
           history: updatedHistory,
           isImportedFromCode,
         },
diff --git a/frontend/src/components/ImageUpload.tsx b/frontend/src/components/ImageUpload.tsx
index 366f104..51336d7 100644
--- a/frontend/src/components/ImageUpload.tsx
+++ b/frontend/src/components/ImageUpload.tsx
@@ -69,8 +69,7 @@ function ImageUpload({ setReferenceImages }: Props) {
 
   const { getRootProps, getInputProps, isFocused, isDragAccept, isDragReject } =
     useDropzone({
-      maxFiles: 1,
-      maxSize: 1024 * 1024 * 20, // 20 MB
+      maxSize: 1024 * 1024 * 5, // 5 MB
       accept: {
         // Image formats
         "image/png": [".png"],
diff --git a/frontend/src/types.ts b/frontend/src/types.ts
index bbabeff..620afe5 100644
--- a/frontend/src/types.ts
+++ b/frontend/src/types.ts
@@ -32,8 +32,7 @@ export enum ScreenRecorderState {
 
 export interface CodeGenerationParams {
   generationType: "create" | "update";
-  inputMode: "image" | "video";
-  image: string;
+  images: string[];
   resultImage?: string;
   history?: string[];
   isImportedFromCode?: boolean;