Update app.py

app.py

@@ -31,18 +31,6 @@ DATA_PATH.mkdir(exist_ok=True)
 
 PREDEFINED_SEED = random.randint(0, MAX_SEED)
 
-async def generate_image(prompt, width, height, seed):
-    try:
-        if seed == -1:
-            seed = PREDEFINED_SEED
-        seed = int(seed)
-        image = await client.text_to_image(
-            prompt=prompt, height=height, width=width, model="enhanceaiteam/Flux-uncensored"
-        )
-        return image, seed
-    except Exception as e:
-        return f"Error al generar imagen: {e}", None
-
 def save_prompt(prompt_text, seed):
     try:
         prompt_file_path = DATA_PATH / f"prompt_{seed}.txt"
@@ -70,15 +58,6 @@ async def gen(prompt, width, height):
 
     return [str(image_path), str(prompt_file_path)]
 
-def save_image(image, seed):
-    try:
-        image_path = DATA_PATH / f"image_{seed}.jpg"
-        image.save(image_path, format="JPEG")
-        return image_path
-    except Exception as e:
-        st.error(f"Error al guardar la imagen: {e}")
-        return None
-
 def get_storage():
     files = [file for file in DATA_PATH.glob("*.jpg") if file.is_file()]
     files.sort(key=lambda x: x.stat().st_mtime, reverse=True)
@@ -89,68 +68,63 @@ def get_prompts():
     prompt_files = [file for file in DATA_PATH.glob("*.txt") if file.is_file()]
     return {file.stem.replace("prompt_", ""): file for file in prompt_files}
 
+async def generate_image(prompt, width, height):
+    try:
+        image = await client.text_to_image(
+            prompt=prompt, height=height, width=width, model="enhanceaiteam/Flux-uncensored"
+        )
+        return image
+    except Exception as e:
+        return f"Error al generar imagen: {e}"
+
+def save_image(image, seed):
+    try:
+        image_path = DATA_PATH / f"image_{seed}.jpg"
+        image.save(image_path, format="JPEG")
+        return image_path
+    except Exception as e:
+        st.error(f"Error al guardar la imagen: {e}")
+        return None
+
 def main():
-    # Configuración de página
     st.set_page_config(layout="wide")
-
-    # Autenticación
     name, authentication_status, username = authenticator.login("Login", "main")
 
     if authentication_status:
         st.success(f"Bienvenido {name}")
 
-        # Aquí comienza la lógica de generación de imágenes después de autenticarse
         st.title("Generador de Imágenes FLUX")
         prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=500)
         format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9"])
 
-        if format_option == "9:16":
-            width = 720
-            height = 1280
-        else:
-            width = 1280
-            height = 720
+        width, height = (720, 1280) if format_option == "9:16" else (1280, 720)
 
         if st.sidebar.button("Generar Imagen"):
             with st.spinner("Generando imagen..."):
-                result = asyncio.run(gen(prompt, width, height))
-                image_paths = result[0]
-                prompt_file = result[1]
-
-            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.")
-
+                image = await generate_image(prompt, width, height)
+                
+            if isinstance(image, str) and image.startswith("Error"):
+                st.error(image)
+            else:
+                image_path = save_image(image, PREDEFINED_SEED)
+                if image_path and Path(image_path).exists():
+                    st.image(image_path, caption="Imagen Generada")
+
+        st.subheader("Galería de Imágenes Generadas")
         files, usage = get_storage()
-        st.text(usage)
-        cols = st.columns(6)
-        prompts = get_prompts()
+        cols = st.columns(3)  
 
         for idx, file in enumerate(files):
-            with cols[idx % 6]:
+            with cols[idx % 3]:
                 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}")
+                st.image(image, caption=f"Imagen {idx + 1}")
 
-                if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
+                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.")
+                        st.success(f"Imagen {idx + 1} fue borrada.")
                     except Exception as e:
-                        st.error(f"Error al borrar la imagen o prompt: {e}")
+                        st.error(f"Error al borrar la imagen: {e}")
 
     elif authentication_status == False:
         st.error("Nombre de usuario/contraseña incorrectos")