Update app.py

app.py

@@ -124,14 +124,14 @@ def login_form():
         else:
             st.error("Credenciales incorrectas. Intenta de nuevo.")
 
-async def generate_variations(prompt, num_variants, use_enhanced):
+async def generate_variations(prompt, num_variants, use_enhanced, style):
     prompts = set() 
     while len(prompts) < num_variants:
         if use_enhanced:
-            enhanced_prompt = await improve_prompt(prompt)
+            enhanced_prompt = await improve_prompt(f"{prompt}, estilo: {style}")
             prompts.add(enhanced_prompt)
         else:
-            prompts.add(prompt)
+            prompts.add(f"{prompt}, estilo: {style}")
     return list(prompts)
 
 async def main():
@@ -146,6 +146,8 @@ async def main():
     format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9", "1:1"]) 
     prompt_checkbox = st.sidebar.checkbox("Mejorar Prompt")
     model_option = st.sidebar.selectbox("Modelo", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-dev"])
+    style_option = st.sidebar.selectbox("Estilo", ["realista", "hiperrealista", "photorealista", "imaginativo", "abstracto", "moebius"])
+    
     width, height = (360, 640) if format_option == "9:16" else (640, 360) if format_option == "16:9" else (640, 640)
 
     if prompt_checkbox:
@@ -155,9 +157,9 @@ async def main():
 
     if prompt_checkbox:
         with st.spinner("Generando prompts mejorados..."):
-            prompts = await generate_variations(prompt, num_variants, True)
+            prompts = await generate_variations(prompt, num_variants, True, style_option)
     else:
-        prompts = [prompt] 
+        prompts = [f"{prompt}, estilo: {style_option}"]
 
     if st.sidebar.button("Generar Imágenes"):
         with st.spinner("Generando imágenes..."):
@@ -171,5 +173,6 @@ async def main():
 
     display_gallery()
 
+
 if __name__ == "__main__":
     asyncio.run(main())