Update app.py

app.py

@@ -1,13 +1,25 @@
 import gradio as gr
 import requests
 from datetime import datetime, timedelta
-import json
+import pycountry
 
 # NewsAPI key (이것을 실제 API 키로 대체해야 합니다)
 API_KEY = "37d83e266422487b8b2e4cb6e1ff0aa6"
 
-def get_news(keyword):
-    base_url = "https://newsapi.org/v2/everything"
+# 국가 코드 리스트
+COUNTRY_CODES = "ae ar at au be bg br ca ch cn co cu cz de eg fr gb gr hk hu id ie il in it jp kr lt lv ma mx my ng nl no nz ph pl pt ro rs ru sa se sg si sk th tr tw ua us ve za".split()
+
+# 국가 코드와 이름 매핑
+COUNTRIES = {'all': 'All Countries'}
+for code in COUNTRY_CODES:
+    try:
+        country = pycountry.countries.get(alpha_2=code.upper())
+        COUNTRIES[code] = country.name
+    except AttributeError:
+        COUNTRIES[code] = code.upper()  # 국가 이름을 찾지 못한 경우 코드를 대문자로 사용
+
+def get_news(keyword, article_count, country):
+    base_url = "https://newsapi.org/v2/top-headlines" if country != 'all' else "https://newsapi.org/v2/everything"
     
     two_days_ago = (datetime.utcnow() - timedelta(hours=48)).isoformat()
 
@@ -16,9 +28,13 @@ def get_news(keyword):
         'q': keyword,
         'from': two_days_ago,
         'language': 'en',
-        'sortBy': 'publishedAt'
+        'sortBy': 'publishedAt',
+        'pageSize': article_count
     }
-    
+
+    if country != 'all':
+        params['country'] = country
+
     try:
         response = requests.get(base_url, params=params, timeout=10)
         response.raise_for_status()
@@ -35,8 +51,8 @@ def get_news(keyword):
         return (f"<p>No recent news found for the keyword '<strong>{keyword}</strong>' within the last 48 hours.<br>"
                 f"Try a different keyword or check back later.</p>")
 
-    html_output = f"<h2>News results for '{keyword}'</h2>"
-    for article in articles[:10]:  # 최대 10개의 기사만 표시
+    html_output = f"<h2>News results for '{keyword}' in {COUNTRIES[country]}</h2>"
+    for article in articles:
         title = article['title']
         link = article['url']
         pub_date = datetime.strptime(article['publishedAt'], "%Y-%m-%dT%H:%M:%SZ")
@@ -55,10 +71,12 @@ def get_news(keyword):
 iface = gr.Interface(
     fn=get_news,
     inputs=[
-        gr.Textbox(label="Enter keyword")
+        gr.Textbox(label="Enter keyword"),
+        gr.Slider(minimum=10, maximum=100, step=10, value=10, label="Number of Articles"),
+        gr.Dropdown(choices=list(COUNTRIES.values()), value="All Countries", label="Select Country")
     ],
     outputs=gr.HTML(),
-    title="Visual News Search",
+    title="Enhanced Visual News Search",
     description="Search for news articles from the last 48 hours using NewsAPI.",
     theme=gr.themes.Soft()
 )