Update app.py

app.py

@@ -6,21 +6,17 @@ from datetime import datetime, timedelta
 from huggingface_hub import InferenceClient  # LLM 사용을 위해 필요
 
 # 환경 변수에서 API 키 가져오기
-API_KEY = os.getenv("SERPHOUSE_API_KEY")  # 본인의 SerpHouse API 키를 환경 변수로 설정하세요.
-HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face API 토큰을 환경 변수로 설정하세요.
-
-MAJOR_COUNTRIES = [
-    "United States", "United Kingdom", "Canada", "Australia", "Germany",
-    "France", "Japan", "South Korea", "China", "India",
-    "Brazil", "Mexico", "Russia", "Italy", "Spain",
-    "Netherlands", "Sweden", "Switzerland", "Norway", "Denmark",
-    "Finland", "Belgium", "Austria", "New Zealand", "Ireland",
-    "Singapore", "Hong Kong", "Israel", "United Arab Emirates", "Saudi Arabia",
-    "South Africa", "Turkey", "Egypt", "Poland", "Czech Republic",
-    "Hungary", "Greece", "Portugal", "Argentina", "Chile",
-    "Colombia", "Peru", "Venezuela", "Thailand", "Malaysia",
-    "Indonesia", "Philippines", "Vietnam", "Pakistan", "Bangladesh"
-]
+API_KEY = os.getenv("SERPHOUSE_API_KEY")
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# 국가 이름과 위치 ID를 매핑
+COUNTRY_CODE_MAPPING = {
+    "United States": "2840",
+    "South Korea": "2458",
+    # 다른 국가들의 위치 ID를 추가하세요
+}
+
+MAJOR_COUNTRIES = list(COUNTRY_CODE_MAPPING.keys())
 
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
@@ -29,11 +25,14 @@ def search_serphouse(query, country, page=1, num_result=10):
     yesterday = now - timedelta(days=1)
     date_range = f"{yesterday.strftime('%Y-%m-%d')},{now.strftime('%Y-%m-%d')}"
 
