Update app.py

app.py

@@ -636,6 +636,9 @@ def main():
         st.session_state.current_url = None
         st.session_state.google_creds = None
         st.session_state.selected_files = []
+        st.session_state.do_deep_search = False  # Add this
+        st.session_state.deep_search_url = None  # Add this
+        st.session_state.search_results = []     # Add this
 
     st.title("Advanced File Downloader")
     
@@ -874,6 +877,35 @@ def main():
         st.header("Bing Search Mode")
         query = st.text_input("Enter search query", key="search_query_input")
         num_results = st.slider("Number of results", 1, 50, 5, key="num_results_slider")
+        
+        # Check if deep search was requested
+        if st.session_state.get('do_deep_search', False):
+            url_to_search = st.session_state.get('deep_search_url')
+            st.write(f"Running deep search on: {url_to_search}")
+            
+            async def perform_deep_search():
+                async with DownloadManager(
+                    use_proxy=use_proxy,
+                    proxy=proxy
+                ) as dm:
+                    files = await dm.deep_search(
+                        url=url_to_search,
+                        custom_ext_list=custom_extensions.split(',') if custom_extensions else [],
+                        sublink_limit=max_sublinks,
+                        timeout=sublink_timeout
+                    )
+                    if files:
+                        st.session_state.discovered_files = files
+                        st.session_state.current_url = url_to_search
+                        st.session_state.selected_files = []
+                    else:
+                        st.warning("No files found on this page.")
+                    
+                    # Clear the deep search flag after execution
+                    st.session_state.do_deep_search = False
+            
+            asyncio.run(perform_deep_search())
+            
         if st.button("Search", key="search_btn"):
             if query:
                 async def run_search():
@@ -886,26 +918,116 @@ def main():
                         with st.spinner("Searching..."):
                             urls = await dm.search_bing()
                             if urls:
+                                st.session_state.search_results = urls  # Store URLs in session state
                                 st.success(f"Found {len(urls)} results!")
                                 for i, url in enumerate(urls, 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,
-                                                custom_ext_list=custom_extensions.split(',') if custom_extensions else [],
-                                                sublink_limit=max_sublinks,
-                                                timeout=sublink_timeout
-                                            )
-                                            if files:
-                                                st.session_state.discovered_files = files
-                                                st.session_state.current_url = url
-                                                st.session_state.selected_files = []
-                                                safe_rerun()
-                                            else:
-                                                st.warning("No files found on this page.")
+                                            st.session_state.deep_search_url = url  # Store the URL to search
+                                            st.session_state.do_deep_search = True  # Flag to perform deep search
+                                            safe_rerun()  # Rerun to apply state change
                             else:
                                 st.warning("No search results found.")
                 asyncio.run(run_search())
+        
+        # Display search results if they exist
+        if hasattr(st.session_state, 'search_results') and st.session_state.search_results:
+            urls = st.session_state.search_results
+            st.success(f"Found {len(urls)} results!")
+            for i, url in enumerate(urls, 1):
+                with st.expander(f"Result {i}: {url}", expanded=(i == 1)):
+                    if st.button(f"Deep Search Result {i}", key=f"deep_search_result_saved_{i}"):
+                        st.session_state.deep_search_url = url
+                        st.session_state.do_deep_search = True
+                        safe_rerun()
+        
+        # If files were discovered in a previous search, show them
+        if st.session_state.discovered_files:
+            files = st.session_state.discovered_files
+            st.success(f"Found {len(files)} files on {st.session_state.current_url}!")
+            
+            # File selection and download UI
+            col1, col2 = st.columns([1, 4])
+            with col1:
+                if st.button("Select All", key="bing_select_all_btn"):
+                    st.session_state.selected_files = list(range(len(files)))
+                    safe_rerun()
+                if st.button("Clear Selection", key="bing_clear_selection_btn"):
+                    st.session_state.selected_files = []
+                    safe_rerun()
+            
+            selected_files = st.multiselect(
+                "Select files to download",
+                options=list(range(len(files))),
+                default=st.session_state.selected_files,
+                format_func=lambda x: f"{files[x]['filename']} ({files[x]['size']})",
+                key="bing_file_multiselect"
+            )
+            st.session_state.selected_files = selected_files
+            
+            if selected_files:
+                col1, col2, col3, col4 = st.columns(4)
+                with col1:
+                    download_dir = st.text_input("Download Directory", value="./downloads", key="bing_download_dir_input")
+                with col2:
+                    create_zip = st.checkbox("Create ZIP file", value=True, key="bing_create_zip_checkbox")
+                with col3:
+                    delete_after = st.checkbox("Delete after creating ZIP", key="bing_delete_after_checkbox")
+                with col4:
+                    upload_to_drive = st.checkbox("Upload to Google Drive", key="bing_upload_drive_checkbox")
+                
+                if st.button("Download Selected", key="bing_download_btn"):
+                    if not os.path.exists(download_dir):
+                        os.makedirs(download_dir)
+                    
+                    async def download_files():
+                        downloaded_paths = []
+                        progress_bar = st.progress(0)
+                        status_text = st.empty()
+                        async with DownloadManager(use_proxy=use_proxy, proxy=proxy) as dm:
+                            for i, idx in enumerate(selected_files):
+                                progress = (i + 1) / len(selected_files)
+                                file_info = files[idx]
+                                status_text.text(f"Downloading {file_info['filename']}... ({i+1}/{len(selected_files)})")
+                                progress_bar.progress(progress)
+                                path = await dm.download_file(file_info, download_dir, st.session_state.current_url)
+                                if path:
+                                    downloaded_paths.append(path)
+                            status_text.empty()
+                            progress_bar.empty()
+                            return downloaded_paths
+                    
+                    downloaded = asyncio.run(download_files())
+                    
+                    if downloaded:
+                        st.success(f"Successfully downloaded {len(downloaded)} files")
+                        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)
+                                    if not isinstance(drive_id, str) or not drive_id.startswith("Error"):
+                                        st.success(f"Uploaded to Google Drive. File ID: {drive_id}")
+                                    else:
+                                        st.error(drive_id)
+                            
+                            if delete_after:
+                                for path in downloaded:
+                                    try:
+                                        os.remove(path)
+                                    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))
     
     else:  # PDF Summarizer mode
         if summarizer is None:
@@ -931,7 +1053,6 @@ def main():
                         except Exception as e:
                             st.error(f"Error summarizing PDF: {e}")
 
-
 if __name__ == "__main__":
     try:
         main()