Update app.py

app.py

@@ -40,7 +40,11 @@ def authenticate_user(username, password):
 
 def list_saved_images():
     try:
-        image_files = sorted(DATA_PATH.glob("*.jpg"))
+        image_files = sorted(
+            DATA_PATH.glob("*.jpg"),
+            key=lambda x: x.stat().st_mtime,
+            reverse=True
+        )
         return image_files
     except Exception as e:
         st.error(f"Error al listar imágenes guardadas: {e}")
@@ -76,7 +80,13 @@ def generate_image(prompt, width, height, seed, model_name):
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
     try:
-        image = 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,
+            seed=seed  
+        )
         return image, seed
     except Exception as e:
         st.error(f"Error al generar imagen: {e}")
@@ -139,19 +149,25 @@ def gen(prompts, width, height, model_name, num_variants=1):
     
     while len(seeds) < num_variants:
         new_seed = random.randint(0, MAX_SEED)
-        if new_seed not in seeds: 
+        if new_seed not in seeds:
             seeds.append(new_seed)
-            
+    
     try:
-        for idx, (prompt, seed) in enumerate(zip(prompts[:num_variants], seeds)):
-            image, _ = generate_image(prompt, width, height, seed, model_name)
-            image_path = save_image(image, f"generated_image_{seed}.jpg")
-            if image_path:
-                save_prompt(f"generated_image_{seed}.jpg: {prompt}")
-                st.success(f"Imagen {idx + 1} generada")
-                images.append(str(image_path))
+        for idx, (seed, prompt) in enumerate(zip(seeds, prompts[:num_variants])):
+            image, used_seed = generate_image(prompt, width, height, seed, model_name)
+            if image:
+                image_path = save_image(image, f"generated_image_{used_seed}.jpg")
+                if image_path:
+                    if ": " in prompt:
+                        clean_prompt = prompt.split(": ")[-1]
+                    else:
+                        clean_prompt = prompt
+                    save_prompt(f"generated_image_{used_seed}.jpg: {clean_prompt}")
+                    st.success(f"Imagen {idx + 1} generada")
+                    images.append(str(image_path))
     except Exception as e:
         st.error(f"Error al generar imágenes: {e}")
+    
     return images
 
 def get_prompt_for_image(image_name):
@@ -160,7 +176,12 @@ def get_prompt_for_image(image_name):
         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()
+                    full_prompt = line.split(": ", 1)[1].strip()
+                    if ":" in full_prompt:
+                        prompt = full_prompt.split(": ")[-1].strip()
+                    else:
+                        prompt = full_prompt
+                    prompts[image_name] = prompt
     except FileNotFoundError:
         return "No hay prompt asociado."
     return prompts.get(image_name, "No hay prompt asociado.")