Update page/reduce.py

page/reduce.py

@@ -3,9 +3,15 @@ from numpy.core.defchararray import lower
 import streamlit as st
 import numpy as np
 import pandas as pd
-from zipfile import ZipFile
+import zipfile
 import io
 import os
+from streamlit.elements.image import image_to_url
+import gzip
+import requests
+from io import BytesIO
+from PIL import Image, ImageDraw
+import base64
 
 def dell(ix):
     print("!!!!")
@@ -18,27 +24,7 @@ def app():
 
     st.subheader('Choose which images you would like to remove from your working set.')
 
-    os.chdir(r"temp/")
-    all_files = os.listdir()
-    for f in all_files:
-        os.remove(f)
-
-    # create a ZipFile object
-    zipObj = ZipFile('ai_architecture.zip', 'w')
-    # Add multiple files to the zip
-    for ix,file in enumerate( st.session_state.results ):
-        file['image'].save("temp/"+str(ix)+".jpeg")
-        zipObj.write("temp/"+str(ix)+".jpeg")
-
-    zipObj.close()
-    
-    st.download_button(
-         label="Download images as zip",
-         data=open('ai_architecture.zip', 'rb'),
-         file_name='ai_architecture.zip',
-         mime='application/zip'
-     )
-
+    zipObj = zipfile.ZipFile('ai_architecture.zip', 'w')
 
     deleteButtons = []
 
@@ -48,11 +34,29 @@ def app():
             col1,col2 = st.columns(2)
 
             with col1:
-                st.image(result['image'])
+                t = st.image(result['image'])
+
+                with io.BytesIO() as output:
+                    result['image'].save(output, format="JPEG")
+                    contents = output.getvalue()
+
+                    txt = result['prompt']+" (temperature:"+ str(result['crazy']) + ", top k:" + str(result['k']) + ")"
+        
+                    zipObj.writestr(txt+".jpg",  contents )
+
+
             with col2:
                 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'),
+         file_name='ai_architecture.zip',
+         mime='application/zip'
+    )