Update app.py

app.py

@@ -129,53 +129,59 @@ def get_download_link(file):
         b64 = base64.b64encode(bytes).decode()
         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 main():
     st.sidebar.title('Web Datasets Bulk Downloader')
 
-    # Selecting URL input method
-    url_input_method = st.sidebar.radio("Choose URL Input Method", ["Enter URL", "Select from List"])
-    url = ""
-    if url_input_method == "Enter URL":
-        url = st.sidebar.text_input('Please enter a Web URL to bulk download text and files')
-    else:
-        selected_site = st.sidebar.selectbox("Select a Website", list(URLS.keys()))
-        url = URLS[selected_site]
-
-    # Reading or creating history.json
-    if not os.path.exists("history.json"):
-        with open("history.json", "w") as f:
-            json.dump({}, f)
-
-    with open("history.json", "r") as f:
-        history = json.load(f)
-
-    # Handling URL submission
-    if url:
-        subdir = hashlib.md5(url.encode()).hexdigest()
-        if not os.path.exists(subdir):
-            os.makedirs(subdir)
-        if url not in history:
-            history[url] = subdir
-            with open("history.json", "w") as f:
-                json.dump(history, f)
-
-    # Button for downloading content
-    if st.sidebar.button('📥 Get All the Content'):
-        download_html_and_files(url, history[url])
-        show_download_links(history[url])
+    # Check for query parameters for file editing
+    query_params = st.experimental_get_query_params()
+    file_to_edit = query_params.get('file_to_edit', [None])[0]
 
-    # Button for showing download links
-    if st.sidebar.button('📂 Show Download Links'):
-        for subdir in history.values():
-            show_download_links(subdir)
+    if file_to_edit and os.path.exists(file_to_edit):
+        file_editor(file_to_edit)
+    else:
+        # Selecting URL input method
+        url_input_method = st.sidebar.radio("Choose URL Input Method", ["Enter URL", "Select from List"])
+        url = ""
+        if url_input_method == "Enter URL":
+            url = st.sidebar.text_input('Please enter a Web URL to bulk download text and files')
+        else:
+            selected_site = st.sidebar.selectbox("Select a Website", list(URLS.keys()))
+            url = URLS[selected_site]
 
-    # Expander for showing URL history and download links
-    with st.expander("URL History and Downloaded Files"):
-        for url, subdir in history.items():
-            st.markdown(f"#### {url}")
-            show_download_links(subdir)
+        # Reading or creating history.json
+        if not os.path.exists("history.json"):
+            with open("history.json", "w") as f:
+                json.dump({}, f)
+
+        with open("history.json", "r") as f:
+            history = json.load(f)
+
+        # Handling URL submission
+        if url:
+            subdir = hashlib.md5(url.encode()).hexdigest()
+            if not os.path.exists(subdir):
+                os.makedirs(subdir)
+            if url not in history:
+                history[url] = subdir
+                with open("history.json", "w") as f:
+                    json.dump(history, f)
+
+        # Button for downloading content
+        if st.sidebar.button('📥 Get All the Content'):
+            download_html_and_files(url, history[url])
+            show_download_links(history[url])
+
+        # Button for showing download links
+        if st.sidebar.button('📂 Show Download Links'):
+            for subdir in history.values():
+                show_download_links(subdir)
+
+        # Expander for showing URL history and download links
+        with st.expander("URL History and Downloaded Files"):
+            for url, subdir in history.items():
+                st.markdown(f"#### {url}")
+                show_download_links(subdir)
 
-            
 if __name__ == "__main__":
     main()
+