Update app.py

app.py

@@ -7,25 +7,6 @@ from bs4 import BeautifulSoup
 import hashlib
 import threading
 
-# Updated Global settings with more detailed news sources
-NEWS_SOURCES = {
-    "Technology": {
-        "TheNewYorkTimes": "https://rss.nytimes.com/services/xml/rss/nyt/Technology.xml",
-        "TechCrunch": "http://feeds.feedburner.com/TechCrunch/",
-        "Wired": "https://www.wired.com/feed/rss"
-    },
-    "Business": {
-        "TheNewYorkTimes": "https://rss.nytimes.com/services/xml/rss/nyt/Business.xml", 
-        "Bloomberg": "https://www.bloomberg.com/feeds/podcasts/surveillance.rss",
-        "Forbes": "https://www.forbes.com/feeds/forbesasia"
-    },
-    "World News": {
-        "BBC": "http://feeds.bbci.co.uk/news/world/rss.xml",
-        "TheNewYorkTimes": "https://rss.nytimes.com/services/xml/rss/nyt/World.xml",
-        "Reuters": "http://feeds.reuters.com/Reuters/worldNews"
-    }
-}
-
 # Global settings
 SUMMARIZER_MODELS = {
     "Default (facebook/bart-large-cnn)": "facebook/bart-large-cnn",
@@ -35,6 +16,12 @@ CACHE_SIZE = 500
 RSS_FETCH_INTERVAL = timedelta(hours=8)
 ARTICLE_LIMIT = 5
 
+NEWS_SOURCES = {
+    "Technology": {"TheNewYorkTimes": "https://rss.nytimes.com/services/xml/rss/nyt/Technology.xml"},
+    "Business": {"TheNewYorkTimes": "https://rss.nytimes.com/services/xml/rss/nyt/Business.xml"},
+    "World News": {"BBC": "http://feeds.bbci.co.uk/news/world/rss.xml"}
+}
+
 class NewsCache:
     def __init__(self, size):
         self.cache = {}
@@ -54,15 +41,11 @@ class NewsCache:
 
 cache = NewsCache(CACHE_SIZE)
 
-def fetch_rss_news(selected_sources):
+def fetch_rss_news(categories):
     articles = []
     cutoff_time = datetime.now(pytz.UTC) - RSS_FETCH_INTERVAL
-    
-    for category, sources in NEWS_SOURCES.items():
-        for source, url in sources.items():
-            if f"{category} - {source}" not in selected_sources:
-                continue
-            
+    for category in categories:
+        for source, url in NEWS_SOURCES.get(category, {}).items():
             try:
                 feed = feedparser.parse(url)
                 for entry in feed.entries:
@@ -78,7 +61,6 @@ def fetch_rss_news(selected_sources):
                         })
             except Exception:
                 continue
-    
     articles = sorted(articles, key=lambda x: x["published"], reverse=True)[:ARTICLE_LIMIT]
     return articles
 
@@ -110,14 +92,12 @@ def summarize_articles(articles, model_name):
         """)
     return "\n".join(summaries)
 
-def generate_summary(selected_sources, model_name):
-    if not selected_sources:
-        return "Please select at least one news source."
-    
-    articles = fetch_rss_news(selected_sources)
+def generate_summary(selected_categories, model_name):
+    if not selected_categories:
+        return "Please select at least one category."
+    articles = fetch_rss_news(selected_categories)
     if not articles:
-        return "No recent news found in the selected sources."
-    
+        return "No recent news found in the selected categories."
     return summarize_articles(articles, model_name)
 
 # Gradio Interface
@@ -125,44 +105,34 @@ demo = gr.Blocks()
 
 with demo:
     gr.Markdown("# 📰 AI News Summarizer")
-    with gr.Column():
-        # Dynamically generate sources grouped by category
-        sources_checkboxes = {}
-        for category, sources in NEWS_SOURCES.items():
-            with gr.Accordion(category, open=False):
-                sources_checkboxes[category] = gr.CheckboxGroup(
-                    choices=list(sources.keys()),
-                    label=f"{category} Sources",
-                    interactive=True
-                )
-        
-        # Model selector
+    with gr.Row():
+        categories = gr.CheckboxGroup(
+            choices=list(NEWS_SOURCES.keys()),
+            label="Select News Categories"
+        )
         model_selector = gr.Radio(
             choices=list(SUMMARIZER_MODELS.keys()),
             label="Choose Summarization Model",
             value="Default (facebook/bart-large-cnn)"
         )
-    
     summarize_button = gr.Button("Get News Summary")
     summary_output = gr.Textbox(label="News Summary", lines=20)
 
-    def get_summary(sources_checkboxes, model_name):
-        # Flatten the sources selection
-        selected_sources = []
-        for category, sources in sources_checkboxes.items():
-            selected_sources.extend([f"{category} - {source}" for source in sources])
-        
-        model_name = SUMMARIZER_MODELS[model_name]
-        return generate_summary(selected_sources, model_name)
-
-    summarize_button.click(
-        get_summary, 
-        inputs=[sources_checkboxes, model_selector], 
-        outputs=summary_output
-    )
+    def get_summary(selected_categories, selected_model):
+        model_name = SUMMARIZER_MODELS[selected_model]
+        return generate_summary(selected_categories, model_name)
+
+    summarize_button.click(get_summary, inputs=[categories, model_selector], outputs=summary_output)
 
 if __name__ == "__main__":
     demo.launch()
 
 
 
+
+
+
+
+
+
+