diff --git a/src/views/FlexibleGenerationView.vue b/src/views/FlexibleGenerationView.vue
index db3928a..3a75e5c 100644
--- a/src/views/FlexibleGenerationView.vue
+++ b/src/views/FlexibleGenerationView.vue
@@ -269,8 +269,15 @@ const handleGenerate = async () => {
 
         const response = await aiService.runGeneration(payload)
 
-        // Response can be a single generation or an array of grouped generations
-        const generations = Array.isArray(response) ? response : [response]
+        // Response can be a single generation, an array, or a Group response with 'generations'
+        let generations = []
+        if (response && response.generations) {
+            generations = response.generations
+        } else if (Array.isArray(response)) {
+            generations = response
+        } else {
+            generations = [response]
+        }
 
         for (const gen of generations) {
             if (gen && gen.id) {
@@ -892,7 +899,7 @@ const confirmAddToAlbum = async () => {
                                     </div>
                                     <i class="pi pi-spin pi-spinner text-violet-500 text-xl mb-2 relative z-10"></i>
                                     <span class="text-[10px] text-violet-300/70 relative z-10 capitalize">{{ item.status
-                                        }}...</span>
+                                    }}...</span>
                                     <span v-if="item.progress"
                                         class="text-[9px] text-violet-400/60 font-mono mt-1 relative z-10">{{
                                             item.progress }}%</span>
@@ -956,7 +963,7 @@ const confirmAddToAlbum = async () => {
                                                 @click.stop="reuseAsset(item)" />
                                         </div>
                                         <p class="text-[10px] text-white/70 line-clamp-1 leading-tight">{{ item.prompt
-                                            }}</p>
+                                        }}</p>
                                     </div>
                                 </div>
                             </div>
diff --git a/src/views/ImageGenerationView.vue b/src/views/ImageGenerationView.vue
index 80fc38e..9c57c73 100644
--- a/src/views/ImageGenerationView.vue
+++ b/src/views/ImageGenerationView.vue
@@ -211,8 +211,15 @@ const handleGenerate = async () => {
 
         const response = await aiService.runGeneration(payload)
 
-        if (response && response.id) {
-            pollStatus(response.id)
+        let genId = null
+        if (response && response.generations && response.generations.length > 0) {
+            genId = response.generations[0].id
+        } else if (response && response.id) {
+            genId = response.id
+        }
+
+        if (genId) {
+            pollStatus(genId)
         } else {
             // Fallback immediate response
             generatedResult.value = response