Update app.py

app.py

@@ -6,6 +6,8 @@ from io import BytesIO
 from PIL import Image
 import hashlib
 import concurrent.futures
+import zipfile
+import tempfile
 
 # Constants
 FREE_URL = os.environ.get("SERVER_URL_FREE")
@@ -135,11 +137,35 @@ def search_face(file, token=None):
     soc_res = process_response2(response2) if response2 else None
     return process_response(response1, soc_res, url, token)
 
+def export_images(items):
+    if not items:
+        return None
+    # Create a zip file in memory
+    zip_buffer = BytesIO()
+    with zipfile.ZipFile(zip_buffer, 'w') as zip_file:
+        url_text = ""
+        for i, item in enumerate(items):
+            with open(item[0], 'rb') as img_file:
+                zip_file.writestr(f'image_{i}.jpg', img_file.read())
+            url_text += f"image_{i}.jpg: {item[1]}\n"
+        zip_file.writestr("urls.txt", url_text)
+    zip_buffer.seek(0)
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".zip") as temp_file:
+        temp_file.write(zip_buffer.getvalue())
+        temp_file_path = temp_file.name
+
+    return temp_file_path
+
 custom_css = """
 caption.caption {
     user-select: text;
     cursor: text;
 }
+
+div#component-16 {
+    max-height: 63.39px;
+}
 """
 
 js = """
@@ -181,6 +207,7 @@ head = """
 </script>
 """
 
+output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto")
 with gr.Blocks(css=custom_css, head=head) as demo:
     gr.Markdown(
         """
@@ -196,12 +223,18 @@ with gr.Blocks(css=custom_css, head=head) as demo:
             gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676c15b1971244a587ca07cb' target='_blank'>Get Premium Token: Deep Search Including Social Media & Full URL Reveal</a>")            
             gr.HTML("<a href='https://faceonlive.pocketsflow.com/checkout?productId=676d7e7597f8b3b699f84820' target='_blank'>Protect Your Privacy – Start Your Takedown Now</a>")
             search_face_button = gr.Button("Search Face")
+            gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, cache_mode='lazy', fn=search_face, outputs=[output])
+            
         with gr.Column(scale=2):
-            output = gr.Gallery(label="Found Images", columns=[4], object_fit="contain", height="auto")
-    
-    gr.Examples(['examples/1.jpg', 'examples/2.jpg'], inputs=image, cache_examples=True, cache_mode='lazy', fn=search_face, outputs=[output])
+            gr.Markdown("> ### **⚠️ Reminder:** Export images before refreshing the page by clicking the 'Export Images' button.")
+            output.render()
+            with gr.Row():
+                export_button = gr.Button("Export Images")
+                export_file = gr.File(label="Download")
+        
 
     search_face_button.click(search_face, inputs=[image, token], outputs=[output], api_name=False)
+    export_button.click(export_images, inputs=[output], outputs=export_file, api_name=False)
 
     gr.HTML('<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online"><img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FFaceOnLive%2FFace-Search-Online&labelColor=%23ff8a65&countColor=%2337d67a&style=flat&labelStyle=upper" /></a>')
     html = gr.HTML()