Update app.py

app.py

@@ -574,9 +574,10 @@ def generate_report(title, info, progress=gr.Progress()):
     try:
         progress(0.1, desc="리포팅 생성 준비 중...")
         
-        # HTML 태그 제거
+        # HTML 태그 제거 및 텍스트 추출
         title_text = re.sub(r'#*\s*\[(.*?)\].*', r'\1', title)
-        info_text = re.sub(r'\*\*(.*?)\*\*', r'\1', info)
+        info_text = re.sub(r'\*\*(.*?)\*\*|\n|AI 요약:|작성자:|시간:|점수:|댓글:', ' ', info)
+        info_text = ' '.join(info_text.split())
         
         progress(0.3, desc="프롬프트 생성 중...")
         
@@ -598,25 +599,13 @@ def generate_report(title, info, progress=gr.Progress()):
             repetition_penalty=1.2
         )
         
-        progress(0.8, desc="결과 처리 중...")
+        progress(1.0, desc="완료!")
+        
+        return response, gr.update(visible=True)
         
-        if response:
-            return [
-                gr.update(value=response, visible=True),
-                gr.update(value="접기", visible=True)
-            ]
-        else:
-            return [
-                gr.update(value="리포팅 생성에 실패했습니다.", visible=True),
-                gr.update(value="접기", visible=True)
-            ]
-            
     except Exception as e:
         print(f"Report generation error: {str(e)}")
-        return [
-            gr.update(value="리포팅 생성 중 오류가 발생했습니다.", visible=True),
-            gr.update(value="접기", visible=True)
-        ]
+        return "리포팅 생성 중 오류가 발생했습니다.", gr.update(visible=True)
 
 def toggle_report(report_content):
     """리포트 표시/숨김 토글"""
@@ -677,9 +666,15 @@ footer {visibility: hidden;}
     border: 1px solid #ddd !important;
     border-radius: 4px !important;
 }
+.report-section {
+    margin-top: 15px;
+    padding: 15px;
+    border-top: 1px solid #eee;
+    background: #f9f9f9;
+}
 
 .hn-article-group {
-    height: auto;
+    height: auto !important;
     min-height: 250px;
     margin-bottom: 20px;
     padding: 15px;
@@ -687,7 +682,7 @@ footer {visibility: hidden;}
     border-radius: 5px;
     background: white;
 }
-.report-section {
+.report-content {
     margin-top: 15px;
     padding: 15px;
     border-top: 1px solid #eee;
@@ -695,7 +690,6 @@ footer {visibility: hidden;}
 }
 """
 
-
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as iface:
     with gr.Tabs():
         # 국가별 탭
@@ -781,25 +775,24 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
                         with gr.Group(visible=False, elem_classes="hn-article-group") as article_group:
                             title = gr.Markdown()
                             info = gr.Markdown()
-                            with gr.Row():
-                                report_button = gr.Button("리포팅 생성", size="sm", variant="primary")
-                            with gr.Column(visible=False) as report_section:
-                                show_report = gr.Button("펼쳐 보기", size="sm")
-                                report_content = gr.Markdown()
+                            report_button = gr.Button("리포팅 생성", size="sm", variant="primary")
+                            report_content = gr.Markdown(visible=False)
+                            show_report = gr.Button("펼쳐 보기", size="sm", visible=False)
                     
                             hn_article_components.append({
                                 'group': article_group,
                                 'title': title,
                                 'info': info,
                                 'report_button': report_button,
-                                'report_section': report_section,
-                                'show_report': show_report,
                                 'report_content': report_content,
+                                'show_report': show_report,
                                 'index': i,
                             })
 
 
 
+
+
  
     
 # 기존 함수들
@@ -971,24 +964,27 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
     ]
 
 
-
-# AI 리포터 탭 이벤트 연결
+    # AI 리포터 탭이벤트 연결
     for comp in hn_article_components:
         comp['report_button'].click(
-            generate_report,
-            inputs=[comp['title'], comp['info']],
-            outputs=[comp['report_content'], comp['show_report']],
+            fn=generate_report,
+            inputs=[
+                comp['title'],
+                comp['info']
+            ],
+            outputs=[
+                comp['report_content'],
+                comp['show_report']
+            ],
             show_progress=True
         )
         
         comp['show_report'].click(
-            toggle_report,
+            fn=toggle_report,
             inputs=[comp['report_content']],
             outputs=[comp['report_content']]
         )
     
-
-
     hn_outputs = [status_message_hn]
     for comp in hn_article_components:
         hn_outputs.extend([
@@ -996,12 +992,12 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css, title="NewsAI 서비스") as
             comp['title'],
             comp['info'],
             comp['report_button'],
-            comp['show_report'],
-            comp['report_content']
+            comp['report_content'],
+            comp['show_report']
         ])
 
     refresh_button.click(
-        refresh_hn_stories,
+        fn=refresh_hn_stories,
         outputs=hn_outputs
     )