Update app.py

app.py

@@ -184,9 +184,9 @@ def main():
     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:
-        prompts = generate_variations(prompt, num_variants=num_variants, use_enhanced=prompt_enhance)
-        if st.sidebar.button("Generar Imágenes"):
+    if st.sidebar.button("Generar Imágenes"):
+        if prompt:
+            prompts = generate_variations(prompt, num_variants=num_variants, use_enhanced=prompt_enhance)
             generated_images = gen(prompts, width, height, model_option, num_variants)
             
             st.header("Imágenes Generadas")
@@ -195,9 +195,11 @@ def main():
                 with cols[i % 4]:
                     st.image(image_path, use_column_width=True)
                     st.caption(image_prompt)
-    
+        else:
+            st.warning("Por favor, ingrese una descripción de imagen")
+        
     upload_image_to_gallery()
     display_gallery()
 
-if __name__ == "__main__": 
+if __name__ == "__main__":
     main()