Update app.py

app.py

@@ -141,13 +141,16 @@ if st.sidebar.button("Mejorar prompt"):
 
 if st.sidebar.button("Generar Imagen"):
     with st.spinner("Generando imagen..."):
-        image_paths, prompt_file = run_gen()
+        result = run_gen()  
+        image_paths = result[0] 
+        prompt_file = result[1] 
     
     if image_paths:
-        st.image(image_paths[0], caption="Imagen Generada")
+        st.image(image_paths, caption="Imagen Generada")
         prompt_text = Path(prompt_file).read_text()
         st.write(f"Prompt utilizado: {prompt_text}")
 
+
 files, usage = get_storage()
 st.text(usage)