Update app.py

app.py

@@ -121,6 +121,23 @@ def create_video_from_images():
 def main():
     st.set_page_config(layout="wide")
     st.title("Generador de Imágenes")
+    
+    if "authenticated" not in st.session_state:
+        st.session_state.authenticated = False
+
+    if not st.session_state.authenticated:
+        st.subheader("Iniciar sesión")
+        username = st.text_input("Usuario")
+        password = st.text_input("Contraseña", type="password")
+
+        if st.button("Ingresar"):
+            if username == credentials["username"] and password == credentials["password"]:
+                st.session_state.authenticated = True
+                st.success("Inicio de sesión exitoso.")
+            else:
+                st.error("Usuario o contraseña incorrectos.")
+        return
+
     st.warning("Este espacio contiene contenido que no es adecuado para todas las audiencias. Se recomienda discreción.")
     agree = st.checkbox("Soy mayor de 18 años y entiendo que el contenido puede no ser apropiado.")
 
@@ -128,60 +145,60 @@ def main():
         prompt = st.sidebar.text_input("Descripción de la imagen", max_chars=500)
         format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9"])
         model_option = st.sidebar.selectbox("Modelo", ["enhanceaiteam/Flux-Uncensored-V2", "enhanceaiteam/Flux-uncensored"])
-        
+
         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, model_option))
                 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.")
-    
+
         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)  
+                        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 st.sidebar.button("Borrar todas las imágenes"):
             delete_all_images()
-    
+
         if st.sidebar.button("Descargar imágenes en .zip"):
             download_images_as_zip()
-    
+
         if st.button("Generar video con las imágenes"):
-                video_path = create_video_from_images()
-                if video_path:
-                    st.video(str(video_path), format="video/mp4")
-    
-if __name__ == "__main__": 
+            video_path = create_video_from_images()
+            if video_path:
+                st.video(str(video_path), format="video/mp4")
+
+if __name__ == "__main__":
     main()