Update app.py

app.py

@@ -1,7 +1,7 @@
 import os; import numpy as np; import random; from pathlib import Path; from PIL import Image; import streamlit as st; from huggingface_hub import AsyncInferenceClient; import asyncio; import yaml
 
 with open("config.yaml") as f: config = yaml.safe_load(f); username = config["credentials"]["username"]; password = config["credentials"]["password"]
-
+    
 client = AsyncInferenceClient(); DATA_PATH = Path("./data"); DATA_PATH.mkdir(exist_ok=True); MAX_SEED = np.iinfo(np.int32).max; PREDEFINED_SEED = random.randint(0, MAX_SEED)
 
 async def generate_image(prompt, width, height, seed):
@@ -43,26 +43,50 @@ def get_prompts():
     return {file.stem.replace("prompt_", ""): file for file in DATA_PATH.glob("*.txt") if file.is_file()}
 
 def main():
-    st.set_page_config(layout="wide"); st.title("Generador de Imágenes")
-    prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=500)
-    width, height = (720, 1280) if st.sidebar.selectbox("Formato", ["9:16", "16:9"]) == "9:16" else (1280, 720)
+    st.set_page_config(layout="wide")
+    st.title("Generador de Imágenes")
+    st.sidebar.header("Inicio de Sesión")
+    input_username = st.sidebar.text_input("Usuario")
+    input_password = st.sidebar.text_input("Contraseña", type="password")
+
+    if st.sidebar.button("Iniciar Sesión"):
+        if input_username == username and input_password == password:
+            st.sidebar.success("Inicio de sesión exitoso!")
+            prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=500)
+            width, height = (720, 1280) if st.sidebar.selectbox("Formato", ["9:16", "16:9"]) == "9:16" else (1280, 720)
 
-    if st.sidebar.button("Generar Imagen"):
-        with st.spinner("Generando imagen..."):
-            result = asyncio.run(gen(prompt, width, height))
-            if (image_paths := result[0]) and Path(image_paths).exists(): st.image(image_paths, caption="Imagen Generada")
-            else: st.error("El archivo de imagen no existe.")
-            if (prompt_file := result[1]) and Path(prompt_file).exists():
-                st.write(f"Prompt utilizado: {Path(prompt_file).read_text()}")
-            else: st.write("El archivo del prompt no está disponible.")
+            if st.sidebar.button("Generar Imagen"):
+                with st.spinner("Generando imagen..."):
+                    result = asyncio.run(gen(prompt, width, height))
+                    if (image_paths := result[0]) and Path(image_paths).exists():
+                        st.image(image_paths, caption="Imagen Generada")
+                    else:
+                        st.error("El archivo de imagen no existe.")
+                    if (prompt_file := result[1]) and Path(prompt_file).exists():
+                        st.write(f"Prompt utilizado: {Path(prompt_file).read_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)
-            st.image(image, caption=f"Imagen {idx+1}"); st.write(f"Prompt: {Path(prompt_file).read_text() if prompt_file else 'No disponible'}")
-            if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
-                try: os.remove(file); os.remove(prompt_file) if prompt_file else None; 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}")
+            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)
+                    st.image(image, caption=f"Imagen {idx+1}")
+                    st.write(f"Prompt: {Path(prompt_file).read_text() if prompt_file else 'No disponible'}")
+                    if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
+                        try:
+                            os.remove(file)
+                            os.remove(prompt_file) if prompt_file else None
+                            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}")
+        else:
+            st.sidebar.error("Usuario o contraseña incorrectos.")
+    else:
+        st.sidebar.warning("Por favor, inicie sesión.")
 
 if __name__ == "__main__": main()