added download feature

app.py

@@ -183,6 +183,39 @@ with gr.Blocks(css=css) as demo:
                 output_text = gr.Textbox(label="Output Text", elem_id="output")
 
         submit_btn.click(run_example, [input_img, model_selector], [output_text])
+    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")
+
+    def create_zip(image, text, fname):
+        if not image or not text or not fname:
+            return None
+        
+        # Create a temporary directory
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Save image with same extension
+            img_ext = image.format.lower() if hasattr(image, 'format') else 'png'
+            img_path = os.path.join(temp_dir, f"{fname}.{img_ext}")
+            image.save(img_path)
+            
+            # Save text
+            txt_path = os.path.join(temp_dir, f"{fname}.txt")
+            with open(txt_path, 'w', encoding='utf-8') as f:
+                f.write(text)
+            
+            # Create zip file
+            zip_path = os.path.join(temp_dir, f"{fname}.zip")
+            with zipfile.ZipFile(zip_path, 'w') as zipf:
+                zipf.write(img_path, os.path.basename(img_path))
+                zipf.write(txt_path, os.path.basename(txt_path))
+            
+            return zip_path
+
+    download_btn.click(
+        create_zip,
+        inputs=[input_img, output_text, filename],
+        outputs=gr.File(label="Download")
+    )
 
 demo.queue(api_open=False)
 demo.launch(debug=True)