Update app.py

app.py

@@ -156,28 +156,18 @@ def gen(prompt: str, system_prompt: str, token: str, accountId: str):
     max_attempts_per_image = 3
     used_prompts = set()  # Untuk tracking prompt yang sudah digunakan
     
-    # Generate initial batch of prompts
-    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
         
         while attempts < max_attempts_per_image and not success:
             try:
-                # 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)
+                # Generate prompt baru yang unik
+                current_prompt = generate_text(prompt, system_prompt)
+                while current_prompt in used_prompts:
+                    current_prompt = generate_text(prompt, system_prompt)
                 
-                # 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
+                used_prompts.add(current_prompt)
                 
                 img, used_prompt = generate_image_with_fallback(current_prompt)
                 
@@ -189,7 +179,7 @@ def gen(prompt: str, system_prompt: str, token: str, accountId: str):
                     failed_prompts.append(used_prompt)
                     attempts += 1
             except Exception as e:
-                failed_prompts.append(f"{current_prompt} (Error: {str(e)})")
+                failed_prompts.append(f"Error: {str(e)}")
                 attempts += 1
         
         if not success: