Update backup.py

backup.py

@@ -6,12 +6,21 @@ import base64
 from bs4 import BeautifulSoup
 import hashlib
 import json
+import uuid 
 
 EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md','.gitattributes', "backup.py","Dockerfile"]
 URLS = {
     "Chordify - Play Along Chords": "https://chordify.net/",
     "National Guitar Academy - Guitar Learning": "https://www.guitaracademy.com/",
     "Ultimate Guitar - Massive Song Database": "https://www.ultimate-guitar.com/",
+    "Wolf Alice": "https://www.chordie.com/song.php/songartist/Wolf+Alice/index.html",
+    "Everclear": "https://www.chordie.com/song.php/songartist/Everclear/index.html",
+    "Jungle": "https://www.ultimate-guitar.com/artist/jungle_47745",
+    "Mylie Cyrus": "https://www.ultimate-guitar.com/search.php?title=mile+cyrus&spelling=Mylie+cyrus",
+    "Kanye": "https://www.ultimate-guitar.com/search.php?search_type=title&value=Kanye%20west",
+    "Cat Stevens": "https://www.ultimate-guitar.com/search.php?search_type=title&value=cat%20stevens",
+    "Metric": "https://www.ultimate-guitar.com/search.php?search_type=title&value=Metric",
+    "John Lennon": "https://www.ultimate-guitar.com/search.php?search_type=title&value=John%20Lennon",
 }
 
 if not os.path.exists("history.json"):
@@ -50,36 +59,67 @@ def list_files(directory_path='.'):
     files = [f for f in os.listdir(directory_path) if os.path.isfile(os.path.join(directory_path, f))]
     return [f for f in files if f not in EXCLUDED_FILES]
 
-def show_file_operations(file_path):
+def file_editor(file_path):
+    st.write(f"Editing File: {os.path.basename(file_path)}")
+    file_content = ""
+
+    with open(file_path, "r") as f:
+        file_content = f.read()
+
+    file_content = st.text_area("Edit the file content:", value=file_content, height=250)
+
+    if st.button("💾 Save"):
+        with open(file_path, "w") as f:
+            f.write(file_content)
+        st.success(f"File '{os.path.basename(file_path)}' saved!")
+
+
+def show_file_operations(file_path, sequence_number):
     st.write(f"File: {os.path.basename(file_path)}")
     unique_key = hashlib.md5(file_path.encode()).hexdigest()
+
+    file_content = ""
+
     col1, col2, col3 = st.columns(3)
 
     with col1:
-        if st.button(f"✏️ Edit", key=f"edit_{unique_key}"):
-            file_content = ""
+        edit_key = f"edit_{unique_key}_{sequence_number}"
+        if st.button(f"✏️ Edit", key=edit_key):
             with open(file_path, "r") as f:
                 file_content = f.read()
-            file_content = st.text_area("Edit the file content:", value=file_content, height=250, key=f"text_area_{unique_key}")
+            text_area_key = f"text_area_{unique_key}_{sequence_number}"
+            file_content = st.text_area("Edit the file content:", value=file_content, height=250, key=text_area_key)
 
     with col2:
-        if st.button(f"💾 Save", key=f"save_{unique_key}"):
-            with open(file_path, "w") as f:
-                f.write(file_content)
-            st.success(f"File saved!")
+        save_key = f"save_{unique_key}_{sequence_number}"
+        if st.button(f"💾 Save", key=save_key):
+            if file_content:  # Ensure file_content is not empty
+                with open(file_path, "w") as f:
+                    f.write(file_content)
+                st.success(f"File saved!")
 
     with col3:
-        if st.button(f"🗑️ Delete", key=f"delete_{unique_key}"):
+        delete_key = f"delete_{unique_key}_{sequence_number}"
+        if st.button(f"🗑️ Delete", key=delete_key):
             os.remove(file_path)
             st.markdown(f"File deleted!")
 
+
+file_sequence_numbers = {}
+
 def show_download_links(subdir):
-    st.write(f'Files for {subdir}:')
+    global file_sequence_numbers
     for file in list_files(subdir):
         file_path = os.path.join(subdir, file)
+        if file_path not in file_sequence_numbers:
+            file_sequence_numbers[file_path] = 1
+        else:
+            file_sequence_numbers[file_path] += 1
+        sequence_number = file_sequence_numbers[file_path]
+
         if os.path.isfile(file_path):
             st.markdown(get_download_link(file_path), unsafe_allow_html=True)
-            show_file_operations(file_path)
+            show_file_operations(file_path, sequence_number)
         else:
             st.write(f"File not found: {file}")
 
@@ -92,6 +132,8 @@ def get_download_link(file):
 
 def main():
     st.sidebar.title('Web Datasets Bulk Downloader')
+
+    # Selecting URL input method
     url_input_method = st.sidebar.radio("Choose URL Input Method", ["Enter URL", "Select from List"])
     url = ""
     if url_input_method == "Enter URL":
@@ -100,9 +142,15 @@ def main():
         selected_site = st.sidebar.selectbox("Select a Website", list(URLS.keys()))
         url = URLS[selected_site]
 
+    # Reading or creating history.json
+    if not os.path.exists("history.json"):
+        with open("history.json", "w") as f:
+            json.dump({}, f)
+
     with open("history.json", "r") as f:
         history = json.load(f)
 
+    # Handling URL submission
     if url:
         subdir = hashlib.md5(url.encode()).hexdigest()
         if not os.path.exists(subdir):
@@ -112,18 +160,22 @@ def main():
             with open("history.json", "w") as f:
                 json.dump(history, f)
 
+    # Button for downloading content
     if st.sidebar.button('📥 Get All the Content'):
         download_html_and_files(url, history[url])
         show_download_links(history[url])
 
+    # Button for showing download links
     if st.sidebar.button('📂 Show Download Links'):
         for subdir in history.values():
             show_download_links(subdir)
 
+    # Expander for showing URL history and download links
     with st.expander("URL History and Downloaded Files"):
         for url, subdir in history.items():
             st.markdown(f"#### {url}")
             show_download_links(subdir)
 
+            
 if __name__ == "__main__":
     main()