Update app.py

app.py

@@ -184,26 +184,12 @@ if uploaded_file and api_key:
         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>'
-        
-            st.markdown(href, unsafe_allow_html=True)
-            st.markdown(
-                """
-                <script>
-                document.getElementById('auto-download-link').click();
-                </script>
-                """,
-                unsafe_allow_html=True
+            zip_and_auto_download(
+                file_bytes=csv_bytes,
+                inner_filename="updated_with_linkedin_links.csv",
+                zip_prefix="LinkedIn_Links"
             )
         
-            st.success("✅ Processing complete. Your file is downloading automatically! Or click below if it didn't start.")
-            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
+            shutil.rmtree(os.path.dirname(output_file))
 elif not api_key:
     st.warning("⚠️ Please enter your BrightData SERP API key to proceed.")