Update app.py

app.py

@@ -123,7 +123,6 @@ hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus-08-2024", token=HF_
 
 def summarize_article(title, snippet):
     try:
-        # 기사 제목과 스니펫을 기반으로 요약 생성
         prompt = f"다음 뉴스 제목과 요약을 바탕으로 한국어로 3문장으로 요약하세요:\n제목: {title}\n요약: {snippet}"
         summary = hf_client.text_generation(prompt, max_new_tokens=500)
         return summary
@@ -146,6 +145,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("검색")
 
+        # 기사 데이터를 저장할 상태 변수
+        articles_state = gr.State()
+
         # 최대 10개의 기사에 대한 컴포넌트를 미리 생성합니다.
         article_components = []
         for i in range(10):
@@ -165,93 +167,84 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                     'info': info,
                     'analyze_button': analyze_button,
                     'summary_output': summary_output,
+                    '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:
-                outputs.extend([
-                    gr.update(visible=False),  # group
-                    gr.update(),  # title
-                    gr.update(),  # image
-                    gr.update(),  # snippet
-                    gr.update(),  # info
-                    gr.update(),  # analyze_button
-                    gr.update(visible=False),  # summary_output
-                ])
-            return outputs
-        else:
-            # 기사 컴포넌트 업데이트
-            for idx, comp in enumerate(article_components):
-                if idx < len(articles):
-                    article = articles[idx]
-                    comp['group'].visible = True
-                    comp['title'].value = f"### [{article['title']}]({article['link']})"
-                    if article['image_url'] and not article['image_url'].startswith("data:image"):
-                        comp['image'].value = article['image_url']
-                        comp['image'].visible = True
-                    else:
-                        comp['image'].visible = False
-                    comp['snippet'].value = f"**요약:** {article['snippet']}"
-                    comp['info'].value = f"**출처:** {article['channel']} | **시간:** {article['time']}"
-                    comp['summary_output'].visible = False  # 초기에는 요약 숨김
-
-                    # 분석 버튼 클릭 이벤트 정의
-                    def create_analyze_function(article_title, article_snippet):
-                        def analyze_article():
-                            summary = summarize_article(article_title, article_snippet)
-                            return gr.update(value=summary, visible=True)
-                        return analyze_article
-
-                    comp['analyze_button'].click(
-                        create_analyze_function(article['title'], article['snippet']),
-                        inputs=[],
-                        outputs=comp['summary_output']
-                    )
-
-                    outputs.extend([
-                        gr.update(visible=True),  # group
-                        gr.update(),  # title
-                        gr.update(),  # image
-                        gr.update(),  # snippet
-                        gr.update(),  # info
-                        gr.update(),  # analyze_button
-                        gr.update(visible=False),  # summary_output
-                    ])
-                else:
-                    # 남은 컴포넌트 숨기기
-                    comp['group'].visible = False
+        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:
                     outputs.extend([
                         gr.update(visible=False),  # group
-                        gr.update(),  # title
-                        gr.update(),  # image
-                        gr.update(),  # snippet
-                        gr.update(),  # info
-                        gr.update(),  # analyze_button
+                        None,  # title
+                        None,  # image
+                        None,  # snippet
+                        None,  # 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]
+                        outputs.extend([
+                            gr.update(visible=True),  # group
+                            gr.update(value=f"### [{article['title']}]({article['link']})"),  # title
+                            gr.update(value=article['image_url'] if article['image_url'] else None, visible=bool(article['image_url'])),  # 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 = []
-    search_outputs.append(gr.Markdown(visible=False))  # 오류 메시지 출력용
-    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['analyze_button'])
-        search_outputs.append(comp['summary_output'])
-
-    search_button.click(
-        search_and_display,
-        inputs=[query, country],
-        outputs=search_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
+
+            comp['analyze_button'].click(
+                create_analyze_function(idx),
+                inputs=[articles_state],
+                outputs=comp['summary_output']
+            )
 
 iface.launch(auth=("gini", "pick"))