Update app.py

app.py

@@ -7,6 +7,8 @@ from PIL import Image
 import imagehash
 import requests
 
+from zipfile import ZipFile
+
 from time import sleep
 sleep_time = 0.5
 
@@ -99,6 +101,17 @@ def search(prompt, search_name, im, func_name):
     return t2_list
     #return list(map(lambda x: x["name"], t2_list))
 
+def zip_ims(g):
+    from uuid import uuid1
+    if g is None:
+        return None
+    l = list(map(lambda x: x["name"], g))
+    zip_file_name ="tmp.zip"
+    with ZipFile(zip_file_name ,"w") as zipObj:
+        for ele in l:
+            zipObj.write(ele, "{}.png".format(uuid1()))
+        #zipObj.write(file2.name, "file2")
+    return zip_file_name
 
 with gr.Blocks(css="custom.css") as demo:
     title = gr.HTML(
@@ -119,6 +132,11 @@ with gr.Blocks(css="custom.css") as demo:
                 text_button = gr.Button("Retrieve Images", elem_id="run_button")
             i = gr.Image(elem_id="result-image", label = "Image upload or selected", height = 768 - 256 - 32)
 
+            with gr.Row():
+                with gr.Tab(label = "Download"):
+                    zip_button = gr.Button("Zip Images to Download", elem_id="zip_button")
+                    downloads = gr.File(label = "Image zipped", elem_id = "zip_file")
+
         with gr.Column():
             title = gr.Markdown(
                     value="### Click on a Image in the gallery to select it, and the grid order will change",
@@ -128,10 +146,11 @@ with gr.Blocks(css="custom.css") as demo:
             order_func_name = gr.Radio(choices=hash_func_name,
             value=hash_func_name[0], label="Order by", elem_id="order_radio")
 
-            outputs = gr.Gallery(lable='Output gallery', elem_id="gallery",).style(grid=5,height=768 - 256,
+            outputs = gr.Gallery(lable='Output gallery', elem_id="gallery",).style(grid=5,height=768 + 64 + 32,
                     allow_preview=False, label = "retrieve Images")
             #gr.Dataframe(label='prompts for corresponding images')]
 
+
     with gr.Row():
         gr.Examples(
             [
@@ -170,6 +189,8 @@ with gr.Blocks(css="custom.css") as demo:
     #text_button.click(enterpix, inputs=inputs, outputs=outputs)
     text_button.click(search, inputs=[inputs, search_func_name, i, order_func_name], outputs=outputs)
 
-
+    zip_button.click(
+        zip_ims, inputs = outputs, outputs=downloads
+    )
 
 demo.launch("0.0.0.0")