fixed download functions

app.py

@@ -10,6 +10,8 @@ import numpy as np
 import os
 from gliner import GLiNER
 import json
+import tempfile
+import zipfile
 
 # Initialize GLiNER model
 gliner_model = GLiNER.from_pretrained("knowledgator/modern-gliner-bi-large-v1.0")
@@ -251,6 +253,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         filename = gr.Textbox(label="Save filename (without extension)", placeholder="Enter filename to save")
         download_btn = gr.Button("Download Image & Text", elem_classes="submit-btn")
+        download_output = gr.File(label="Download")
 
     def create_zip(image, text_data, fname):
         if not isinstance(image, (Image.Image, np.ndarray)) or not text_data or not fname:
@@ -304,5 +307,12 @@ with gr.Blocks(css=css) as demo:
             
             return zip_path
 
+    # Add the click handler for the download button
+    download_btn.click(
+        create_zip,
+        inputs=[input_img, output_text, filename],
+        outputs=[download_output]
+    )
+
 demo.queue(api_open=False)
 demo.launch(debug=True)