Update app.py

app.py

@@ -149,29 +149,36 @@ def gen(prompt: str, system_prompt: str, token: str, accountId: str):
     if token and accountId:
         setAccount(token, accountId)
         
-    # Generate multiple optimized prompts
-    optimized_prompts = generate_multiple_prompts(prompt, system_prompt)
-    
-    # Generate images for each prompt
     images = []
     successful_prompts = []
     failed_prompts = []
+    target_images = 10  # Jumlah gambar yang diinginkan
     
-    for opt_prompt in optimized_prompts:
-        img, used_prompt = generate_image_with_fallback(opt_prompt)
-        if img is not None:
-            images.append(img)
-            successful_prompts.append(used_prompt)
-        else:
-            failed_prompts.append(used_prompt)
+    while len(images) < target_images:
+        # Generate batch of prompts
+        optimized_prompts = generate_multiple_prompts(prompt, system_prompt)
+        
+        # Try each prompt until we get enough images
+        for opt_prompt in optimized_prompts:
+            if len(images) >= target_images:
+                break
+                
+            img, used_prompt = generate_image_with_fallback(opt_prompt)
+            if img is not None:
+                images.append(img)
+                successful_prompts.append(used_prompt)
+            else:
+                failed_prompts.append(used_prompt)
+                # Jika gagal, generate prompt baru dan coba lagi
+                continue
     
     # Prepare output text
-    text = f"Original Prompt: {prompt}\n\nSuccessful Generations:\n"
+    text = f"Original Prompt: {prompt}\n\nSuccessful Generations ({len(successful_prompts)}):\n"
     for i, p in enumerate(successful_prompts, 1):
         text += f"\n{i}. {p}"
     
     if failed_prompts:
-        text += "\n\nFailed Generations (NSFW or Error):\n"
+        text += f"\n\nFailed Attempts ({len(failed_prompts)}):\n"
         for i, p in enumerate(failed_prompts, 1):
             text += f"\n{i}. {p}"