Update app.py

app.py

@@ -113,22 +113,25 @@ def save_prompt(prompt):
         st.success("Prompt guardado.")
 
 def generate_variations(prompt, num_variants, use_enhanced):
-    instructions = [
-        "With this words, create a photorealistic description for a detailed txt2img prompt in English in 200 characters maximum",
-        "With this idea, write a creative, realistic, and detailed text-to-image prompt in English in 200 characters maximum",
-        "With this text, generate a descriptive and True to life txt2img prompt in English in 200 characters maximum",
-        "With my idea, describe a photorealistic scene with detailed illumination for a txt2img prompt in English in 200 characters maximum",
-        "With this concept, give a realistic, elegant txt2img prompt in English, emphasizing photorealism in 200 characters maximum",
-        "With this perspective, conform a visually dynamic and hyperrealistic txt2img prompt in English in 200 characters maximum",
-        "With this inspiration, realize a cinematic txt2img prompt in English with hyperrealistic elements in 200 characters maximum",
-        "With my idea, make a lifelike and txt2img prompt in English, focusing on photorealistic depth in 200 characters maximum"
-    ]
-    prompts = set()
-    while len(prompts) < num_variants:
-        instruction = random.choice(instructions)
-        enhanced_prompt = f"{prompt}"
-        prompts.add(enhanced_prompt)
-    return list(prompts)
+    if use_enhanced:
+        instructions = [
+            "With this words, create a photorealistic description for a detailed txt2img prompt in English in 200 characters maximum",
+            "With this idea, write a creative, realistic, and detailed text-to-image prompt in English in 200 characters maximum",
+            "With this text, generate a descriptive and True to life txt2img prompt in English in 200 characters maximum",
+            "With my idea, describe a photorealistic scene with detailed illumination for a txt2img prompt in English in 200 characters maximum",
+            "With this concept, give a realistic, elegant txt2img prompt in English, emphasizing photorealism in 200 characters maximum",
+            "With this perspective, conform a visually dynamic and hyperrealistic txt2img prompt in English in 200 characters maximum",
+            "With this inspiration, realize a cinematic txt2img prompt in English with hyperrealistic elements in 200 characters maximum",
+            "With my idea, make a lifelike and txt2img prompt in English, focusing on photorealistic depth in 200 characters maximum"
+        ]
+        prompts = set()
+        while len(prompts) < num_variants:
+            instruction = random.choice(instructions)
+            enhanced_prompt = f"{instruction}: {prompt}"
+            prompts.add(enhanced_prompt)
+        return list(prompts)
+    else:
+        return [prompt] * num_variants
 
 def get_prompt_for_image(image_name):
     prompts = {}
@@ -203,17 +206,18 @@ def main():
     prompt_enhance = st.sidebar.checkbox("Mejorar Prompt", True)
     num_variants = st.sidebar.slider("Número de imágenes", 1, 8, 8)
     width, height = (720, 1280) if format_option == "9:16" else (1280, 720) if format_option == "16:9" else (1280, 1280)
-    if prompt_enhance and prompt:
-        prompts = generate_variations(prompt, num_variants=num_variants, use_enhanced=True)
-    else:
-        prompts = [prompt] * num_variants
-    if st.sidebar.button("Generar Imágenes"):
-        images = gen(prompts, width, height, model_option, num_variants)
+    
+    if prompt:
+        prompts = generate_variations(prompt, num_variants=num_variants, use_enhanced=prompt_enhance)
+        if st.sidebar.button("Generar Imágenes"):
+            images = gen(prompts, width, height, model_option, num_variants)
+            
     if generated_image_path and upscale_checkbox:
         upscale_factor = st.sidebar.slider("Factor de Escalado", 1, 4, 2)
         improved_image = get_upscale_finegrain(prompt, generated_image_path, upscale_factor)
         if improved_image:
             st.image(improved_image, caption="Imagen Escalada", use_column_width=True)
+            
     upload_image_to_gallery()
     display_gallery()