Update app.py

app.py

@@ -144,6 +144,7 @@ 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_type = st.sidebar.selectbox("Estilo", ["realismo", "hiperrealismo", "photorealismo", "ilustración", "abstracto", "imaginativo"])
     style_option = st.sidebar.selectbox("Camara", ["Over Showder", "Medium Shot", "Long Shot", "Close-UP", "Dutch Angle", "High Angle", "Low Angle", "Oblique Angle"])
     
     width, height = (360, 640) if format_option == "9:16" else (640, 360) if format_option == "16:9" else (640, 640)
@@ -155,7 +156,7 @@ async def main():
 
     if prompt_checkbox:
         with st.spinner("Generando prompts mejorados..."):
-            prompts = await generate_variations(prompt, num_variants, True, style_option)
+            prompts = await generate_variations(f"{prompt}, estilo: {style_option}, {style_type}", num_variants, prompt_checkbox, style_option) if prompt_checkbox else [f"{prompt}, estilo: {style_option}, {style_type}"]
     else:
         prompts = [f"{prompt}, estilo: {style_option}"]