Update app.py

app.py

@@ -135,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():
@@ -144,6 +144,9 @@ 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("검색")
 
+        # 상태 메시지 컴포넌트 제거 (Progress를 사용하므로 불필요)
+        # status_message = gr.Markdown(visible=False)
+
         # 기사 데이터를 저장할 상태 변수
         articles_state = gr.State([])  # 초기값을 빈 리스트로 설정
 
@@ -170,47 +173,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                 })
 
     def search_and_display(query, country, articles_state):
-        # Progress 객체 생성
-        progress = gr.Progress(track_tqdm=True)
-        progress(0, desc="처리중입니다. 잠시만 기다리세요.")
-
-        error_message, articles = serphouse_search(query, country)
-        progress(50)  # 진행률 50%로 업데이트
-
-        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
@@ -219,9 +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)
-        progress(100)  # 진행률 100%로 업데이트
+                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 클릭 시 업데이트될 출력 컴포넌트 목록 생성
@@ -236,6 +236,7 @@ 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)
+    # status_message 컴포넌트 제거
 
     search_button.click(
         search_and_display,
@@ -247,16 +248,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
     for idx, comp in enumerate(article_components):
         def create_analyze_function(index=idx):
             def analyze_article(articles):
-                progress = gr.Progress(track_tqdm=True)
-                progress(0, desc="처리중입니다. 잠시만 기다리세요.")
-                if articles and index < len(articles):
-                    article = articles[index]
-                    summary = summarize_article(article['title'], article['snippet'])
-                    progress(100)  # 진행률 100%로 업데이트
-                    return gr.update(value=summary, visible=True)
-                else:
-                    progress(100)  # 진행률 100%로 업데이트
-                    return gr.update(value="기사 정보를 찾을 수 없습니다.", visible=True)
+                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(