Update app.py

app.py

@@ -129,14 +129,21 @@ def gen(prompts, width, height, model_name, num_variants=1):
             images.append(str(image_path))
     return images
 
+def save_prompt(prompt):
+    clean_prompt = prompt.split(": ", 1)[-1]  
+    with open(DATA_PATH / "prompts.txt", "a") as f:
+        f.write(clean_prompt + "\n")  
+        st.success("Prompt guardado.")
+
 def get_prompt_for_image(image_name):
     prompts = {}
-    try:
-        with open(DATA_PATH / "prompts.txt", "r") as f:
+    prompt_file = DATA_PATH / "prompts.txt"
+    if prompt_file.exists():
+        with open(prompt_file, "r") as f:
             for line in f:
                 if line.startswith(image_name):
                     prompts[image_name] = line.split(": ", 1)[1].strip()
-    except FileNotFoundError:
+    else:
         return "No hay prompt asociado."
     return prompts.get(image_name, "No hay prompt asociado.")