Update app.py

app.py

@@ -1,94 +1,97 @@
+%%writefile app.py
+
 import streamlit as st
-import sqlite3
 import requests
-from datetime import datetime
 
-# Connect to the SQLite database
-def connect_db():
-    conn = sqlite3.connect("news_aggregator.db")
-    return conn
+API_KEY = "fe1e6bcbbf384b3e9220a7a1138805e0"  # Replace with your actual API key
 
-# Function to create or update the articles in the database
-def insert_article(title, description, link, topic, date_published):
-    conn = connect_db()
-    cursor = conn.cursor()
-    
-    # Insert article into the database
-    cursor.execute('''
-    INSERT INTO articles (title, description, link, topic, date_published)
-    VALUES (?, ?, ?, ?, ?)
-    ''', (title, description, link, topic, date_published))
-    
-    conn.commit()
-    conn.close()
+publisher_articles = []
 
-# Function to fetch news articles from an API (e.g., NewsAPI)
-def fetch_news_api(topic, keyword):
-    api_key = 'fe1e6bcbbf384b3e9220a7a1138805e0'  # Replace with your News API key
-    url = f'https://newsapi.org/v2/everything?q={keyword}&category={topic}&apiKey={api_key}'
+def fetch_news(topic, keyword):
+    query = f"{topic} {keyword}" if keyword else topic
+    url = f"https://newsapi.org/v2/everything?q={query}&apiKey={API_KEY}"
     response = requests.get(url)
-    
     if response.status_code == 200:
-        return response.json()['articles']
+        return response.json().get("articles", [])
     else:
+        st.error("Failed to fetch news. Check your API key or internet connection.")
         return []
 
-# Function to fetch articles from the database based on topic and filter
-def fetch_articles_from_db(topic, keyword_filter=None):
-    conn = connect_db()
-    cursor = conn.cursor()
-    
-    query = 'SELECT * FROM articles WHERE topic=?'
-    params = (topic,)
+def main():
+    st.title("📰 News Aggregator")
+    st.write("Publish and subscribe to topics to get the latest updates!")
+
+    menu = ["Home", "Publisher Panel", "Subscriber Panel"]
+    choice = st.sidebar.selectbox("Choose a panel:", menu)
 
-    if keyword_filter:
-        query += ' AND description LIKE ?'
-        params = (topic, f'%{keyword_filter}%')
+    if choice == "Home":
+        st.header("Welcome to the News Aggregator!")
+        st.write("Navigate to the **Publisher Panel** to add news articles or to the **Subscriber Panel** to view articles.")
+    
+    elif choice == "Publisher Panel":
+        st.header("📝 Publisher Panel")
+        st.write("Publish articles to relevant topics.")
+        
+        title = st.text_input("Article Title:")
+        description = st.text_area("Article Description:")
+        link = st.text_input("Article Link:")
+        topic = st.selectbox("Select Topic:", ["Sports", "Politics", "Technology", "Entertainment", "Health", "Science"])
 
-    cursor.execute(query, params)
-    articles = cursor.fetchall()
-    conn.close()
+        if st.button("Publish Article"):
+            if title and description and link and topic:
+                publisher_articles.append({"title": title, "description": description, "link": link, "topic": topic})
+                st.success(f"Article on '{topic}' published successfully!")
+            else:
+                st.warning("Please fill in all fields before publishing.")
 
-    return articles
+    elif choice == "Subscriber Panel":
+        st.header("🔔 Subscriber Panel")
+        st.write("Subscribe to topics and view news articles.")
+        
+        topics = ["Sports", "Politics", "Technology", "Entertainment", "Health", "Science"]
+        selected_topics = st.multiselect("Select topics to subscribe to:", topics)
+        keyword_filter = st.text_input("🔍 Enter a keyword to filter news:")
 
-# Streamlit UI code
-def main():
-    st.title("News Aggregator")
+        if st.button("Fetch News"):
+            if not selected_topics:
+                st.warning("Please select at least one topic to subscribe to.")
+            else:
+                # Notify user about subscription
+                st.success(f"🔔 You have subscribed to: {', '.join(selected_topics)}")
+                st.info("Fetching news articles...")
 
-    # Topic selection
-    topic = st.selectbox("Select Topic", ['Sports', 'Politics', 'Technology'])
-    keyword_filter = st.text_input("Filter by Keyword")
+                all_articles = []
 
-    # Fetch news articles
-    if st.button("Fetch News"):
-        # First, fetch from News API
-        articles = fetch_news_api(topic, keyword_filter)
+                # Fetch publisher articles for selected topics
+                for article in publisher_articles:
+                    if article["topic"] in selected_topics:
+                        all_articles.append(article)
+                
+                # Fetch external news articles from API
+                for topic in selected_topics:
+                    fetched_articles = fetch_news(topic, keyword_filter)
+                    for article in fetched_articles:
+                        all_articles.append({
+                            "title": article.get("title", "No Title"),
+                            "description": article.get("description", "No Description"),
+                            "link": article.get("url", "#"),
+                            "topic": topic
+                        })
 
-        # Insert the fetched articles into the database
-        for article in articles:
-            insert_article(
-                article['title'],
-                article['description'],
-                article['url'],
-                topic,
-                article['publishedAt']
-            )
-        
-        st.success(f"Fetched {len(articles)} articles from News API.")
+                # Filter articles by keyword if provided
+                if keyword_filter:
+                    all_articles = [a for a in all_articles if keyword_filter.lower() in a["title"].lower()]
 
-    # Fetch and display articles from the database
-    st.subheader("Articles from Database")
-    fetched_articles = fetch_articles_from_db(topic, keyword_filter)
-    
-    if fetched_articles:
-        for article in fetched_articles:
-            st.write(f"**Title**: {article[1]}")
-            st.write(f"**Description**: {article[2]}")
-            st.write(f"**Link**: {article[3]}")
-            st.write(f"**Published**: {article[5]}")
-            st.write("---")
-    else:
-        st.write("No articles found for this topic.")
+                if all_articles:
+                    st.success(f"Found {len(all_articles)} articles!")
+                    for article in all_articles:
+                        st.subheader(article["title"])
+                        st.write(article["description"])
+                        st.write(f"Topic: {article['topic']}")
+                        st.write(f"[Read more]({article['link']})")
+                        st.write("---")
+                else:
+                    st.info("No articles found for the selected topics or keyword.")
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()