Update app.py

app.py

@@ -44,7 +44,6 @@ def get_upscale_finegrain(prompt, img_path, upscale_factor):
         )
         return result[1] if isinstance(result, list) and len(result) > 1 else None
     except Exception as e:
-        st.error(f"Error en el escalado: {e}")
         return None
 
 async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora):
@@ -76,7 +75,7 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
             upscale_image = Image.open(upscale_image_path)
             upscale_image.save(DATA_PATH / f"upscale_image_{seed}.jpg", format="JPEG")
             progress_bar.progress(100)
-            image_path.unlink()  
+            image_path.unlink()
             return [str(DATA_PATH / f"upscale_image_{seed}.jpg"), str(prompt_file_path)]
         else:
             progress_bar.empty()
@@ -93,8 +92,7 @@ async def improve_prompt(prompt):
         improved_text = response['generated_text'].strip() if 'generated_text' in response else response.strip()
         return improved_text
     except Exception as e:
-        st.error(f"Error mejorando el prompt: {e}")
-        return ""
+        return f"Error mejorando el prompt: {e}"
 
 def get_storage():
     files = [{"name": str(file.resolve()), "size": file.stat().st_size,}
@@ -111,10 +109,20 @@ def run_gen():
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     prompt_to_use = st.session_state.get('improved_prompt', prompt)
-    result = loop.run_until_complete(gen(prompt_to_use, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora))
-    return result
+    return loop.run_until_complete(gen(prompt_to_use, basemodel, width, height, scales, steps, seed, upscale_factor, process_upscale, lora_model, process_lora))
+
+def delete_image(image_path):
+    try:
+        if Path(image_path).exists():
+            Path(image_path).unlink()
+            st.success(f"Imagen {image_path} borrada.")
+        else:
+            st.error("El archivo de imagen no existe.")
+    except Exception as e:
+        st.error(f"Error al borrar la imagen: {e}")
 
 st.set_page_config(layout="wide")
+st.title("Generador de Imágenes FLUX")
 prompt = st.sidebar.text_input("Descripción de la imagen")
 basemodel = st.sidebar.selectbox("Modelo Base", ["black-forest-labs/FLUX.1-schnell", "black-forest-labs/FLUX.1-DEV"])
 lora_model = st.sidebar.selectbox("LORA Realismo", ["Shakker-Labs/FLUX.1-dev-LoRA-add-details", "XLabs-AI/flux-RealismLora"])
@@ -141,12 +149,12 @@ if st.sidebar.button("Mejorar prompt"):
 
 if st.sidebar.button("Generar Imagen"):
     with st.spinner("Generando imagen..."):
-        result = run_gen()
-        image_paths = result[0]
+        result = run_gen()  
+        image_paths = result[0]  
         prompt_file = result[1]
     
-    st.write(f"Image paths: {image_paths}")
-
+    st.write(f"Image paths: {image_paths}")  
+    
     if image_paths:
         if Path(image_paths).exists():
             st.image(image_paths, caption="Imagen Generada")
@@ -161,13 +169,17 @@ if st.sidebar.button("Generar Imagen"):
 
 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}")
+        st.write(f"Prompt: {prompt_text}")
+        if st.button(f"Borrar Imagen {idx+1}", key=f"delete_{idx}"):
+            os.remove(file)
+            if prompt_file:
+                os.remove(prompt_file)