Update app.py

app.py

@@ -3,12 +3,10 @@ import requests
 import xml.etree.ElementTree as ET
 import time
 
-RSS_FEED_URL = "http://www.yonhapnewstv.co.kr/category/news/headline/feed/"
-
-def get_news_from_rss():
-    """연합뉴스TV RSS 피드에서 헤드라인 뉴스를 가져옵니다."""
+def get_news_from_rss(rss_url):
+    """RSS 피드에서 뉴스를 가져옵니다."""
     try:
-        response = requests.get(RSS_FEED_URL)
+        response = requests.get(rss_url)
         if response.status_code == 200:
             root = ET.fromstring(response.content)
             items = root.findall(".//item")
@@ -22,10 +20,8 @@ def get_news_from_rss():
         print(f"Error fetching RSS feed: {e}")
     return []
 
-def display_news():
-    """RSS 피드에서 뉴스를 가져와 요약 및 표시합니다."""
-    news_items = get_news_from_rss()
-
+def display_news(news_items):
+    """뉴스를 요약 및 표시합니다."""
     if not news_items:
         st.warning("뉴스를 가져오는데 실패했습니다.")
         return
@@ -74,7 +70,18 @@ def main():
         </style>
     """, unsafe_allow_html=True)
 
-    display_news()
+    # 뉴스 선택 버튼 추가
+    news_source = st.sidebar.radio("뉴스 선택", ["jtbc 뉴스", "연합뉴스", "SBS 뉴스"])
+
+    if news_source == "jtbc 뉴스":
+        rss_url = "https://news-ex.jtbc.co.kr/v1/get/rss/issue"
+    elif news_source == "연합뉴스":
+        rss_url = "http://www.yonhapnewstv.co.kr/category/news/headline/feed/"
+    else:
+        rss_url = "https://news.sbs.co.kr/news/headlineRssFeed.do?plink=RSSREADER"
+
+    news_items = get_news_from_rss(rss_url)
+    display_news(news_items)
 
     # 새로고침 타이머 설정
     if 'last_run' not in st.session_state: