Update app.py

app.py

@@ -615,12 +615,12 @@ def main():
                     
                     files = asyncio.run(run_deep_search())
                     if files:
-                        # Save all discovered files even if duplicated
+                        # Save all discovered files—even duplicates
                         st.session_state.discovered_files = files
                         st.session_state.current_url = url
                         st.success(f"Found {len(files)} files!")
                         
-                        # File selection block 1
+                        # File selection block (Select/Clear)
                         col1, col2 = st.columns([1, 4])
                         with col1:
                             if st.button("Select All", key="select_all_btn"):
@@ -664,7 +664,7 @@ def main():
                                             file_info = files[idx]
                                             status_text.text(f"Downloading {file_info['filename']}... ({i+1}/{len(selected_files)})")
                                             progress_bar.progress(progress)
-                                            # In DownloadManager.download_file(), consider adding a fallback to requests if needed.
+                                            # Download the file (ensure DownloadManager.download_file downloads duplicates)
                                             path = await dm.download_file(file_info, download_dir, url)
                                             if path:
                                                 downloaded_paths.append(path)
@@ -676,9 +676,14 @@ def main():
                                 
                                 if downloaded:
                                     st.success(f"Successfully downloaded {len(downloaded)} files")
-                                    if create_zip or upload_to_drive:
+                                    # If the user chose to create a ZIP, generate it and offer a download button
+                                    if create_zip:
                                         zip_path = create_zip_file(downloaded, download_dir)
                                         st.success(f"Created ZIP file: {zip_path}")
+                                        with open(zip_path, "rb") as f:
+                                            zip_data = f.read()
+                                        st.download_button("Download ZIP", data=zip_data, file_name=os.path.basename(zip_path), mime="application/zip")
+                                        
                                         if upload_to_drive and st.session_state.get('google_creds'):
                                             with st.spinner("Uploading to Google Drive..."):
                                                 drive_id = google_drive_upload(zip_path, st.session_state.google_creds)
@@ -693,10 +698,16 @@ def main():
                                                 except Exception as e:
                                                     st.warning(f"Could not delete {path}: {e}")
                                             st.info("Deleted original files after ZIP creation")
+                                    else:
+                                        # Otherwise, generate an individual download button for each file
+                                        for path in downloaded:
+                                            with open(path, "rb") as f:
+                                                file_data = f.read()
+                                            st.download_button(f"Download {os.path.basename(path)}", data=file_data, file_name=os.path.basename(path))
                     else:
                         st.warning("No files found.")
         
-        # File selection block 2 (if files have already been discovered)
+        # If files were discovered in a previous search, show them here as well.
         if st.session_state.discovered_files:
             files = st.session_state.discovered_files
             st.success(f"Found {len(files)} files!")
@@ -749,9 +760,12 @@ def main():
                     downloaded = asyncio.run(download_files())
                     if downloaded:
                         st.success(f"Successfully downloaded {len(downloaded)} files")
-                        if create_zip or upload_to_drive:
+                        if create_zip:
                             zip_path = create_zip_file(downloaded, download_dir)
                             st.success(f"Created ZIP file: {zip_path}")
+                            with open(zip_path, "rb") as f:
+                                zip_data = f.read()
+                            st.download_button("Download ZIP", data=zip_data, file_name=os.path.basename(zip_path), mime="application/zip")
                             if upload_to_drive and st.session_state.get('google_creds'):
                                 with st.spinner("Uploading to Google Drive..."):
                                     drive_id = google_drive_upload(zip_path, st.session_state.google_creds)
@@ -766,6 +780,11 @@ def main():
                                     except Exception as e:
                                         st.warning(f"Could not delete {path}: {e}")
                                 st.info("Deleted original files after ZIP creation")
+                        else:
+                            for path in downloaded:
+                                with open(path, "rb") as f:
+                                    file_data = f.read()
+                                st.download_button(f"Download {os.path.basename(path)}", data=file_data, file_name=os.path.basename(path))
     
     elif mode == "Bing Search":
         st.header("Bing Search Mode")
@@ -785,7 +804,7 @@ def main():
                             if urls:
                                 st.success(f"Found {len(urls)} results!")
                                 for i, url in enumerate(urls, 1):
-                                    with st.expander(f"Result {i}: {url}", expanded=i == 1):
+                                    with st.expander(f"Result {i}: {url}", expanded=(i == 1)):
                                         if st.button(f"Deep Search Result {i}", key=f"deep_search_result_{i}"):
                                             files = await dm.deep_search(
                                                 url=url,