Update app.py

app.py

@@ -133,6 +133,27 @@ def generate_variations(prompt, num_variants, use_enhanced):
     else:
         return [prompt] * num_variants
 
+def gen(prompts, width, height, model_name, num_variants=1):
+    images = []
+    seeds = []
+    
+    while len(seeds) < num_variants:
+        new_seed = random.randint(0, MAX_SEED)
+        if new_seed not in seeds: 
+            seeds.append(new_seed)
+            
+    try:
+        for idx, (prompt, seed) in enumerate(zip(prompts[:num_variants], seeds)):
+            image, _ = generate_image(prompt, width, height, seed, model_name)
+            image_path = save_image(image, f"generated_image_{seed}.jpg")
+            if image_path:
+                save_prompt(f"generated_image_{seed}.jpg: {prompt}")
+                st.success(f"Imagen {idx + 1} generada")
+                images.append(str(image_path))
+    except Exception as e:
+        st.error(f"Error al generar imágenes: {e}")
+    return images
+
 def get_prompt_for_image(image_name):
     prompts = {}
     try: