Update app.py

app.py

@@ -226,6 +226,17 @@ def page_1():
                 st.session_state.text_collection = text_collection
                 st.session_state.video_collection = video_collection
 
+                import shutil
+                # Path to the folder you want to download
+                folder_path = "mm_vdb2"
+                zip_path = "mm_vdb2.zip"
+                
+                # Compress the folder
+                shutil.make_archive(base_name="mm_vdb2", format="zip", root_dir=folder_path)
+                with open(zip_path, "rb") as file:
+                    st.download_button(label="Download mm_vdb2.zip", data=file, file_name="mm_vdb2.zip")
+
+
                 # Simulate a delay for finalizing (if needed)
                 time.sleep(1)  # Simulate final step