Update app.py

app.py

@@ -604,13 +604,14 @@ def main():
     st.title("Advanced File Downloader")
     
     # Mode Selection
-    mode = st.radio("Select Mode", ["Manual URL", "Bing Search", "PDF Summarizer"])
+    mode = st.radio("Select Mode", ["Manual URL", "Bing Search", "PDF Summarizer"], key="mode_select")
     
     # Advanced Options
     with st.expander("Advanced Options"):
         custom_extensions = st.text_input(
             "Custom File Extensions",
-            placeholder=".csv, .txt, .epub"
+            placeholder=".csv, .txt, .epub",
+            key="custom_ext_input"
         )
         max_sublinks = st.number_input(
             "Maximum Sublinks to Process",
@@ -618,7 +619,8 @@ def main():
             max_value=10000,
             value=100,
             step=50,
-            help="Maximum number of sublinks to process from the main page"
+            help="Maximum number of sublinks to process from the main page",
+            key="max_sublinks_input"
         )
         sublink_timeout = st.number_input(
             "Search Timeout (seconds per sublink)",
@@ -626,37 +628,37 @@ def main():
             max_value=3000,
             value=30,
             step=5,
-            help="Maximum time to spend searching each sublink"
+            help="Maximum time to spend searching each sublink",
+            key="timeout_input"
         )
-        use_proxy = st.checkbox("Use Proxy")
-        proxy = st.text_input("Proxy URL", placeholder="http://proxy:port")
+        use_proxy = st.checkbox("Use Proxy", key="proxy_checkbox")
+        proxy = st.text_input("Proxy URL", placeholder="http://proxy:port", key="proxy_input")
 
     # Google Drive Integration
     with st.expander("Google Drive Integration"):
-        if st.button("Start Google Sign-In"):
+        if st.button("Start Google Sign-In", key="google_signin_btn"):
             auth_url = get_google_auth_url()
             st.markdown(f"[Click here to authorize]({auth_url})")
             
-        auth_code = st.text_input("Enter authorization code")
-        if st.button("Complete Sign-In") and auth_code:
+        auth_code = st.text_input("Enter authorization code", key="auth_code_input")
+        if st.button("Complete Sign-In", key="complete_signin_btn") and auth_code:
             creds, msg = exchange_code_for_credentials(auth_code)
             st.session_state.google_creds = creds
             st.write(msg)
 
-    url = st.text_input("Enter URL", placeholder="https://example.com")
-
     if mode == "Manual URL":
         st.header("Manual URL Mode")
-        url = st.text_input("Enter URL", placeholder="https://example.com")
+        url = st.text_input("Enter URL", placeholder="https://example.com", key="url_input")
         
-        if st.button("Deep Search", use_container_width=True):
+        if st.button("Deep Search", use_container_width=True, key="deep_search_btn"):
             if url:
                 async def run_deep_search():
                     async with DownloadManager(use_proxy=use_proxy, proxy=proxy) as dm:
                         files = await dm.deep_search(
                             url=url,
                             custom_ext_list=custom_extensions.split(',') if custom_extensions else [],
-                            sublink_limit=max_sublinks
+                            sublink_limit=max_sublinks,
+                            timeout=sublink_timeout
                         )
                         if files:
                             st.session_state.discovered_files = files
@@ -670,10 +672,10 @@ def main():
                     # Select All/Clear Selection buttons
                     col1, col2 = st.columns([1, 4])
                     with col1:
-                        if st.button("Select All"):
+                        if st.button("Select All", key="select_all_btn"):
                             st.session_state.selected_files = list(range(len(files)))
                             st.experimental_rerun()
-                        if st.button("Clear Selection"):
+                        if st.button("Clear Selection", key="clear_selection_btn"):
                             st.session_state.selected_files = []
                             st.experimental_rerun()
                     
@@ -682,7 +684,8 @@ def main():
                         "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']})"
+                        format_func=lambda x: f"{files[x]['filename']} ({files[x]['size']})",
+                        key="file_multiselect"
                     )
                     
                     # Update session state
