Update app.py

app.py

@@ -78,7 +78,7 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
             upscale_image.save(DATA_PATH / f"upscale_image_{seed}.jpg", format="JPEG")
             progress_bar.progress(100)
             image_path.unlink()  
-            return [str(DATA_PATH / f"upscale_image_{seed}.jpg"), str(prompt_file_path)]
+            return [str(DATA_PATH / f"upscale_image_{seed}.jpg"), str(prompt_file_path")]
         else:
             progress_bar.empty()
             return [str(image_path), str(prompt_file_path)]
@@ -124,71 +124,75 @@ def delete_image(image_path):
     except Exception as e:
         st.error(f"Error al borrar la imagen: {e}")
 
-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)
-
-with st.sidebar.expander("Opciones avanzadas", expanded=False):
-    basemodel = st.selectbox("Modelo Base", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"])
-    lora_model = st.selectbox("LORA Realismo", ["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"])
-    format_option = st.selectbox("Formato", ["9:16", "16:9"])
-    process_lora = st.checkbox("Procesar LORA")
-    process_upscale = st.checkbox("Procesar Escalador")
-    upscale_factor = st.selectbox("Factor de Escala", [2, 4, 8], index=0)
-    scales = st.slider("Escalado", 1, 20, 10)
-    steps = st.slider("Pasos", 1, 100, 20)
-    seed = st.number_input("Semilla", value=-1)
-
-if format_option == "9:16":
-    width = 720
-    height = 1280
-else:
-    width = 1280
-    height = 720
-
-if st.sidebar.button("Generar Imagen"):
-    with st.spinner("Mejorando y generando imagen..."):
-        improved_prompt = await improve_prompt(prompt)
-        st.session_state.improved_prompt = improved_prompt
-        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]
-
-    st.write(f"Image paths: {image_paths}")
-
-    if image_paths:
-        if Path(image_paths).exists():
-            st.image(image_paths, caption="Imagen Generada")
-        else:
-            st.error("El archivo de imagen no existe.")
-
-        if prompt_file and Path(prompt_file).exists():
-            prompt_text = Path(prompt_file).read_text()
-            st.write(f"Prompt utilizado: {prompt_text}")
-        else:
-            st.write("El archivo del prompt no está disponible.")
-
-files, usage = get_storage()
-st.text(usage)
-cols = st.columns(6)
-prompts = get_prompts()
-
-for idx, file in enumerate(files):
-    with cols[idx % 6]:
-        image = Image.open(file)
-        prompt_file = prompts.get(Path(file).stem.replace("image_", ""), None)
-        prompt_text = Path(prompt_file).read_text() if prompt_file else "No disponible"
-        
-        st.image(image, caption=f"Imagen {idx+1}")
-        st.write(f"Prompt: {prompt_text}")
-        
-        if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
-            try:
-                os.remove(file)  
-                if prompt_file:
-                    os.remove(prompt_file)  
-                st.success(f"Imagen {idx+1} y su prompt fueron borrados.")
-            except Exception as e:
-                st.error(f"Error al borrar la imagen o prompt: {e}")
+def main():
+    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)
+
+    with st.sidebar.expander("Opciones avanzadas", expanded=False):
+        basemodel = st.selectbox("Modelo Base", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"])
+        lora_model = st.selectbox("LORA Realismo", ["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"])
+        format_option = st.selectbox("Formato", ["9:16", "16:9"])
+        process_lora = st.checkbox("Procesar LORA")
+        process_upscale = st.checkbox("Procesar Escalador")
+        upscale_factor = st.selectbox("Factor de Escala", [2, 4, 8], index=0)
+        scales = st.slider("Escalado", 1, 20, 10)
+        steps = st.slider("Pasos", 1, 100, 20)
+        seed = st.number_input("Semilla", value=-1)
+
+    if format_option == "9:16":
+        width = 720
+        height = 1280
+    else:
+        width = 1280
+        height = 720
+
+    if st.sidebar.button("Generar Imagen"):
+        with st.spinner("Mejorando y generando imagen..."):
+            improved_prompt = asyncio.run(improve_prompt(prompt))
+            st.session_state.improved_prompt = improved_prompt
+            prompt_to_use = st.session_state.get('improved_prompt', prompt)
+            result = asyncio.run(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]
+
+        st.write(f"Image paths: {image_paths}")
+
+        if image_paths:
+            if Path(image_paths).exists():
+                st.image(image_paths, caption="Imagen Generada")
+            else:
+                st.error("El archivo de imagen no existe.")
+
+            if prompt_file and Path(prompt_file).exists():
+                prompt_text = Path(prompt_file).read_text()
+                st.write(f"Prompt utilizado: {prompt_text}")
+            else:
+                st.write("El archivo del prompt no está disponible.")
+
+    files, usage = get_storage()
+    st.text(usage)
+    cols = st.columns(6)
+    prompts = get_prompts()
+
+    for idx, file in enumerate(files):
+        with cols[idx % 6]:
+            image = Image.open(file)
+            prompt_file = prompts.get(Path(file).stem.replace("image_", ""), None)
+            prompt_text = Path(prompt_file).read_text() if prompt_file else "No disponible"
+            
+            st.image(image, caption=f"Imagen {idx+1}")
+            st.write(f"Prompt: {prompt_text}")
+            
+            if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
+                try:
+                    os.remove(file)  
+                    if prompt_file:
+                        os.remove(prompt_file)  
+                    st.success(f"Imagen {idx+1} y su prompt fueron borrados.")
+                except Exception as e:
+                    st.error(f"Error al borrar la imagen o prompt: {e}")
+
+if __name__ == "__main__":
+    main()