Update app.py

app.py

@@ -5,8 +5,10 @@ import os
 import urllib
 import base64
 
+EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md']
+
 def download_file(url, local_filename):
-    if url.startswith('http://') or url.startswith('https://'):  # add this line
+    if url.startswith('http://') or url.startswith('https://'):
         try:
             with requests.get(url, stream=True) as r:
                 r.raise_for_status()
@@ -15,8 +17,7 @@ def download_file(url, local_filename):
                         f.write(chunk)
             return local_filename
         except requests.exceptions.HTTPError as err:
-            print(f"HTTP error occurred: {err}")  # or use logging
-
+            print(f"HTTP error occurred: {err}")
 
 def download_html_and_files(url):
     html_content = requests.get(url).text
@@ -27,25 +28,16 @@ def download_html_and_files(url):
     for link in soup.find_all('a'):
         file_url = urllib.parse.urljoin(base_url, link.get('href'))
         local_filename = urllib.parse.urlparse(file_url).path.split('/')[-1]
-        if local_filename:  # add this line
+        if local_filename:
             link['href'] = local_filename
             download_file(file_url, local_filename)
     
     with open("index.html", "w") as file:
         file.write(str(soup))
 
-        
 def list_files(directory_path='.'):
-    return [f for f in os.listdir(directory_path) if os.path.isfile(os.path.join(directory_path, f))]
-
-def main():
-    st.sidebar.title('Bulk Download Tool')
-    url = st.sidebar.text_input('Please enter a URL to bulk download text and files')
-    if st.sidebar.button('📥 Get All the Content'):
-        download_html_and_files(url)
-        st.sidebar.write('Download complete. Here are the files you can download:')
-        for file in list_files():
-            st.sidebar.markdown(get_download_link(file), unsafe_allow_html=True)
+    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 get_download_link(file):
     with open(file, "rb") as f:
@@ -54,5 +46,19 @@ def get_download_link(file):
         href = f'<a href="data:file/octet-stream;base64,{b64}" download=\'{file}\'>Click to download {file}</a>'
     return href
 
+def show_download_links():
+    st.sidebar.write('Here are the files you can download:')
+    for file in list_files():
+        st.sidebar.markdown(get_download_link(file), unsafe_allow_html=True)
+
+def main():
+    st.sidebar.title('Bulk Download Tool')
+    url = st.sidebar.text_input('Please enter a URL to bulk download text and files')
+    if st.sidebar.button('📥 Get All the Content'):
+        download_html_and_files(url)
+        show_download_links()
+    if st.sidebar.button('📂 Show Download Links'):
+        show_download_links()
+
 if __name__ == "__main__":
     main()