Update app.py

app.py

@@ -68,18 +68,18 @@ def swap_faces(source_image, source_face_index, destination_image, destination_f
     result = swapper.get(destination_image, res_face, source_face, paste_back=True)
     return result
 
-async def generate_image(prompt, width, height, seed, model_name):
+def generate_image(prompt, width, height, seed, model_name):
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
-    image = await client.text_to_image(prompt=prompt, height=height, width=width, model=model_name)
+    image = client.text_to_image(prompt=prompt, height=height, width=width, model=model_name)
     return image, seed
 
-async def gen(prompts, width, height, model_name, num_variants=1, use_enhanced=True):
+def gen(prompts, width, height, model_name, num_variants=1, use_enhanced=True):
     images = []
     try:
         for idx, prompt in enumerate(prompts):
             seed = random.randint(0, MAX_SEED)
-            image, seed = await generate_image(prompt, width, height, seed, model_name)
+            image, seed = generate_image(prompt, width, height, seed, model_name)
             image_path = save_image(image, f"generated_image_{seed}.jpg")
             if image_path:
                 st.success(f"Imagen {idx + 1} generada")
@@ -88,43 +88,7 @@ async def gen(prompts, width, height, model_name, num_variants=1, use_enhanced=T
         st.error(f"Error al generar imágenes: {e}")
     return images
 
-def list_saved_images():
-    return list(DATA_PATH.glob("*.jpg"))
-
-def display_gallery():
-    st.header("Galería de Imágenes Guardadas")
-    images = list_saved_images()
-    if images:
-        cols = st.columns(8)  
-        for i, image_file in enumerate(images):
-            with cols[i % 8]:  
-                st.image(str(image_file), caption=image_file.name, use_column_width=True)
-                prompt = get_prompt_for_image(image_file.name)  
-                st.write(prompt[:300])  
-
-                if st.button(f"FaceSwap", key=f"select_{i}_{image_file.name}"):
-                    st.session_state['generated_image_path'] = str(image_file)
-                    st.success("Imagen seleccionada")
-
-                if st.button(f"Borrar", key=f"delete_{i}_{image_file.name}"):
-                    if os.path.exists(image_file):
-                        os.remove(image_file)
-                        st.success("Imagen borrada")
-                        display_gallery()
-                    else:
-                        st.warning("La imagen no existe.")
-    else:
-        st.info("No hay imágenes guardadas.")
-
-def save_prompt(prompt):
-    with open(DATA_PATH / "prompts.txt", "a") as f:
-        f.write(prompt + "\n")
-        st.success("Prompt guardado.")
-
-def run_async(func, *args):
-    return asyncio.run(func(*args))
-
-async def improve_prompt(prompt):
+def improve_prompt(prompt):
     try:
         instructions = [
             "With my idea create a vibrant description for a detailed txt2img prompt, 300 characters max.",
@@ -133,8 +97,7 @@ async def improve_prompt(prompt):
             "With my idea describe a photorealistic with illumination txt2img prompt in English, 300 characters max.",
             "With my idea give a realistic and elegant txt2img prompt in English, 300 characters max.",
             "With my idea conform a visually dynamic and surreal txt2img prompt in English, 300 characters max.",
-            "With my idea realize an artistic and cinematic txt2img prompt in English, 300 characters max.",
-            "With my idea make a narrative and immersive txt2img prompt in English, 300 characters max."
+            "With my idea realize an artistic and cinematic txt2img prompt in English, 300 characters max."
         ]
         instruction = random.choice(instructions)
         formatted_prompt = f"{prompt}: {instruction}"
@@ -143,58 +106,35 @@ async def improve_prompt(prompt):
     except Exception as e:
         return f"Error mejorando el prompt: {e}"
 
-async def generate_variations(prompt, num_variants, use_enhanced):
+def generate_variations(prompt, num_variants, use_enhanced):
     prompts = set() 
     while len(prompts) < num_variants:
         if use_enhanced:
-            enhanced_prompt = await improve_prompt(prompt)
+            enhanced_prompt = improve_prompt(prompt)
             prompts.add(enhanced_prompt)
         else:
             prompts.add(prompt)
     return list(prompts)
 
-def get_prompt_for_image(image_name):
-    prompts = {}
+def delete_all_images():
     try:
-        with open(DATA_PATH / "prompts.txt", "r") as f:
-            for line in f:
-                if line.startswith(image_name):
-                    prompts[image_name] = line.split(": ", 1)[1].strip()
-    except FileNotFoundError:
-        return "No hay prompt asociado."
-    
-    return prompts.get(image_name, "No hay prompt asociado.")
-
-def login_form():
-    st.title("Iniciar Sesión")
-    username = st.text_input("Usuario", value="admin")
-    password = st.text_input("Contraseña", value="flux3x", type="password") 
-    if st.button("Iniciar Sesión"):
-        if authenticate_user(username, password):
-            st.success("Autenticación exitosa.")
-            st.session_state['authenticated'] = True
-        else:
-            st.error("Credenciales incorrectas. Intenta de nuevo.")
-
-def save_image(image, filename):
-    try:
-        image_path = DATA_PATH / filename
-        image.save(image_path)
-        return image_path
+        for image_file in list_saved_images():
+            os.remove(image_file)
+        st.success("Todas las imágenes han sido eliminadas.")
     except Exception as e:
-        st.error(f"Error al guardar la imagen: {e}")
-        return None
-
-def upload_image_to_gallery():
-    uploaded_image = st.sidebar.file_uploader("Sube una imagen a la galería", type=["jpg", "jpeg", "png"])
-    if uploaded_image:
-        image = Image.open(uploaded_image)
-        image_path = save_image(image, f"{uploaded_image.name}")
-        if image_path:
-            save_prompt("uploaded by user")
-            st.sidebar.success(f"Imagen subida: {image_path}")
+        st.error(f"Error al eliminar las imágenes: {e}")
+
+def upload_images_to_gallery():
+    uploaded_images = st.sidebar.file_uploader("Sube imágenes a la galería", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
+    if uploaded_images:
+        for uploaded_image in uploaded_images:
+            image = Image.open(uploaded_image)
+            image_path = save_image(image, f"{uploaded_image.name}")
+            if image_path:
+                save_prompt("uploaded by user")
+                st.sidebar.success(f"Imagen subida: {image_path}")
 
-async def main():
+def main():
     st.set_page_config(layout="wide")
 
     if 'authenticated' not in st.session_state or not st.session_state['authenticated']:
@@ -202,29 +142,33 @@ async def main():
         return
 
     st.title("Flux +Upscale +Prompt Enhancer +FaceSwap")
+    
+    if st.sidebar.button("Borrar todas las imágenes"):
+        delete_all_images()
+
     generated_image_path = st.session_state.get('generated_image_path')
-    st.header("Generador de Imágenes")
     prompt = st.sidebar.text_area("Descripción de la imagen", height=150, max_chars=500)
     format_option = st.sidebar.selectbox("Formato", ["9:16", "16:9"])
     model_option = st.sidebar.selectbox("Modelo", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-dev", "enhanceaiteam/Flux-Uncensored-V2", "enhanceaiteam/Flux-Uncensored"])
     prompt_checkbox = st.sidebar.checkbox("Prompt Enhancer")
     upscale_checkbox = st.sidebar.checkbox("Escalar imagen")
-    width, height = (360, 640) if format_option == "9:16" else (640, 360) if format_option == "16:9" else (640, 640)
     
+    width, height = (360, 640) if format_option == "9:16" else (640, 360) if format_option == "16:9" else (640, 640)
     num_variants = st.sidebar.slider("Número de imágenes a generar", 1, 8, 1) if prompt_checkbox else 1
     
     if prompt_checkbox:
         with st.spinner("Generando prompts mejorados..."):
-            prompts = await generate_variations(prompt, num_variants, True)
+            prompts = generate_variations(prompt, num_variants, True)
     else:
         prompts = [prompt] 
 
-    upload_image_to_gallery()
+    # Subida múltiple de imágenes
+    upload_images_to_gallery()
 
     if st.sidebar.button("Generar Imágenes"):
         with st.spinner("Generando imágenes..."):
             try:
-                results = await gen(prompts, width, height, model_option, num_variants, prompt_checkbox)
+                results = gen(prompts, width, height, model_option, num_variants, prompt_checkbox)
                 st.session_state['generated_image_paths'] = results
                 for result in results:
                     st.image(result, caption="Imagen Generada")
@@ -274,4 +218,4 @@ async def main():
     display_gallery()
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    main()