Update app.py

app.py

@@ -146,7 +146,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             search_button = gr.Button("검색")
 
         # 기사 데이터를 저장할 상태 변수
-        articles_state = gr.State()
+        articles_state = gr.State([])  # 초기값을 빈 리스트로 설정
 
         # 최대 10개의 기사에 대한 컴포넌트를 미리 생성합니다.
         article_components = []
@@ -170,88 +170,89 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                     'index': i,
                 })
 
-        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:
+    def search_and_display(query, country, articles_state):
+        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
-                        None,  # title
-                        None,  # image
-                        None,  # snippet
-                        None,  # info
+                        gr.update(),  # title
+                        gr.update(),  # image
+                        gr.update(),  # snippet
+                        gr.update(),  # info
                         gr.update(visible=False),  # summary_output
                     ])
-                outputs.append(gr.update(value=[]))  # 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
-                            None,  # title
-                            None,  # image
-                            None,  # snippet
-                            None,  # info
-                            gr.update(visible=False),  # summary_output
-                        ])
-                outputs.append(gr.update(value=articles))  # articles_state 업데이트
-            return outputs
-
-        # search_button 클릭 시 업데이트될 출력 컴포넌트 목록 생성
-        search_outputs = []
-        error_output = gr.Markdown(visible=False)
-        search_outputs.append(error_output)
-        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['summary_output'])
-        search_outputs.append(articles_state)
-
-        search_button.click(
-            search_and_display,
-            inputs=[query, country],
-            outputs=search_outputs
-        )
-
-        # 분석 버튼 클릭 이벤트 설정
-        for idx, comp in enumerate(article_components):
-            def create_analyze_function(index):
-                def analyze_article(articles):
-                    if 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
+            articles_state = articles  # articles_state 업데이트
+        outputs.append(articles_state)
+        return outputs
+
+    # search_button 클릭 시 업데이트될 출력 컴포넌트 목록 생성
+    search_outputs = []
+    error_output = gr.Markdown(visible=False)
+    search_outputs.append(error_output)
+    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['summary_output'])
+    search_outputs.append(articles_state)
+
+    search_button.click(
+        search_and_display,
+        inputs=[query, country, articles_state],
+        outputs=search_outputs
+    )
+
+    # 분석 버튼 클릭 이벤트 설정
+    for idx, comp in enumerate(article_components):
+        def create_analyze_function(index=idx):
+            def analyze_article(articles):
+                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(idx),
-                inputs=[articles_state],
-                outputs=comp['summary_output']
-            )
+        comp['analyze_button'].click(
+            create_analyze_function(),
+            inputs=[articles_state],
+            outputs=comp['summary_output']
+        )
 
-    iface.launch(auth=("gini", "pick"))
+iface.launch(auth=("gini", "pick"))