Update app.py

app.py

@@ -88,7 +88,7 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
 
 async def improve_prompt(prompt):
     try:
-        instruction = ("With this idea, describe in English a detailed txt2img prompt in 300 character at most...")
+        instruction = ("With this idea, describe in English a detailed txt2img prompt in 300 characters at most...")
         formatted_prompt = f"{instruction}"
         response = llm_client.text_generation(formatted_prompt, max_new_tokens=300)
         improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
@@ -124,12 +124,6 @@ def delete_image(image_path):
     except Exception as e:
         st.error(f"Error al borrar la imagen: {e}")
 
-def run_gen():
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    prompt_to_use = st.session_state.get('improved_prompt', prompt)
-    return loop.run_until_complete(gen(prompt_to_use, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora))
-
 st.set_page_config(layout="wide")
 st.title("Generador de Imágenes FLUX")
 prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=200)
@@ -153,13 +147,12 @@ else:
     height = 720
 
 if st.sidebar.button("Generar Imagen"):
-    with st.spinner("Mejorando prompt..."):
-        improved_prompt = asyncio.run(improve_prompt(prompt))
+    with st.spinner("Mejorando y generando imagen..."):
+        improved_prompt = await improve_prompt(prompt)
         st.session_state.improved_prompt = improved_prompt
-        st.write(f"Prompt mejorado: {improved_prompt}")
-    
-    with st.spinner("Generando imagen..."):
-        result = run_gen()
+        prompt_to_use = st.session_state.get('improved_prompt', prompt)
+        result = await gen(prompt_to_use, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora)
+
         image_paths = result[0]
         prompt_file = result[1]