Update app.py

app.py

@@ -1,97 +1,94 @@
 import streamlit as st
+import sqlite3
 import requests
+from datetime import datetime
 
-# Replace this with your actual News API key
-API_KEY = "fe1e6bcbbf384b3e9220a7a1138805e0"  # 🔑 Add your NewsAPI.org API key here
+# Connect to the SQLite database
+def connect_db():
+    conn = sqlite3.connect("news_aggregator.db")
+    return conn
 
-publisher_articles = []
+# 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()
 
-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}"
+# 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}'
     response = requests.get(url)
+    
     if response.status_code == 200:
-        return response.json().get("articles", [])
+        return response.json()['articles']
     else:
-        st.error("Failed to fetch news. Check your API key or internet connection.")
         return []
 
-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 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.")
+# 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()
     
-    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"])
+    query = 'SELECT * FROM articles WHERE topic=?'
+    params = (topic,)
 
-        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.")
+    if keyword_filter:
+        query += ' AND description LIKE ?'
+        params = (topic, f'%{keyword_filter}%')
 
-    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:")
+    cursor.execute(query, params)
+    articles = cursor.fetchall()
+    conn.close()
 
-        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...")
+    return articles
 
-                all_articles = []
+# Streamlit UI code
+def main():
+    st.title("News Aggregator")
 
-                # 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
-                        })
+    # Topic selection
+    topic = st.selectbox("Select Topic", ['Sports', 'Politics', 'Technology'])
+    keyword_filter = st.text_input("Filter by Keyword")
 
-                # 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 news articles
+    if st.button("Fetch News"):
+        # First, fetch from News API
+        articles = fetch_news_api(topic, keyword_filter)
 
-                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.")
+        # 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.")
 
-if __name__ == "__main__":
-    main()
+    # 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 __name__ == '__main__':
+    main()