Update app.py

app.py

@@ -6,17 +6,21 @@ from datetime import datetime, timedelta
 from huggingface_hub import InferenceClient  # LLM 사용을 위해 필요
 
 # 환경 변수에서 API 키 가져오기
-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())
+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"
+]
 
 def search_serphouse(query, country, page=1, num_result=10):
     url = "https://api.serphouse.com/serp/live"
@@ -25,14 +29,11 @@ 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": loc_id,  # 위치 ID 사용
+            "loc": country,  # 국가 이름을 직접 사용합니다.
             "lang": "en",
             "device": "desktop",
             "serp_type": "news",
@@ -135,7 +136,7 @@ footer {
 """
 
 # Gradio 인터페이스 구성
-with gr.Blocks(css=css, title="NewsAI 서비스") as iface:
+with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
     gr.Markdown("검색어를 입력하고 원하는 국가를 선택하면, 검색어와 일치하는 24시간 이내 뉴스를 최대 10개 출력합니다.")
 
     with gr.Column():
@@ -144,8 +145,8 @@ with gr.Blocks(css=css, title="NewsAI 서비스") as iface:
             country = gr.Dropdown(MAJOR_COUNTRIES, label="국가", value="South Korea")
             search_button = gr.Button("검색")
 
-        # 상태 메시지 컴포넌트 제거 (Progress를 사용하므로 불필요)
-        # status_message = gr.Markdown(visible=False)
+        # 상태 메시지 컴포넌트 추가
+        status_message = gr.Markdown(visible=False)
 
         # 기사 데이터를 저장할 상태 변수
         articles_state = gr.State([])  # 초기값을 빈 리스트로 설정
@@ -173,14 +174,44 @@ with gr.Blocks(css=css, title="NewsAI 서비스") as iface:
                 })
 
     def search_and_display(query, country, articles_state):
-        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:
+        # 상태 메시지 표시
+        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:
                     outputs.extend([
                         gr.update(visible=False),  # group
                         gr.update(),  # title
@@ -189,39 +220,10 @@ with gr.Blocks(css=css, title="NewsAI 서비스") as iface:
                         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:
-                        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는 자동으로 사라집니다)
+            articles_state = articles  # articles_state 업데이트
+        outputs.append(articles_state)
+        # 상태 메시지 숨김
+        outputs.append(gr.update(visible=False))
         return outputs
 
     # search_button 클릭 시 업데이트될 출력 컴포넌트 목록 생성
@@ -236,32 +238,37 @@ with gr.Blocks(css=css, title="NewsAI 서비스") as iface:
         search_outputs.append(comp['info'])
         search_outputs.append(comp['summary_output'])
     search_outputs.append(articles_state)
-    # status_message 컴포넌트 제거
+    search_outputs.append(status_message)  # 상태 메시지 출력에 추가
 
     search_button.click(
         search_and_display,
         inputs=[query, country, articles_state],
-        outputs=search_outputs
+        outputs=search_outputs,
+        show_progress=False  # Gradio의 기본 로딩 표시를 끕니다.
     )
 
     # 분석 버튼 클릭 이벤트 설정
     for idx, comp in enumerate(article_components):
         def create_analyze_function(index=idx):
             def analyze_article(articles):
-                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)
+                # 상태 메시지 표시
+                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)
             return analyze_article
 
+        # 분석 버튼의 출력에 상태 메시지를 추가합니다.
         comp['analyze_button'].click(
             create_analyze_function(),
             inputs=[articles_state],
-            outputs=comp['summary_output']
+            outputs=[comp['summary_output'], status_message],
+            show_progress=False  # Gradio의 기본 로딩 표시를 끕니다.
         )
 
 iface.launch(auth=("gini", "pick"))