Update page/reduce.py

page/reduce.py

@@ -23,7 +23,9 @@ def app():
 
     st.title('AI-Generated Architecture')
 
-    st.subheader('Choose which images you would like to remove from your working set.')
+    st.subheader('Download your images or choose which images you would like to remove from your working set.')
+
+
 
     d = datetime.datetime.now()
 
@@ -34,7 +36,7 @@ def app():
     for ix,result in enumerate( st.session_state.results ):
 
         with st.container():
-            col1,col2 = st.columns(2)
+            cols1,col2 = st.columns(2)
 
             with col1:
                 t = st.image(result['image'])
@@ -49,13 +51,15 @@ def app():
 
 
             with col2:
-                st.button("delete ", key=ix, on_click=dell, kwargs=dict(ix=ix) )
+                if(len(st.session_state.results) > 1):
+                    st.button("delete ", key=ix, on_click=dell, kwargs=dict(ix=ix) )
 
             m = st.markdown("""
             <hr />""", unsafe_allow_html=True)
 
     zipObj.close()
 
+
     st.download_button(
          label="Download images as zip",
          data=open('ai_architecture.zip', 'rb'),
@@ -63,3 +67,5 @@ def app():
          mime='application/zip'
     )
 
+    
+