salomonsky commited on
Commit
d99c194
·
verified ·
1 Parent(s): 283d760

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -30
app.py CHANGED
@@ -52,39 +52,41 @@ def get_storage():
52
  usage = sum([file.stat().st_size for file in files])
53
  return [str(file.resolve()) for file in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
54
 
55
- async def main():
56
- name, authentication_status, username = authenticator.login("Login", location="main")
 
 
 
 
 
57
  if authentication_status:
58
- st.success(f"Bienvenido {name}")
59
- st.title("Generador de Imágenes FLUX")
60
- prompt, format_option = st.sidebar.text_input("Descripción de la imagen", max_chars=500), st.sidebar.selectbox("Formato", ["9:16", "16:9"])
61
- width, height = (720, 1280) if format_option == "9:16" else (1280, 720)
62
- if st.sidebar.button("Generar Imagen"):
63
- with st.spinner("Generando imagen..."):
64
- image = await generate_image(prompt, width, height)
65
- if isinstance(image, str) and image.startswith("Error"):
66
- st.error(image)
 
 
 
 
67
  else:
68
- image_path = save_image(image, PREDEFINED_SEED)
69
- if image_path and Path(image_path).exists():
70
- st.image(image_path, caption="Imagen Generada")
71
- st.subheader("Galería de Imágenes Generadas")
72
- files, usage = get_storage()
73
- cols = st.columns(3)
74
- for idx, file in enumerate(files):
75
- with cols[idx % 3]:
76
- image = Image.open(file)
77
- st.image(image, caption=f"Imagen {idx + 1}")
78
- if st.button(f"Borrar Imagen {idx + 1}", key=f"delete_{idx}"):
79
- try:
80
- os.remove(file)
81
- st.success(f"Imagen {idx + 1} fue borrada.")
82
- except Exception as e:
83
- st.error(f"Error al borrar la imagen: {e}")
84
  elif authentication_status == False:
85
- st.error("Nombre de usuario/contraseña incorrectos")
86
  elif authentication_status == None:
87
- st.warning("Por favor, ingrese sus credenciales")
88
 
89
  if __name__ == "__main__":
90
- asyncio.run(main())
 
52
  usage = sum([file.stat().st_size for file in files])
53
  return [str(file.resolve()) for file in files], f"Uso total: {usage/(1024.0 ** 3):.3f}GB"
54
 
55
+ def main():
56
+ name, authentication_status, username = authenticator.login(
57
+ "Login",
58
+ location="main",
59
+ fields={"Form name": "Login", "Username": "Username", "Password": "Password", "Login": "Login"}
60
+ )
61
+
62
  if authentication_status:
63
+ st.title("Generador de Imágenes con FLUX")
64
+
65
+ archivos_excel = listar_archivos_excel()
66
+ archivo_seleccionado = st.sidebar.selectbox("Selecciona un archivo Excel", archivos_excel)
67
+
68
+ prompt = st.text_input("Introduce el prompt para generar la imagen")
69
+ imagen_base = st.file_uploader("Sube una imagen base (opcional)", type=['jpg', 'jpeg', 'png'])
70
+
71
+ if st.button("Generar Imagen"):
72
+ if prompt:
73
+ imagen_generada = generar_imagen(prompt, imagen_base)
74
+ if imagen_generada:
75
+ st.image(imagen_generada, caption="Imagen Generada", use_column_width=True)
76
  else:
77
+ st.error("Por favor, introduce un prompt.")
78
+
79
+ if archivo_seleccionado:
80
+ with open(archivo_seleccionado, "rb") as file:
81
+ datos = leer_archivo(file)
82
+ if datos is not None:
83
+ num_filas = st.sidebar.slider("Número de filas a mostrar", min_value=1, max_value=100, value=10)
84
+ st.write(datos.head(num_filas))
85
+
 
 
 
 
 
 
 
86
  elif authentication_status == False:
87
+ st.error('Nombre de usuario/contraseña incorrectos')
88
  elif authentication_status == None:
89
+ st.warning('Por favor, introduce tu nombre de usuario y contraseña')
90
 
91
  if __name__ == "__main__":
92
+ main()