Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import requests
 from datetime import datetime, timedelta
-import concurrent.futures
+
 
 # Google Custom Search API 키와 검색 엔진 ID
 API_KEY = "AIzaSyDUz3wkGal0ewRtPlzeMit88bV4hS4ZIVY"
@@ -68,7 +68,8 @@ COUNTRIES = {
     'Wallis and Futuna': 'countryWF', 'Western Sahara': 'countryEH', 'Yemen': 'countryYE', 'Zambia': 'countryZM', 'Zimbabwe': 'countryZW'
 }
 
-def search_news(keyword, country, search_all_countries=False):
+
+def search_news(keyword, country):
     # 24시간 전 날짜 계산
     one_day_ago = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d")
 
@@ -85,7 +86,7 @@ def search_news(keyword, country, search_all_countries=False):
         'siteSearch': 'news.google.com',  # Google News로 제한
     }
 
-    if not search_all_countries and country != 'All Countries':
+    if country != 'All Countries':
         params['cr'] = COUNTRIES[country]
 
     # API 요청
@@ -106,46 +107,16 @@ def search_news(keyword, country, search_all_countries=False):
 
     return formatted_results
 
-def search_all_countries(keyword):
-    def search_country(country_name):
-        result = search_news(keyword, country_name)
-        if "No news found" not in result:
-            return country_name, result
-        return None
-
-    with concurrent.futures.ThreadPoolExecutor(max_workers=20) as executor:
-        future_to_country = {executor.submit(search_country, country): country for country in COUNTRIES.keys()}
-        matching_countries = []
-        for future in concurrent.futures.as_completed(future_to_country):
-            result = future.result()
-            if result:
-                matching_countries.append(result)
-
-    if matching_countries:
-        output = "<h2>Countries with matching news:</h2>"
-        for country, news in matching_countries:
-            output += f"<h3>{country}</h3>{news}<hr>"
-        return output
-    else:
-        return f"No news found for '{keyword}' in any country within the last 24 hours."
-
-def news_search(keyword, country, search_all):
-    if search_all:
-        return search_all_countries(keyword)
-    else:
-        return search_news(keyword, country)
-
 # Gradio 인터페이스 생성
 iface = gr.Interface(
-    fn=news_search,
+    fn=search_news,
     inputs=[
         gr.Textbox(label="Enter keyword (in English)"),
-        gr.Dropdown(choices=['All Countries'] + list(COUNTRIES.keys()), label="Select Country"),
-        gr.Checkbox(label="Search all countries")
+        gr.Dropdown(choices=['All Countries'] + list(COUNTRIES.keys()), label="Select Country")
     ],
     outputs=gr.HTML(),
     title="Google News Search",
-    description="Search for news articles from the last 24 hours using Google Custom Search API. You can search in a specific country or across all countries."
+    description="Search for news articles from the last 24 hours using Google Custom Search API."
 )
 
 # 애플리케이션 실행 (공개 링크 생성)