+    # 국가 이름을 위치 ID로 변환
+    loc_id = COUNTRY_CODE_MAPPING.get(country, "2840")  # 기본값은 미국
+
     payload = {
         "data": {
             "q": query,
             "domain": "google.com",
-            "loc": country,  # 국가 이름을 직접 사용합니다.
+            "loc": loc_id,  # 위치 ID 사용
             "lang": "en",
             "device": "desktop",
             "serp_type": "news",
@@ -136,7 +135,7 @@ footer {
 """
 
 # 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시간 이내 뉴스를 최대 10개 출력합니다.")
 
     with gr.Column():
@@ -145,8 +144,8 @@ 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("검색")
 
-        # 상태 메시지 컴포넌트 추가
-        status_message = gr.Markdown(visible=False)
+        # 상태 메시지 컴포넌트 제거 (Progress를 사용하므로 불필요)
+        # status_message = gr.Markdown(visible=False)
 
         # 기사 데이터를 저장할 상태 변수
         articles_state = gr.State([])  # 초기값을 빈 리스트로 설정
@@ -174,44 +173,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                 })
 
     def search_and_display(query, country, articles_state):
-        # 상태 메시지 표시
-        status_update = gr.update(value="처리중입니다. 잠시만 기다리세요.", visible=True)
-
-        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(visible=False),  # summary_output
-                ])
-            articles_state = []
-        else:
-            outputs.append(gr.update(value="", visible=False))
-            for idx, comp in enumerate(article_components):
-                if idx < len(articles):
-                    article = articles[idx]
-                    # 이미지 처리 수정
-                    image_url = article['image_url']
-                    if image_url and not image_url.startswith('data:image'):
-                        image_update = gr.update(value=image_url, visible=True)
-                    else:
-                        image_update = gr.update(value=None, visible=False)
-
-                    outputs.extend([
-                        gr.update(visible=True),  # group
-                        gr.update(value=f"### [{article['title']}]({article['link']})"),  # title
-                        image_update,  # image
-                        gr.update(value=f"**요약:** {article['snippet']}"),  # snippet
-                        gr.update(value=f"**출처:** {article['channel']} | **시간:** {article['time']}"),  # info
-                        gr.update(visible=False),  # summary_output
-                    ])
-                else:
+        with gr.Progress() as progress:
+            progress(0, desc="처리중입니다. 잠시만 기다리세요.")
+
+            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
@@ -220,10 +189,39 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                         gr.update(),  # info
                         gr.update(visible=False),  # summary_output
                     ])
-            articles_state = articles  # articles_state 업데이트
-        outputs.append(articles_state)
-        # 상태 메시지 숨김
-        outputs.append(gr.update(visible=False))
+                articles_state = []
+            else:
+                outputs.append(gr.update(value="", visible=False))
+                for idx, comp in enumerate(article_components):
+                    if idx < len(articles):
+                        article = articles[idx]
+                        # 이미지 처리 수정
+                        image_url = article['image_url']
+                        if image_url and not image_url.startswith('data:image'):
+                            image_update = gr.update(value=image_url, visible=True)
+                        else:
+                            image_update = gr.update(value=None, visible=False)
+
+                        outputs.extend([
+                            gr.update(visible=True),  # group
+                            gr.update(value=f"### [{article['title']}]({article['link']})"),  # title
+                            image_update,  # image
+                            gr.update(value=f"**요약:** {article['snippet']}"),  # snippet
+                            gr.update(value=f"**출처:** {article['channel']} | **시간:** {article['time']}"),  # info
+                            gr.update(visible=False),  # summary_output
+                        ])
+                    else:
+                        outputs.extend([
+                            gr.update(visible=False),  # group
+                            gr.update(),  # title
+                            gr.update(),  # image
+                            gr.update(),  # snippet
+                            gr.update(),  # info
+                            gr.update(visible=False),  # summary_output
+                        ])
+                articles_state = articles  # articles_state 업데이트
+            outputs.append(articles_state)
+            # 상태 메시지 숨김 (Progress는 자동으로 사라집니다)
         return outputs
 
     # search_button 클릭 시 업데이트될 출력 컴포넌트 목록 생성
@@ -238,37 +236,32 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         search_outputs.append(comp['info'])
         search_outputs.append(comp['summary_output'])
     search_outputs.append(articles_state)
-    search_outputs.append(status_message)  # 상태 메시지 출력에 추가
+    # status_message 컴포넌트 제거
 
     search_button.click(
         search_and_display,
         inputs=[query, country, articles_state],
-        outputs=search_outputs,
-        show_progress=False  # Gradio의 기본 로딩 표시를 끕니다.
+        outputs=search_outputs
     )
 
     # 분석 버튼 클릭 이벤트 설정
     for idx, comp in enumerate(article_components):
         def create_analyze_function(index=idx):
             def analyze_article(articles):
-                # 상태 메시지 표시
-                status_update = gr.update(value="처리중입니다. 잠시만 기다리세요.", visible=True)
-                if articles and index < len(articles):
-                    article = articles[index]
-                    summary = summarize_article(article['title'], article['snippet'])
-                    # 상태 메시지 숨김
-                    return gr.update(value=summary, visible=True), gr.update(visible=False)
-                else:
-                    # 상태 메시지 숨김
-                    return gr.update(value="기사 정보를 찾을 수 없습니다.", visible=True), gr.update(visible=False)
+                with gr.Progress() as progress:
+                    progress(0, desc="처리중입니다. 잠시만 기다리세요.")
+                    if articles and index < len(articles):
+                        article = articles[index]
+                        summary = summarize_article(article['title'], article['snippet'])
+                        return gr.update(value=summary, visible=True)
+                    else:
+                        return gr.update(value="기사 정보를 찾을 수 없습니다.", visible=True)
             return analyze_article
 
-        # 분석 버튼의 출력에 상태 메시지를 추가합니다.
         comp['analyze_button'].click(
             create_analyze_function(),
             inputs=[articles_state],
-            outputs=[comp['summary_output'], status_message],
-            show_progress=False  # Gradio의 기본 로딩 표시를 끕니다.
+            outputs=comp['summary_output']
         )
 
 iface.launch(auth=("gini", "pick"))