Update app.py

app.py

@@ -7,12 +7,20 @@ import json
 # NewsAPI key
 API_KEY = "37d83e266422487b8b2e4cb6e1ff0aa6"
 
+
 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'}
 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, extend_search=False):
+def get_related_keywords(keyword):
+    # 간단한 관련 키워드 생성 로직 (실제로는 더 복잡한 알고리즘이나 API를 사용할 수 있습니다)
+    return [f"{keyword} news", f"{keyword} technology", f"{keyword} company", f"latest {keyword}"]
+
+def get_news(keyword, article_count, country, extend_search=False, search_all_countries=False):
+    if search_all_countries:
+        country = 'All Countries'
+    
     country_code = next((code for code, name in COUNTRIES.items() if name == country), 'all')
     
     if country_code == 'all':
@@ -60,19 +68,24 @@ def get_news(keyword, article_count, country, extend_search=False):
     if not articles:
         if not extend_search:
             return get_news(keyword, article_count, country, extend_search=True)
+        elif not search_all_countries and country != 'All Countries':
+            return get_news(keyword, article_count, 'All Countries', extend_search=True, search_all_countries=True)
         else:
-            return (f"<p>No news found for the keyword '<strong>{keyword}</strong>' in {country}.<br>"
-                    f"Suggestions:</p>"
-                    f"<ul>"
-                    f"<li>Try a different keyword</li>"
-                    f"<li>Expand your search to all countries</li>"
-                    f"<li>Increase the number of articles</li>"
-                    f"</ul>"
-                    f"<pre>{debug_info}</pre>")
+            related_keywords = get_related_keywords(keyword)
+            suggestions = (f"<p>No news found for the keyword '<strong>{keyword}</strong>' in {country}.</p>"
+                           f"<p>Suggestions:</p>"
+                           f"<ul>"
+                           f"<li>Try one of these related keywords: {', '.join(related_keywords)}</li>"
+                           f"<li>Increase the number of articles</li>"
+                           f"<li>Check for any spelling errors in your keyword</li>"
+                           f"</ul>")
+            return suggestions + f"<pre>{debug_info}</pre>"
 
     html_output = f"<h2>News results for '{keyword}' in {country}</h2>"
     if extend_search:
         html_output += "<p><em>Note: Search range was extended due to limited recent results.</em></p>"
+    if search_all_countries:
+        html_output += "<p><em>Note: Search was expanded to all countries due to limited results in the specified country.</em></p>"
     
     for article in articles:
         title = article['title']
@@ -101,8 +114,8 @@ iface = gr.Interface(
         gr.Dropdown(choices=list(COUNTRIES.values()), value="All Countries", label="Select Country")
     ],
     outputs=gr.HTML(),
-    title="Enhanced Visual News Search",
-    description="Search for news articles using NewsAPI. If recent results are limited, the search range will be automatically extended.",
+    title="Advanced Visual News Search",
+    description="Search for news articles using NewsAPI. The search will automatically extend if results are limited.",
     theme=gr.themes.Soft()
 )