Update app.py

app.py

@@ -48,17 +48,22 @@ 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_download_links(subdir):
+    st.write(f'Files for {subdir}:')
+    for file in list_files(subdir):
+        file_path = os.path.join(subdir, file)
+        if os.path.isfile(file_path):
+            st.markdown(get_download_link(file_path), unsafe_allow_html=True)
+        else:
+            st.write(f"File not found: {file}")
+
 def get_download_link(file):
     with open(file, "rb") as f:
         bytes = f.read()
         b64 = base64.b64encode(bytes).decode()
-        href = f'<a href="data:file/octet-stream;base64,{b64}" download=\'{file.split("/")[-1]}\'>Click to download {file.split("/")[-1]}</a>'
+        href = f'<a href="data:file/octet-stream;base64,{b64}" download=\'{os.path.basename(file)}\'>Click to download {os.path.basename(file)}</a>'
     return href
 
-def show_download_links(subdir):
-    st.write(f'Files for {subdir}:')
-    for file in list_files(subdir):
-        st.markdown(get_download_link(file), unsafe_allow_html=True)
 
 def main():
     st.sidebar.title('Web Datasets Bulk Downloader')