Updated the save button

app.py

@@ -32,7 +32,7 @@ def load_css(file_path):
 
 
 # Load CSS file
-# load_css('styles2.css')
+load_css('styles2.css')
 
 
 # ----- FUNCTIONS ----
@@ -318,26 +318,23 @@ def main():
 
                 with col1:
                     saved_button = False
-                    if st.button("Save", key="b_save"):
-                        file_name = "output.txt"
-
-                        # Define the folder path
-                        folder_path = "extracted data"
-
-                        # Create the folder if it doesn't exist
-                        os.makedirs(folder_path, exist_ok=True)
-
-                        # Define the file path
-                        file_path = os.path.join(folder_path, file_name)
-                        if is_a_sitemap:
-                            saved_data = st.session_state.sitemap_data
-                            # Save string variable to file
-                            with open(file_path, "w", encoding="utf-8") as file:
-                                file.write(saved_data)
-                        else:
-                            with open(file_path, "w", encoding="utf-8") as file:
-                                file.write(data)
-                        saved_button = True
+
+                    if is_a_sitemap:
+                        saved_data = st.session_state.sitemap_data
+ 
+                        if st.download_button(
+                            label="Save",
+                            data=saved_data                            
+                        ):
+                            saved_button = True
+
+                        
+                    else:
+                        if st.download_button(
+                            label="Save",
+                            data=data                            
+                        ):
+                            saved_button = True
 
                 with col2:
                     if st.button("Clear"):