Update app.py

app.py

@@ -1,11 +1,10 @@
-
 import gradio as gr
 import requests
 from datetime import datetime, timedelta
 import pycountry
 import json
 
-# NewsAPI key (이것을 실제 API 키로 대체해야 합니다)
+# NewsAPI key
 API_KEY = "37d83e266422487b8b2e4cb6e1ff0aa6"
 
 # 국가 코드 리스트
@@ -21,21 +20,29 @@ for code in COUNTRY_CODES:
         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"
+    country_code = [code for code, name in COUNTRIES.items() if name == country][0]
+    
+    if country_code == 'all':
+        base_url = "https://newsapi.org/v2/everything"
+        params = {
+            'apiKey': API_KEY,
+            'q': keyword,
+            'language': 'en',
+            'sortBy': 'publishedAt',
+            'pageSize': article_count
+        }
+    else:
+        base_url = "https://newsapi.org/v2/top-headlines"
+        params = {
+            'apiKey': API_KEY,
+            'q': keyword,
+            'language': 'en',
+            'country': country_code,
+            'pageSize': article_count
+        }
     
     two_days_ago = (datetime.utcnow() - timedelta(hours=48)).isoformat()
-
-    params = {
-        'apiKey': API_KEY,
-        'q': keyword,
-        'from': two_days_ago,
-        'language': 'en',
-        'sortBy': 'publishedAt',
-        'pageSize': article_count
-    }
-
-    if country != 'all':
-        params['country'] = country
+    params['from'] = two_days_ago
 
     debug_info = f"API Request URL: {base_url}\n"
     debug_info += f"Parameters: {json.dumps(params, indent=2)}\n\n"
@@ -61,7 +68,7 @@ def get_news(keyword, article_count, country):
         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><pre>{debug_info}</pre>")
 
-    html_output = f"<h2>News results for '{keyword}' in {COUNTRIES[country]}</h2>"
+    html_output = f"<h2>News results for '{keyword}' in {country}</h2>"
     for article in articles:
         title = article['title']
         link = article['url']
@@ -87,8 +94,8 @@ iface = gr.Interface(
         gr.Dropdown(choices=list(COUNTRIES.values()), value="All Countries", label="Select Country")
     ],
     outputs=gr.HTML(),
-    title="Debug: Enhanced Visual News Search",
-    description="Search for news articles from the last 48 hours using NewsAPI. Debug information is included.",
+    title="Enhanced Visual News Search",
+    description="Search for news articles from the last 48 hours using NewsAPI.",
     theme=gr.themes.Soft()
 )