Update app.py

app.py

@@ -326,8 +326,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                     query = gr.Textbox(label="검색어")
                     country = gr.Dropdown(MAJOR_COUNTRIES, label="국가", value="South Korea")
                 
+                # 검색 상태 메시지    
                 status_message = gr.Markdown("", visible=True)
+                
+                # 번역 결과를 보여주는 컴포넌트
                 translated_query_display = gr.Markdown(visible=False)
+                
                 search_button = gr.Button("검색", variant="primary")
                 
                 progress = gr.Progress()
@@ -381,10 +385,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                         })
 
     def search_and_display(query, country, articles_state, progress=gr.Progress()):
-        # 기존 search_and_display 함수 코드 유지
+        # 검색 상태 메시지 업데이트
         status_msg = "검색을 진행중입니다. 잠시만 기다리세요..."
         
         progress(0, desc="검색어 번역 중...")
+        
+        # 검색어 번역
         translated_query = translate_query(query, country)
         translated_display = f"**원본 검색어:** {query}\n**번역된 검색어:** {translated_query}" if translated_query != query else f"**검색어:** {query}"
         
@@ -393,8 +399,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
         progress(0.5, desc="결과 처리 중...")
         
         outputs = []
-        outputs.append(gr.update(value=status_msg, visible=True))
-        outputs.append(gr.update(value=translated_display, visible=True))
+        outputs.append(gr.update(value=status_msg, visible=True))  # 상태 메시지
+        outputs.append(gr.update(value=translated_display, visible=True))  # 번역 결과
         
         if error_message:
             outputs.append(gr.update(value=error_message, visible=True))
@@ -414,6 +420,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                     image_url = article['image_url']
                     image_update = gr.update(value=image_url, visible=True) if image_url and not image_url.startswith('data:image') else gr.update(value=None, visible=False)
                     
