Update page/reduce.py

page/reduce.py

@@ -12,6 +12,7 @@ import requests
 from io import BytesIO
 from PIL import Image, ImageDraw
 import base64
+import datetime
 
 def dell(ix):
     print("!!!!")
@@ -24,6 +25,8 @@ def app():
 
     st.subheader('Choose which images you would like to remove from your working set.')
 
+    d = datetime.datetime.now()
+
     zipObj = zipfile.ZipFile('ai_architecture.zip', 'w')
 
     deleteButtons = []
@@ -40,7 +43,7 @@ def app():
                     result['image'].save(output, format="JPEG")
                     contents = output.getvalue()
 
-                    txt = result['prompt']+" (temperature:"+ str(result['crazy']) + ", top k:" + str(result['k']) + ")"
+                    txt = str(ix+1)+") " + result['prompt']+" (temperature:"+ str(result['crazy']) + ", top k:" + str(result['k']) + ")"
         
                     zipObj.writestr(txt+".jpg",  contents )
 
@@ -56,7 +59,7 @@ def app():
     st.download_button(
          label="Download images as zip",
          data=open('ai_architecture.zip', 'rb'),
-         file_name='ai_architecture.zip',
+         file_name='ai_architecture '+d.strftime("%m-%d-%Y")+'.zip',
          mime='application/zip'
     )