Update app.py

app.py

@@ -9,6 +9,48 @@ from difflib import SequenceMatcher
 import json
 from urllib.parse import quote_plus
 import base64
+import zipfile
+from datetime import datetime
+
+
+# -----------zip_and_auto_download helper function--------------
+def zip_and_auto_download(file_bytes, inner_filename, zip_prefix="Download"):
+    """
+    Wraps given bytes in a ZIP and auto-downloads it via base64 + fallback button.
+    """
+    # Create temp ZIP file
+    zip_filename = f"{zip_prefix}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip"
+    zip_file_path = tempfile.NamedTemporaryFile(delete=False, suffix=".zip").name
+
+    with zipfile.ZipFile(zip_file_path, 'w') as zipf:
+        zipf.writestr(inner_filename, file_bytes)
+
+    # Read ZIP bytes
+    with open(zip_file_path, "rb") as f:
+        zip_bytes = f.read()
+
+        # Fallback download button
+        st.download_button(
+            label=f"📥 Download {zip_filename}",
+            data=zip_bytes,
+            file_name=zip_filename,
+            mime="application/zip"
+        )
+
+        # Auto-download link
+        b64 = base64.b64encode(zip_bytes).decode()
+        href = f'<a href="data:application/zip;base64,{b64}" download="{zip_filename}" id="auto-download-link"></a>'
+        st.markdown(href, unsafe_allow_html=True)
+        st.markdown(
+            """
+            <script>
+            document.getElementById('auto-download-link').click();
+            </script>
+            """,
+            unsafe_allow_html=True
+        )
+
+    os.remove(zip_file_path)
 
 # -----------------------------------------------
 # 🔧 UTILITY FUNCTIONS
@@ -141,7 +183,7 @@ if uploaded_file and api_key:
     if output_file:
         with open(output_file, "rb") as f:
             csv_bytes = f.read()
-            
+        
             b64 = base64.b64encode(csv_bytes).decode()
             href = f'<a href="data:text/csv;base64,{b64}" download="updated_with_linkedin_links.csv" id="auto-download-link"></a>'
         
@@ -154,15 +196,14 @@ if uploaded_file and api_key:
                 """,
                 unsafe_allow_html=True
             )
-    
+        
             st.success("✅ Processing complete. Your file is downloading automatically! Or click below if it didn't start.")
-            # st.success("✅ Processing complete. Download the updated file below.")
             st.download_button(
                 label="📥 Download CSV with LinkedIn Links",
                 data=f,
                 file_name="updated_with_linkedin_links.csv",
                 mime="text/csv"
-            )
+            )            
             shutil.rmtree(os.path.dirname(output_file))  # Cleanup temp directory
 elif not api_key:
     st.warning("⚠️ Please enter your BrightData SERP API key to proceed.")