Update app.py

app.py

@@ -12,20 +12,17 @@ import zipfile
 from PIL import Image
 
 EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md','.gitattributes', "backup.py","Dockerfile"]
-# Emoji mapping for different file types
 FILE_EMOJIS = {
     ".jpg": "🖼️", ".jpeg": "🖼️", ".png": "🖼️",
     ".html": "🌐", ".htm": "🌐", ".txt": "📄",
     ".pdf": "📚", ".doc": "📝", ".docx": "📝",
     ".xls": "📊", ".xlsx": "📊", ".ppt": "📊", ".pptx": "📊",
-    # Add more mappings as needed
 }
 URLS = {
     "Lumiere": "https://lumiere-video.github.io/",
     "ChatDev": "https://github.com/OpenBMB/ChatDev",
     "Autogen": "https://microsoft.github.io/autogen/",
     "Mixtral": "https://github.com/open-compass/MixtralKit",
-    
     "World Health Organization": "https://www.who.int/",
     "CMS - Centers for Medicare & Medicaid Services": "https://www.cms.gov/",
     "Mayo Clinic": "https://www.mayoclinic.org/",
@@ -105,7 +102,6 @@ def download_html_and_files(url, subdir):
         file.write(str(soup))
     return file_urls
 
-
 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]
@@ -147,8 +143,6 @@ def show_file_operations(file_path, sequence_number):
             os.remove(file_path)
             st.markdown(f"File deleted!")
 
-file_sequence_numbers = {}
-
 def show_file_content(file_path, original_url):
     _, file_extension = os.path.splitext(file_path)
     emoji = FILE_EMOJIS.get(file_extension, "📁")  # Default emoji for unknown file types
@@ -167,6 +161,7 @@ def show_file_content(file_path, original_url):
     except Exception as e:
         st.error(f"Error reading file {file_path}: {e}")
 
+file_sequence_numbers = {}
 def show_download_links(subdir):
     global file_sequence_numbers
     for file in list_files(subdir):
@@ -218,19 +213,14 @@ def main():
         else:
             selected_site = st.sidebar.selectbox("Select a Website", list(URLS.keys()), index=0)
             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:
             try:
                 history = json.load(f)
             except:
                 print('error')
-                
-        # Handling URL submission
         if url:
             subdir = hashlib.md5(url.encode()).hexdigest()
             if not os.path.exists(subdir):
@@ -264,11 +254,13 @@ def main():
                         st.write(f"Deleted: {file_path}")
                     os.rmdir(subdir)  # Remove the empty directory
             st.experimental_rerun()
+
         if st.sidebar.button("⬇️ Download All", help="Download all files in a zip"):
             start_directory = '.'  # Current directory
             for zip_file in zip_subdirs(start_directory):
                 st.sidebar.markdown(zip_file, unsafe_allow_html=True)
                 st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
+
         with st.expander("URL History and Downloaded Files"):
             try:
                 for url, subdir in history.items():