Update app.py

app.py

@@ -65,7 +65,6 @@ async def gen(prompt, basemodel, width, height, scales, steps, seed, upscale_fac
     image_path = DATA_PATH / f"image_{seed}.jpg"
     image.save(image_path, format="JPEG")
 
-    # Save the prompt used in a text file
     prompt_file_path = DATA_PATH / f"prompt_{seed}.txt"
     with open(prompt_file_path, "w") as prompt_file:
         prompt_file.write(combined_prompt)
@@ -142,14 +141,22 @@ if st.sidebar.button("Mejorar prompt"):
 if st.sidebar.button("Generar Imagen"):
     with st.spinner("Generando imagen..."):
         result = run_gen()  
-        image_paths = result[0] 
-        prompt_file = result[1] 
+        image_paths = result[0]  
+        prompt_file = result[1]
+    
+    st.write(f"Image paths: {image_paths}")  
     
     if image_paths:
-        st.image(image_paths, caption="Imagen Generada")
-        prompt_text = Path(prompt_file).read_text()
-        st.write(f"Prompt utilizado: {prompt_text}")
-
+        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)