Update app.py

app.py

@@ -3,16 +3,66 @@ import requests
 import json
 import os
 from datetime import datetime, timedelta
-from bs4 import BeautifulSoup  # 웹 페이지에서 텍스트를 추출하기 위해 사용
 from huggingface_hub import InferenceClient  # LLM 사용을 위해 필요
 
-# 필요한 패키지 설치 (필요한 경우 주석을 제거하고 실행)
-# !pip install bs4 huggingface_hub
-
 # 환경 변수에서 API 키 가져오기 (API 키는 안전하게 관리되어야 합니다)
 API_KEY = os.getenv("SERPHOUSE_API_KEY")  # 본인의 SerpHouse API 키를 환경 변수로 설정하세요.
 HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face API 토큰을 환경 변수로 설정하세요.
 
+# 국가 이름과 Google 검색에서 사용하는 국가 코드를 매핑
+COUNTRY_CODE_MAPPING = {
+    "United States": "us",
+    "United Kingdom": "uk",
+    "Canada": "ca",
+    "Australia": "au",
+    "Germany": "de",
+    "France": "fr",
+    "Japan": "jp",
+    "South Korea": "kr",
+    "China": "cn",
+    "India": "in",
+    "Brazil": "br",
+    "Mexico": "mx",
+    "Russia": "ru",
+    "Italy": "it",
+    "Spain": "es",
+    "Netherlands": "nl",
+    "Sweden": "se",
+    "Switzerland": "ch",
+    "Norway": "no",
+    "Denmark": "dk",
+    "Finland": "fi",
+    "Belgium": "be",
+    "Austria": "at",
+    "New Zealand": "nz",
+    "Ireland": "ie",
+    "Singapore": "sg",
+    "Hong Kong": "hk",
+    "Israel": "il",
+    "United Arab Emirates": "ae",
+    "Saudi Arabia": "sa",
+    "South Africa": "za",
+    "Turkey": "tr",
+    "Egypt": "eg",
+    "Poland": "pl",
+    "Czech Republic": "cz",
+    "Hungary": "hu",
+    "Greece": "gr",
+    "Portugal": "pt",
+    "Argentina": "ar",
+    "Chile": "cl",
+    "Colombia": "co",
+    "Peru": "pe",
+    "Venezuela": "ve",
+    "Thailand": "th",
+    "Malaysia": "my",
+    "Indonesia": "id",
+    "Philippines": "ph",
+    "Vietnam": "vn",
+    "Pakistan": "pk",
+    "Bangladesh": "bd"
+}
+
 MAJOR_COUNTRIES = [
     "United States", "United Kingdom", "Canada", "Australia", "Germany",
     "France", "Japan", "South Korea", "China", "India",
@@ -37,7 +87,7 @@ def search_serphouse(query, country, page=1, num_result=100):
         "data": {
             "q": query,
             "domain": "google.com",
-            "loc": country,
+            "loc": COUNTRY_CODE_MAPPING.get(country, "us"),
             "lang": "en",
             "device": "desktop",
             "serp_type": "news",
@@ -127,19 +177,10 @@ def serphouse_search(query, country):
 # LLM 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_TOKEN)
 
-def summarize_article(url):
+def summarize_article(title, snippet):
     try:
-        # 웹 페이지에서 텍스트 추출
-        response = requests.get(url)
-        response.raise_for_status()
-        soup = BeautifulSoup(response.text, 'html.parser')
-        # 모든 텍스트를 추출 (간단한 예시)
-        text = ' '.join([p.get_text() for p in soup.find_all('p')])
-        if not text.strip():
-            return "기사 내용을 가져올 수 없습니다."
-
-        # 요약 생성
-        prompt = f"다음 영어 기사를 한국어로 3문장으로 요약하세요:\n{text}"
+        # 기사 제목과 스니펫을 기반으로 요약 생성
+        prompt = f"다음 뉴스 제목과 요약을 바탕으로 한국어로 3문장으로 요약하세요:\n제목: {title}\n요약: {snippet}"
         summary = hf_client.text_generation(prompt, max_new_tokens=500)
         return summary
     except Exception as e:
@@ -149,10 +190,26 @@ css = """
 footer {
     visibility: hidden;
 }
+/* 분석 버튼 스타일 개선 */
+.analyze-button {
+    background-color: #4CAF50; /* Green */
+    border: none;
+    color: white;
+    padding: 6px 12px;
+    text-align: center;
+    text-decoration: none;
+    font-size: 14px;
+    margin: 2px;
+    cursor: pointer;
+    border-radius: 4px;
+}
+.analyze-button:hover {
+    background-color: #45a049;
+}
 """
 
 # Gradio 인터페이스 구성
-with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
+with gr.Blocks(css=css, title="NewsAI 서비스") as iface:
     gr.Markdown("검색어를 입력하고 원하는 국가를 선택하면, 검색어와 일치하는 24시간 이내 뉴스를 최대 100개 출력합니다.")
 
     with gr.Column():
@@ -170,10 +227,19 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             return f"<p>{error_message}</p>", gr.update(visible=False)
         else:
             # 기사 목록을 HTML 테이블로 생성
-            table_html = "<table border='1' style='width:100%; text-align:left;'><tr><th>번호</th><th>제목</th><th>출처</th><th>시간</th><th>분석</th></tr>"
+            table_html = """
+            <table border='1' style='width:100%; text-align:left;'>
+                <tr>
+                    <th>번호</th>
+                    <th>제목</th>
+                    <th>출처</th>
+                    <th>시간</th>
+                    <th>분석</th>
+                </tr>
+            """
             for article in articles:
                 # 각 기사에 대해 버튼에 해당하는 JavaScript 코드를 삽입
-                analyze_button = f"""<button onclick="analyzeArticle('{article['link']}')">분석</button>"""
+                analyze_button = f"""<button class="analyze-button" onclick="analyzeArticle('{article['index']}')">분석</button>"""
                 row = f"""
                 <tr>
                     <td>{article['index']}</td>
@@ -189,23 +255,36 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             # JavaScript 함수 정의
             js_code = """
             <script>
-            function analyzeArticle(url) {
-                // Gradio의 handle_function을 사용하여 Python 함수 호출
-                gradioApp().querySelector('#article_url_input textarea').value = url;
-                gradioApp().querySelector('#analyze_button').click();
+            function analyzeArticle(index) {
+                // Gradio의 handleFunction을 사용하여 Python 함수 호출
+                const articleData = JSON.parse(document.getElementById('articles_data').textContent);
+                const selectedArticle = articleData.find(article => article.index == index);
+                if (selectedArticle) {
+                    gradioApp().querySelector('#article_title textarea').value = selectedArticle.title;
+                    gradioApp().querySelector('#article_snippet textarea').value = selectedArticle.snippet;
+                    gradioApp().querySelector('#analyze_button').click();
+                }
             }
             </script>
             """
 
-            full_html = table_html + js_code
+            # 기사 데이터를 JSON으로 저장하여 JavaScript에서 접근 가능하도록 함
+            articles_json = json.dumps(articles)
+
+            full_html = f"""
+            <div id="articles_data" style="display:none;">{articles_json}</div>
+            {table_html}
+            {js_code}
+            """
 
             return full_html, gr.update(visible=True, value="")  # summary_output 초기화
 
-    def analyze_article(url):
-        summary = summarize_article(url)
+    def analyze_article(title, snippet):
+        summary = summarize_article(title, snippet)
         return summary
 
-    article_url_input = gr.Textbox(visible=False, elem_id="article_url_input")
+    article_title = gr.Textbox(visible=False, elem_id="article_title")
+    article_snippet = gr.Textbox(visible=False, elem_id="article_snippet")
     analyze_button = gr.Button("분석", visible=False, elem_id="analyze_button")
 
     search_button.click(
@@ -216,7 +295,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
 
     analyze_button.click(
         analyze_article,
-        inputs=[article_url_input],
+        inputs=[article_title, article_snippet],
         outputs=[summary_output]
     )