@@ -691,15 +694,15 @@ def main():
                     if selected_files:
                         col1, col2, col3, col4 = st.columns(4)
                         with col1:
-                            download_dir = st.text_input("Download Directory", value="./downloads")
+                            download_dir = st.text_input("Download Directory", value="./downloads", key="download_dir_input")
                         with col2:
-                            create_zip = st.checkbox("Create ZIP file", value=True)
+                            create_zip = st.checkbox("Create ZIP file", value=True, key="create_zip_checkbox")
                         with col3:
-                            delete_after = st.checkbox("Delete after creating ZIP")
+                            delete_after = st.checkbox("Delete after creating ZIP", key="delete_after_checkbox")
                         with col4:
-                            upload_to_drive = st.checkbox("Upload to Google Drive")
+                            upload_to_drive = st.checkbox("Upload to Google Drive", key="upload_drive_checkbox")
                         
-                        if st.button("Download Selected"):
+                        if st.button("Download Selected", key="download_btn"):
                             if not os.path.exists(download_dir):
                                 os.makedirs(download_dir)
                                 
@@ -763,10 +766,10 @@ def main():
             # Select All/Clear Selection buttons
             col1, col2 = st.columns([1, 4])
             with col1:
-                if st.button("Select All"):
+                if st.button("Select All", key="select_all_btn2"):
                     st.session_state.selected_files = list(range(len(files)))
                     st.experimental_rerun()
-                if st.button("Clear Selection"):
+                if st.button("Clear Selection", key="clear_selection_btn2"):
                     st.session_state.selected_files = []
                     st.experimental_rerun()
             
@@ -775,7 +778,8 @@ def main():
                 "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']})"
+                format_func=lambda x: f"{files[x]['filename']} ({files[x]['size']})",
+                key="file_multiselect2"
             )
             
             # Update session state
@@ -784,15 +788,15 @@ def main():
             if selected_files:
                 col1, col2, col3, col4 = st.columns(4)
                 with col1:
-                    download_dir = st.text_input("Download Directory", value="./downloads")
+                    download_dir = st.text_input("Download Directory", value="./downloads", key="download_dir_input2")
                 with col2:
-                    create_zip = st.checkbox("Create ZIP file", value=True)
+                    create_zip = st.checkbox("Create ZIP file", value=True, key="create_zip_checkbox2")
                 with col3:
-                    delete_after = st.checkbox("Delete after creating ZIP")
+                    delete_after = st.checkbox("Delete after creating ZIP", key="delete_after_checkbox2")
                 with col4:
-                    upload_to_drive = st.checkbox("Upload to Google Drive")
+                    upload_to_drive = st.checkbox("Upload to Google Drive", key="upload_drive_checkbox2")
                 
-                if st.button("Download Selected"):
+                if st.button("Download Selected", key="download_btn2"):
                     if not os.path.exists(download_dir):
                         os.makedirs(download_dir)
                         
@@ -848,10 +852,10 @@ def main():
 
     elif mode == "Bing Search":
         st.header("Bing Search Mode")
-        query = st.text_input("Enter search query")
-        num_results = st.slider("Number of results", 1, 50, 5)
+        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")
         
-        if st.button("Search"):
+        if st.button("Search", key="search_btn"):
             if query:
                 async def run_search():
                     async with DownloadManager(
@@ -866,11 +870,12 @@ def main():
                                 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}"):
+                                        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
+                                                sublink_limit=max_sublinks,
+                                                timeout=sublink_timeout
                                             )
                                             if files:
                                                 st.session_state.discovered_files = files
@@ -889,9 +894,9 @@ def main():
             st.error("PDF summarization is not available due to model loading errors.")
         else:
             st.header("PDF Summarizer")
-            pdf_url = st.text_input("Enter PDF URL")
+            pdf_url = st.text_input("Enter PDF URL", key="pdf_url_input")
             
-            if st.button("Summarize"):
+            if st.button("Summarize", key="summarize_btn"):
                 if pdf_url:
                     with st.spinner("Generating summary..."):
                         try: