Update app.py

app.py

@@ -3,13 +3,9 @@ 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 키 가져오기
 API_KEY = os.getenv("SERPHOUSE_API_KEY")  # 본인의 SerpHouse API 키를 환경 변수로 설정하세요.
 HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face API 토큰을 환경 변수로 설정하세요.
 
@@ -37,7 +33,7 @@ def search_serphouse(query, country, page=1, num_result=10):
         "data": {
             "q": query,
             "domain": "google.com",
-            "loc": country,
+            "loc": country,  # 국가 이름을 직접 사용합니다.
             "lang": "en",
             "device": "desktop",
             "serp_type": "news",
@@ -60,7 +56,7 @@ def search_serphouse(query, country, page=1, num_result=10):
         return response.json()
     except requests.RequestException as e:
         error_msg = f"Error: {str(e)}"
-        if response.text:
+        if hasattr(response, 'text'):
             error_msg += f"\nResponse content: {response.text}"
         return {"error": error_msg}
 
@@ -72,12 +68,11 @@ def format_results_from_raw(results):
         if not isinstance(results, dict):
             raise ValueError("결과가 사전 형식이 아닙니다.")
 
-        # 'results' 키 내부의 구조 확인 (중첩된 'results' 처리)
+        # 'results' 키 내부의 구조 확인
         if 'results' in results:
             results_content = results['results']
             if 'results' in results_content:
                 results_content = results_content['results']
-                # 'news' 키 확인
                 if 'news' in results_content:
                     news_results = results_content['news']
                 else:
@@ -101,6 +96,7 @@ def format_results_from_raw(results):
             image_url = result.get("img", result.get("thumbnail", ""))
 
             articles.append({
+                "index": idx,
                 "title": title,
                 "link": link,
                 "snippet": snippet,
@@ -116,7 +112,6 @@ def format_results_from_raw(results):
         return "Error: " + error_message, []
 
 def serphouse_search(query, country):
-    # 페이지와 결과 수의 기본값을 설정합니다.
     page = 1
     num_result = 10
     results = search_serphouse(query, country, page, num_result)
@@ -126,19 +121,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:
@@ -160,43 +146,112 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             country = gr.Dropdown(MAJOR_COUNTRIES, label="국가", value="South Korea")
             search_button = gr.Button("검색")
 
-        article_outputs = []
-
-        def search_and_display(query, country):
-            error_message, articles = serphouse_search(query, country)
-            if error_message:
-                return gr.update(visible=True, value=error_message)
-            else:
-                # 기존 출력물 제거
-                for components in article_outputs:
-                    for component in components:
-                        component.visible = False
-                article_outputs.clear()
-
-                # 각 기사에 대해 출력 생성
-                for article in articles:
-                    with gr.Column():
-                        title = gr.Markdown(f"### [{article['title']}]({article['link']})")
-                        image = gr.Image(value=article['image_url'], visible=bool(article['image_url']), shape=(200, 150))
-                        snippet = gr.Markdown(f"**요약:** {article['snippet']}")
-                        info = gr.Markdown(f"**출처:** {article['channel']} | **시간:** {article['time']}")
-                        analyze_button = gr.Button("분석")
-                        summary_output = gr.Markdown(visible=False)
-
-                        def analyze_article(url):
-                            summary = summarize_article(url)
-                            summary_output.update(value=summary, visible=True)
-
-                        analyze_button.click(analyze_article, inputs=gr.State(article['link']), outputs=summary_output)
-
-                        article_outputs.append([title, image, snippet, info, analyze_button, summary_output])
-
-                return gr.update()
-
-        search_button.click(
-            search_and_display,
-            inputs=[query, country],
-            outputs=[]
-        )
+        # 최대 10개의 기사에 대한 컴포넌트를 미리 생성합니다.
+        article_components = []
+        for i in range(10):
+            with gr.Group(visible=False) as article_group:
+                title = gr.Markdown()
+                image = gr.Image(width=200, height=150)
+                snippet = gr.Markdown()
+                info = gr.Markdown()
+                analyze_button = gr.Button("분석")
+                summary_output = gr.Markdown(visible=False)
+
+                article_components.append({
+                    'group': article_group,
+                    'title': title,
+                    'image': image,
+                    'snippet': snippet,
+                    'info': info,
+                    'analyze_button': analyze_button,
+                    'summary_output': summary_output,
+                })
+
+    def search_and_display(query, country):
+        error_message, articles = serphouse_search(query, country)
+        outputs = []
+        if error_message:
+            outputs.append(gr.update(value=error_message, visible=True))
+            # 나머지 컴포넌트 숨기기
+            for comp in article_components:
+                outputs.extend([
+                    gr.update(visible=False),  # group
+                    gr.update(),  # title
+                    gr.update(),  # image
+                    gr.update(),  # snippet
+                    gr.update(),  # info
+                    gr.update(),  # analyze_button
+                    gr.update(visible=False),  # summary_output
+                ])
+            return outputs
+        else:
+            # 기사 컴포넌트 업데이트
+            for idx, comp in enumerate(article_components):
+                if idx < len(articles):
+                    article = articles[idx]
+                    comp['group'].visible = True
+                    comp['title'].value = f"### [{article['title']}]({article['link']})"
+                    if article['image_url'] and not article['image_url'].startswith("data:image"):
+                        comp['image'].value = article['image_url']
+                        comp['image'].visible = True
+                    else:
+                        comp['image'].visible = False
+                    comp['snippet'].value = f"**요약:** {article['snippet']}"
+                    comp['info'].value = f"**출처:** {article['channel']} | **시간:** {article['time']}"
+                    comp['summary_output'].visible = False  # 초기에는 요약 숨김
+
+                    # 분석 버튼 클릭 이벤트 정의
+                    def create_analyze_function(article_title, article_snippet):
+                        def analyze_article():
+                            summary = summarize_article(article_title, article_snippet)
+                            return gr.update(value=summary, visible=True)
+                        return analyze_article
+
+                    comp['analyze_button'].click(
+                        create_analyze_function(article['title'], article['snippet']),
+                        inputs=[],
+                        outputs=comp['summary_output']
+                    )
+
+                    outputs.extend([
+                        gr.update(visible=True),  # group
+                        gr.update(),  # title
+                        gr.update(),  # image
+                        gr.update(),  # snippet
+                        gr.update(),  # info
+                        gr.update(),  # analyze_button
+                        gr.update(visible=False),  # summary_output
+                    ])
+                else:
+                    # 남은 컴포넌트 숨기기
+                    comp['group'].visible = False
+                    outputs.extend([
+                        gr.update(visible=False),  # group
+                        gr.update(),  # title
+                        gr.update(),  # image
+                        gr.update(),  # snippet
+                        gr.update(),  # info
+                        gr.update(),  # analyze_button
+                        gr.update(visible=False),  # summary_output
+                    ])
+            return outputs
+
+    # search_button 클릭 시 업데이트될 출력 컴포넌트 목록 생성
+    search_outputs = []
+    search_outputs.append(gr.Markdown(visible=False))  # 오류 메시지 출력용
+    for comp in article_components:
+        search_outputs.append(comp['group'])
+        search_outputs.append(comp['title'])
+        search_outputs.append(comp['image'])
+        search_outputs.append(comp['snippet'])
+        search_outputs.append(comp['info'])
+        search_outputs.append(comp['analyze_button'])
+        search_outputs.append(comp['summary_output'])
+
+    search_button.click(
+        search_and_display,
+        inputs=[query, country],
+        outputs=search_outputs
+    )
 
 iface.launch(auth=("gini", "pick"))