Update app.py

app.py

@@ -11,6 +11,7 @@ import logging
 from typing import Optional, Dict, List, Any
 from pathlib import Path
 from shot_scraper import shotscraper  # importing shot-scraper
+import feedgenerator  # Import RSS feed generator
 
 # set up logging
 logging.basicConfig(level=logging.INFO)
@@ -301,22 +302,49 @@ def show_download_links(subdir: str) -> None:
         else:
             st.write(f"File not found: {file}")
 
+# Generate RSS feed
+def generate_rss_feed():
+    feed = feedgenerator.Rss201rev2Feed(
+        title="Infinite Dataset Hub Updates",
+        link="https://huggingface.co/spaces/infinite-dataset-hub/infinite-dataset-hub",
+        description="Latest updates from the Infinite Dataset Hub",
+        language="en"
+    )
+
+    for i, line in enumerate(urls):
+        dataset_name = line
+        feed.add_item(
+            title=dataset_name,
+            link=urls[dataset_name],
+            description=f"Link to {dataset_name}",
+            pubdate=time.gmtime(time.time() - 86400 * i)
+        )
+
+    return feed.writeString('utf-8')
+
 def main() -> None:
-    """main application function."""
-    st.sidebar.title("infinite dataset hub")
-    st.sidebar.subheader("available urls")
+    """main app"""
+    st.title("RSS Feed and Content Downloader")
+
+    # Initialize history
+    initialize_history()
 
-    # Add a title to the app
-    st.title("Music Download App")
+    # RSS Feed Section
+    st.header("RSS Feed")
+    if st.button("Generate RSS Feed"):
+        rss_feed = generate_rss_feed()
+        st.success("RSS Feed generated successfully!")
+        st.markdown(rss_feed, unsafe_allow_html=True)
 
-    for name, url in urls.items():
-        if st.sidebar.button(name):
-            st.write(f"downloading content from {url}")
-            download_html_and_files(url, name.replace(" ", "_"))
+    # Content Downloader Section
+    st.header("Content Downloader")
+    selected_url = st.selectbox("Select a URL to download content from:", list(urls.keys()))
+    subdir = st.text_input("Enter subdirectory name to save files:", "downloads")
 
-# Initialize history
-initialize_history()
+    if st.button("Download Content"):
+        download_html_and_files(urls[selected_url], subdir)
+        st.success("Content downloaded successfully!")
+        show_download_links(subdir)
 
 if __name__ == "__main__":
-    main()
-    st.write("debugging: app launched successfully.")
+    main()