BrightData_SerpAPI_LinkedIn_Profile_Scraping

Running

App Files Files Community

ElegantSolutions commited on Jun 17

Commit

24cf5cc

verified ·

1 Parent(s): c464196

Update app.py

Browse files

Files changed (1) hide show

app.py +17 -15

app.py CHANGED Viewed

@@ -11,25 +11,26 @@ 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,
@@ -37,18 +38,19 @@ def zip_and_auto_download(file_bytes, inner_filename, zip_prefix="Download"):
             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)

 import base64
 import zipfile
 from datetime import datetime
+import streamlit.components.v1 as components
 # -----------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 provides:
+    - Fallback button
+    - Guaranteed auto-download using components.html
     """
     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)
     with open(zip_file_path, "rb") as f:
         zip_bytes = f.read()
+        # Visible fallback
         st.download_button(
             label=f"📥 Download {zip_filename}",
             data=zip_bytes,
             mime="application/zip"
         )
+        # Auto download using components.html
         b64 = base64.b64encode(zip_bytes).decode()
+        html = f"""
+            <html>
+                <body>
+                    <a id="auto-download-link" href="data:application/zip;base64,{b64}" download="{zip_filename}"></a>
+                    <script>
+                        document.getElementById('auto-download-link').click();
+                    </script>
+                </body>
+            </html>
+        """
+        components.html(html, height=0, width=0)
     os.remove(zip_file_path)