Update app.py

app.py

@@ -1,14 +1,28 @@
 import os
-import random
-import asyncio
 import numpy as np
+import random
 from pathlib import Path
 from PIL import Image
 import streamlit as st
-from huggingface_hub import AsyncInferenceClient
+from huggingface_hub import InferenceClient, AsyncInferenceClient
+import asyncio
 import streamlit_authenticator as stauth
 
-st.set_page_config(layout="wide")
+# Definir usuarios y contraseñas
+names = ["admin"]
+usernames = ["admin"]
+passwords = ["flux33"]
+
+# Crear el objeto de autenticación
+hashed_passwords = stauth.Hasher(passwords).generate()
+authenticator = stauth.Authenticate(
+    names,
+    usernames,
+    hashed_passwords,
+    "imagen_gen_auth",
+    "abcdef",  # Llave para la cookie
+    cookie_expiry_days=30
+)
 
 MAX_SEED = np.iinfo(np.int32).max
 client = AsyncInferenceClient()
@@ -17,21 +31,6 @@ DATA_PATH.mkdir(exist_ok=True)
 
 PREDEFINED_SEED = random.randint(0, MAX_SEED)
 
-credentials = {
-    "usernames": {
-        "admin": {
-            "name": "Admin",
-            "password": stauth.Hasher(["flux33"]).generate()[0]
-        }
-    }
-}
-
-cookie_name = "image_gen"
-key = "abcdef"
-cookie_expiry_days = 1
-
-authenticator = stauth.Authenticate(credentials, cookie_name, key, cookie_expiry_days)
-
 async def generate_image(prompt, width, height, seed):
     try:
         if seed == -1:
@@ -55,7 +54,7 @@ def save_prompt(prompt_text, seed):
         return None
 
 async def gen(prompt, width, height):
-    combined_prompt = prompt
+    combined_prompt = prompt  
 
     seed = PREDEFINED_SEED
     progress_bar = st.progress(0)
@@ -91,9 +90,16 @@ def get_prompts():
     return {file.stem.replace("prompt_", ""): file for file in prompt_files}
 
 def main():
-    name, authentication_status, username = authenticator.login("Login", "sidebar")
+    # 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"])
@@ -145,10 +151,12 @@ def main():
                         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}")
+
     elif authentication_status == False:
-        st.error("Usuario/Contraseña incorrectos")
+        st.error("Nombre de usuario/contraseña incorrectos")
+
     elif authentication_status == None:
-        st.warning("Por favor, ingrese sus credenciales.")
+        st.warning("Por favor, ingrese sus credenciales")
 
 if __name__ == "__main__":
     main()