+                    # 요약 내용 한글 번역
                     korean_summary = translate_to_korean(article['snippet'])
                     
                     outputs.extend([
@@ -432,74 +439,98 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
 
         progress(1.0, desc="완료!")
         outputs.append(articles_state)
+        
+        # 검색 완료 후 상태 메시지 숨김
         outputs[0] = gr.update(value="", visible=False)
         
         return outputs
 
     def search_global(query, articles_state_global, progress=gr.Progress()):
-        status_msg = "전세계 검색을 진행중입니다. 잠시만 기다리세요..."
+        status_msg = "전세계 검색을 진행중입니다. 실시간으로 결과가 출력됩니다..."
         all_results = []
+        displayed_count = 0
+        
+        # 초기 출력 설정
+        outputs = []
+        outputs.append(gr.update(value=status_msg, visible=True))
+        outputs.append(gr.update(value=f"**검색어:** {query}", visible=True))
+        
+        # 1000개의 결과 컴포넌트를 초기에 모두 숨김 처리
+        for _ in global_article_components:
+            outputs.extend([
+                gr.update(visible=False), gr.update(), gr.update(),
+                gr.update(), gr.update()
+            ])
+        outputs.append([])  # articles_state_global
         
-        progress(0, desc="전세계 검색 시작...")
+        yield outputs  # 초기 상태 출력
         
+        # 국가별 검색 및 실시간 결과 출력
         for idx, (country, location) in enumerate(COUNTRY_LOCATIONS.items()):
-            progress(idx / len(COUNTRY_LOCATIONS), f"{country} 검색 중...")
+            progress(idx / len(COUNTRY_LOCATIONS), f"{country} 검색 중... ({idx + 1}/{len(COUNTRY_LOCATIONS)})")
+            
             try:
                 error_message, articles = serphouse_search(query, country)
                 if not error_message and articles:
+                    # 해당 국가의 결과에 국가 정보 추가
                     for article in articles:
                         article['source_country'] = country
+                    
+                    # 새로운 결과 추가 및 정렬
                     all_results.extend(articles)
+                    sorted_results = sorted(all_results, key=lambda x: x.get('time', ''), reverse=True)
+                    
+                    # 중복 제거
+                    seen_urls = set()
+                    unique_results = []
+                    for article in sorted_results:
+                        url = article.get('link', '')
+                        if url not in seen_urls:
+                            seen_urls.add(url)
+                            unique_results.append(article)
+                    
+                    # 최대 1000개까지만 유지
+                    unique_results = unique_results[:1000]
+                    
+                    # 새로운 결과 출력
+                    outputs = []
+                    outputs.append(gr.update(value=f"전세계 검색 진행 중... ({idx + 1}/{len(COUNTRY_LOCATIONS)} 국가 완료)", visible=True))
+                    outputs.append(gr.update(value=f"**검색어:** {query}\n**현재까지 발견된 뉴스:** {len(unique_results)}건", visible=True))
+                    
+                    # 결과 컴포넌트 업데이트
+                    for idx, comp in enumerate(global_article_components):
+                        if idx < len(unique_results):
+                            article = unique_results[idx]
+                            image_url = article.get('image_url', '')
+                            image_update = gr.update(value=image_url, visible=True) if image_url and not image_url.startswith('data:image') else gr.update(value=None, visible=False)
+                            
+                            korean_summary = translate_to_korean(article['snippet'])
+                            
+                            outputs.extend([
+                                gr.update(visible=True),
+                                gr.update(value=f"### [{article['title']}]({article['link']})"),
+                                image_update,
+                                gr.update(value=f"**요약:** {article['snippet']}\n\n**한글 요약:** {korean_summary}"),
+                                gr.update(value=f"**출처:** {article['channel']} | **국가:** {article['source_country']} | **시간:** {article['time']}")
+                            ])
+                        else:
+                            outputs.extend([
+                                gr.update(visible=False), gr.update(), gr.update(),
+                                gr.update(), gr.update()
+                            ])
+                    
+                    outputs.append(unique_results)
+                    yield outputs
+                    
             except Exception as e:
                 print(f"Error searching {country}: {str(e)}")
                 continue
-
-        # 결과 정렬 및 중복 제거
-        seen_urls = set()
-        unique_results = []
-        for article in sorted(all_results, key=lambda x: x.get('time', ''), reverse=True):
-            url = article.get('link', '')
-            if url not in seen_urls:
-                seen_urls.add(url)
-                unique_results.append(article)
-
-        # 최대 1000개로 제한
-        final_results = unique_results[:1000]
         
-        outputs = []
-        outputs.append(gr.update(value=status_msg, visible=True))
-        outputs.append(gr.update(value=f"**검색어:** {query}", visible=True))
-        
-        progress(0.5, desc="결과 처리 중...")
-        
-        for idx, comp in enumerate(global_article_components):
-            if idx < len(final_results):
-                article = final_results[idx]
-                image_url = article.get('image_url', '')
-                image_update = gr.update(value=image_url, visible=True) if image_url and not image_url.startswith('data:image') else gr.update(value=None, visible=False)
-                
-                korean_summary = translate_to_korean(article['snippet'])
-                
-                outputs.extend([
-                    gr.update(visible=True),
-                    gr.update(value=f"### [{article['title']}]({article['link']})"),
-                    image_update,
-                    gr.update(value=f"**요약:** {article['snippet']}\n\n**한글 요약:** {korean_summary}"),
-                    gr.update(value=f"**출처:** {article['channel']} | **국가:** {article['source_country']} | **시간:** {article['time']}")
-                ])
-            else:
-                outputs.extend([
-                    gr.update(visible=False), gr.update(), gr.update(),
-                    gr.update(), gr.update()
-                ])
-        
-        progress(1.0, desc="완료!")
-        outputs.append(final_results)
-        outputs[0] = gr.update(value="", visible=False)
-        
-        return outputs
+        # 최종 상태 업데이트
+        final_status = f"검색 완료! 총 {len(unique_results)}개의 뉴스가 발견되었습니다."
+        outputs[0] = gr.update(value=final_status, visible=True)
+        yield outputs
 
-    # 국가별 검색 이벤트 연결
     search_outputs = [
         status_message,
         translated_query_display,