Update app.py

app.py

@@ -153,24 +153,32 @@ def gen(prompt: str, system_prompt: str, token: str, accountId: str):
     successful_prompts = []
     failed_prompts = []
     target_images = 10
-    max_attempts_per_image = 3  # Maksimal retry per gambar individual
+    max_attempts_per_image = 3
+    used_prompts = set()  # Untuk tracking prompt yang sudah digunakan
     
     # Generate initial batch of prompts
-    optimized_prompts = generate_multiple_prompts(prompt, system_prompt)
+    optimized_prompts = [p.strip() for p in generate_multiple_prompts(prompt, system_prompt)]
+    # Hilangkan duplikat dari batch awal
+    optimized_prompts = list(dict.fromkeys(optimized_prompts))
     
     # Proses setiap target gambar
     for i in range(target_images):
         attempts = 0
         success = False
         
-        # Retry untuk gambar individual ini
         while attempts < max_attempts_per_image and not success:
             try:
-                # Ambil prompt dari batch, atau generate baru jika habis
-                if i + attempts >= len(optimized_prompts):
-                    optimized_prompts.extend(generate_multiple_prompts(prompt, system_prompt))
+                # Cek apakah perlu generate prompt baru
+                while len(optimized_prompts) <= i + attempts or all(p in used_prompts for p in optimized_prompts):
+                    new_prompts = [p.strip() for p in generate_multiple_prompts(prompt, system_prompt)]
+                    # Hanya tambahkan prompt yang belum pernah digunakan
+                    new_prompts = [p for p in new_prompts if p not in used_prompts]
+                    optimized_prompts.extend(new_prompts)
+                
+                # Ambil prompt yang belum digunakan
+                current_prompt = next(p for p in optimized_prompts if p not in used_prompts)
+                used_prompts.add(current_prompt)  # Tandai sebagai sudah digunakan
                 
-                current_prompt = optimized_prompts[i + attempts]
                 img, used_prompt = generate_image_with_fallback(current_prompt)
                 
                 if img is not None:
@@ -184,7 +192,6 @@ def gen(prompt: str, system_prompt: str, token: str, accountId: str):
                 failed_prompts.append(f"{current_prompt} (Error: {str(e)})")
                 attempts += 1
         
-        # Jika masih gagal setelah max attempts
         if not success:
             failed_prompts.append(f"Failed after {max_attempts_per_image} attempts for image #{i+1}")