Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import requests
 from datetime import datetime, timedelta
@@ -12,15 +13,10 @@ COUNTRY_CODES = "ae ar at au be bg br ca ch cn co cu cz de eg fr gb gr hk hu id
 
 # 국가 코드와 이름 매핑
 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()
+COUNTRIES.update({code: pycountry.countries.get(alpha_2=code.upper()).name for code in COUNTRY_CODES if pycountry.countries.get(alpha_2=code.upper())})
 
 def get_news(keyword, article_count, country):
-    country_code = [code for code, name in COUNTRIES.items() if name == country][0]
+    country_code = next((code for code, name in COUNTRIES.items() if name == country), 'all')
     
     if country_code == 'all':
         base_url = "https://newsapi.org/v2/everything"
@@ -36,7 +32,6 @@ def get_news(keyword, article_count, country):
         params = {
             'apiKey': API_KEY,
             'q': keyword,
-            'language': 'en',
             'country': country_code,
             'pageSize': article_count
         }
@@ -65,7 +60,7 @@ def get_news(keyword, article_count, country):
     articles = news_data['articles']
     
     if not articles:
-        return (f"<p>No recent news found for the keyword '<strong>{keyword}</strong>' within the last 48 hours.<br>"
+        return (f"<p>No recent news found for the keyword '<strong>{keyword}</strong>' in {country} 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 {country}</h2>"