Update app.py

app.py

@@ -51,7 +51,7 @@ def save_webpage_as_zip(url):
         asset_full_path = os.path.join(temp_dir, asset_path)
 
         # Skip if asset_path is a directory
-        if os.path.isdir(asset_full_path):
+        if asset_path.endswith('/'):
             print(f"Skipping directory {asset_full_path}")
             continue
 
@@ -84,7 +84,10 @@ def save_webpage_as_zip(url):
 def generate_zip_file(url):
     """Generate ZIP file from a webpage URL."""
     zip_buffer = save_webpage_as_zip(url)
-    return zip_buffer, 'webpage.zip'
+    temp_zip_path = "webpage.zip"
+    with open(temp_zip_path, 'wb') as f:
+        f.write(zip_buffer.read())
+    return temp_zip_path
 
 # Gradio Interface
 with gr.Blocks() as demo: