Update app.py

app.py

@@ -4,6 +4,7 @@ from bs4 import BeautifulSoup
 from urllib.parse import urljoin, urlparse
 from zipfile import ZipFile
 from io import BytesIO
+import gradio as gr
 
 def download_file(url, session):
     """Download a file and return its content."""
@@ -11,7 +12,7 @@ def download_file(url, session):
     response.raise_for_status()
     return response.content
 
-def save_webpage_as_zip(url, output_zip):
+def save_webpage_as_zip(url):
     """Save a webpage and its assets as a ZIP file."""
     session = requests.Session()
     response = session.get(url)
@@ -53,8 +54,9 @@ def save_webpage_as_zip(url, output_zip):
         with open(asset_full_path, 'wb') as f:
             f.write(content)
 
-    # Create a ZIP file
-    with ZipFile(output_zip, 'w') as zipf:
+    # Create a ZIP file in memory
+    zip_buffer = BytesIO()
+    with ZipFile(zip_buffer, 'w') as zipf:
         for root, _, files in os.walk(temp_dir):
             for file in files:
                 file_path = os.path.join(root, file)
@@ -66,7 +68,20 @@ def save_webpage_as_zip(url, output_zip):
             os.remove(os.path.join(root, file))
         os.rmdir(root)
 
-# Example usage
-url = 'http://example.com'
-output_zip = 'webpage.zip'
-save_webpage_as_zip(url, output_zip)
+    zip_buffer.seek(0)
+    return zip_buffer
+
+def generate_zip_file(url):
+    """Generate ZIP file from a webpage URL."""
+    zip_buffer = save_webpage_as_zip(url)
+    return zip_buffer, 'webpage.zip'
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    url_input = gr.Textbox(label="Website URL")
+    download_button = gr.Button("Download as ZIP")
+    output_file = gr.File(label="Download")
+
+    download_button.click(fn=generate_zip_file, inputs=url_input, outputs=output_file)
+
+demo.launch()