Update src/streamlit_app.py

src/streamlit_app.py

@@ -318,44 +318,53 @@ def generate_wordcloud(keywords_dict):
             st.warning("워드클라우드 설치안되어 있습니다.")
             return None
         try:
-            wc= WordCloud(
+            # 기본 WordCloud 객체 (폰트 경로 없이)
+            wc = WordCloud(
                 width=800,
                 height=400,
-                background_color = 'white',
-                colormap = 'viridis',
+                background_color='white',
+                colormap='viridis',
                 max_font_size=150,
                 random_state=42
-            ).generate_from_frequencies(keywords_dict)
+            )
 
             try:
                 import os
-                script_dir = os.path.dirname(os.path.abspath(__file__))  
-                possible_font_paths = ["NanumGothic.ttf", "이름"]
+                script_dir = os.path.dirname(os.path.abspath(__file__))
+                # 사용자가 루트에 넣은 폰트 파일 이름을 지정합니다.
+                # 만약 다른 이름의 폰트를 사용했다면 이 부분을 수정해주세요. (예: "YourFontName.ttf")
+                possible_font_paths = ["NanumGothic.ttf"] 
 
                 font_path = None
-                for path in possible_font_paths:
-                    candidate = os.path.join(script_dir, fname)
+                for path_segment in possible_font_paths:
+                    candidate = os.path.join(script_dir, path_segment)
                     if os.path.exists(candidate):
                         font_path = candidate
                         break
-
+                
+                # font_path가 성공적으로 찾아진 경우에만 폰트 경로를 포함하여 WordCloud 재생성
                 if font_path:
-                    wc= WordCloud(
+                    wc = WordCloud(
                         font_path=font_path,
                         width=800,
                         height=400,
-                        background_color = 'white',
-                        colormap = 'viridis',
+                        background_color='white',
+                        colormap='viridis',
                         max_font_size=150,
                         random_state=42
                     ).generate_from_frequencies(keywords_dict)
-            except Exception as e:
-                print(f"오류발생 {str(e)}")
+                else:
+                    st.warning(f"지정된 한국어 글꼴 파일({', '.join(possible_font_paths)})을 스크립트 디렉터리에서 찾을 수 없습니다. 워드클라우드가 깨질 수 있습니다.")
             
-            return wc
+            except Exception as e:
+                print(f"글꼴 로딩 중 오류 발생: {str(e)}")
+                st.warning(f"글꼴 로딩 중 예상치 못한 오류가 발생했습니다: {str(e)}") # 사용자에게도 경고 표시
+
+            # 최종적으로 wc 객체 반환 (폰트가 적용되었거나, 기본 객체이거나)
+            return wc.generate_from_frequencies(keywords_dict) if isinstance(wc, WordCloud) else None
         
         except Exception as e:
-            st.error(f"오류발생 {str(e)}")
+            st.error(f"워드클라우드 생성 중 오류발생: {str(e)}")
             return None
 
 # 뉴스 분석 함수
@@ -389,7 +398,7 @@ def generate_article(original_content, prompt_text):
             return "OpenAI API 키가 설정되지 않았습니다."
             
         response = st.session_state.openai_client.chat.completions.create(
-            model="gpt-4",  # 또는 사용 가능한 적절한 모델
+            model="gpt-4.1-nano",  # 또는 사용 가능한 적절한 모델
             messages=[
                 {"role": "system", "content": "당신은 전문적인 뉴스 기자입니다. 주어진 내용을 바탕으로 새로운 기사를 작성해주세요."},
                 {"role": "user", "content": f"다음 내용을 바탕으로 {prompt_text}\n\n{original_content[:1000]}"}
@@ -400,6 +409,26 @@ def generate_article(original_content, prompt_text):
     except Exception as e:
         return f"기사 생성 오류: {str(e)}"
 
+# 여러 제목으로부터 기사 생성하는 함수 추가
+def generate_article_from_titles(titles, prompt_text):
+    try:
+        if not st.session_state.openai_client:
+            return "OpenAI API 키가 설정되지 않았습니다."
+            
+        titles_text = "\n".join([f"- {title}" for title in titles])
+        
+        response = st.session_state.openai_client.chat.completions.create(
+            model="gpt-4.1-nano",  # 또는 사용 가능한 적절한 모델
+            messages=[
+                {"role": "system", "content": "당신은 전문적인 뉴스 기자입니다. 주어진 여러 뉴스 제목을 바탕으로 새로운 통합 기사를 작성해주세요."},
+                {"role": "user", "content": f"다음 뉴스 제목들을 바탕으로 {prompt_text}\n\n{titles_text}"}
+            ],
+            max_tokens=2000
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"기사 생성 오류: {str(e)}"
+
 # OpenAI API를 이용한 이미지 생성 (새로운 버전 방식)
 def generate_image(prompt):
     try:
@@ -922,19 +951,23 @@ elif menu == "새 기사 생성하기":
     if not articles:
         st.warning("저장된 기사가 없습니다. 먼저 '뉴스 기사 크롤링' 메뉴에서 기사를 수집해주세요.")
     else:
-        # 기사 선택
-        titles = [article['title'] for article in articles]
-        selected_title = st.selectbox("원본 기사 선택", titles)
+        # 탭 추가: 단일 기사로 생성 vs 다중 제목으로 생성
+        tab1, tab2 = st.tabs(["단일 기사로 생성", "여러 제목으로 생성"])
         
-        selected_article = next((a for a in articles if a['title'] == selected_title), None)
-        
-        if selected_article:
-            st.write(f"**원본 제목:** {selected_article['title']}")
+        with tab1:
+            # 기존 코드: 단일 기사 선택
+            titles = [article['title'] for article in articles]
+            selected_title = st.selectbox("원본 기사 선택", titles, key="single_article")
             
-            with st.expander("원본 기사 내용"):
-                st.write(selected_article['content'])
+            selected_article = next((a for a in articles if a['title'] == selected_title), None)
             
-            prompt_text ="""다음 기사 양식을 따라서 다시 작성해줘. 
+            if selected_article:
+                st.write(f"**원본 제목:** {selected_article['title']}")
+                
+                with st.expander("원본 기사 내용"):
+                    st.write(selected_article['content'])
+                
+                prompt_text ="""다음 기사 양식을 따라서 다시 작성해줘. 
 역할: 당신은 신문사의 기자입니다.
 작업: 최근 일어난 사건에 대한 보도자료를 작성해야 합니다. 자료는 사실을 기반으로 하며, 객관적이고 정확해야 합니다.
 지침:
@@ -943,52 +976,139 @@ elif menu == "새 기사 생성하기":
 기사 내용은 정확하고 간결하며 설득력 있는 문장으로 구성합니다.
 관련자의 인터뷰를 인용 형태로 넣어주세요.
 위의 정보와 지침을 참고하여 신문 보도자료 형식의 기사를 작성해 주세요"""
+                
+                # 이미지 생성 여부 선택 옵션 추가
+                generate_image_too = st.checkbox("기사 생성 후 이미지도 함께 생성하기", value=True, key="single_image")
+                
+                if st.button("새 기사 생성하기", key="generate_single"):
+                    if st.session_state.openai_client:
+                        with st.spinner("기사를 생성 중입니다..."):
+                            new_article = generate_article(selected_article['content'], prompt_text)
+                            
+                            st.write("**생성된 기사:**")
+                            st.write(new_article)
+                            
+                            # 이미지 생성하기 (옵션이 선택된 경우)
+                            if generate_image_too:
+                                with st.spinner("기사 관련 이미지를 생성 중입니다..."):
+                                    image_prompt = f"""신문기사 제목 "{selected_article['title']}" 을 보고 이미지를 만들어줘 
+                                    이미지에는 다음 요소가 포함되어야 합니다:
+                                    - 기사를 이해할 수 있는 도식
+                                    - 기사 내용과 관련된 텍스트 
+                                    - 심플하게 처리
+                                    """
+                                    
+                                    # 이미지 생성
+                                    image = generate_image(image_prompt)
+                                    
+                                    if isinstance(image, BytesIO):
+                                        st.subheader("생성된 이미지:")
+                                        st.image(image, use_column_width=True)
+                                    else:
+                                        st.error(image)
+                            
+                            # 생성된 기사 저장 옵션
+                            if st.button("생성된 기사 저장", key="save_single"):
+                                new_article_data = {
+                                    'title': f"[생성됨] {selected_article['title']}",
+                                    'source': f"AI 생성 (원본: {selected_article['source']})",
+                                    'date': datetime.now().strftime("%Y-%m-%d %H:%M"),
+                                    'description': new_article[:100] + "...",
+                                    'link': "",
+                                    'content': new_article
+                                }
+                                articles.append(new_article_data)
+                                save_articles(articles)
+                                st.success("생성된 기사가 저장되었습니다!")
+                    else:
+                        st.warning("OpenAI API 키를 사이드바에서 설정해주세요.")
+        
+        with tab2:
+            # 새로운 기능: 여러 제목으로 기사 생성
+            st.subheader("여러 제목을 기반으로 하나의 기사 생성하기")
             
-            # 이미지 생성 여부 선택 옵션 추가
-            generate_image_too = st.checkbox("기사 생성 후 이미지도 함께 생성하기", value=True)
+            # 다중 선택 위젯으로 여러 제목 선택 가능
+            titles = [article['title'] for article in articles]
+            selected_titles = st.multiselect("여러 기사 제목 선택 (2개 이상 권장)", titles)
             
-            if st.button("새 기사 생성하기"):
-                if st.session_state.openai_client:
-                    with st.spinner("기사를 생성 중입니다..."):
-                        new_article = generate_article(selected_article['content'], prompt_text)
-                        
-                        st.write("**생성된 기사:**")
-                        st.write(new_article)
-                        
-                        # 이미지 생성하기 (옵션이 선택된 경우)
-                        if generate_image_too:
-                            with st.spinner("기사 관련 이미지를 생성 중입니다..."):
-                                image_prompt = f"""신문기사 제목 "{selected_article['title']}" 을 보고 이미지를 만들어줘 
-                                이미지에는 다음 요소가 포함되어야 합니다:
-                                - 기사를 이해할 수 있는 도식
-                                - 기사 내용과 관련된 텍스트 
-                                - 심플하게 처리
-                                """
+            if selected_titles:
+                st.write(f"**선택된 제목 수:** {len(selected_titles)}개")
+                
+                with st.expander("선택된 제목 목록"):
+                    for i, title in enumerate(selected_titles):
+                        st.write(f"{i+1}. {title}")
+                
+                multi_prompt_text = """다음 뉴스 제목들을 종합하여 하나의 통합된 기사로 작성해줘.
+역할: 당신은 신문사의 기자입니다.
+작업: 여러 뉴스 제목에서 공통 주제를 파악하고, 이를 종합한 보도자료를 작성해야 합니다.
+지침:
+- 제공된 여러 제목을 종합적으로 분석하여 하나의 일관된 기사를 작성하세요.
+- 기사 제목은 제공된 모든 제목의 핵심 주제를 담아야 합니다.
+- 기사 내용은 제목들이 다루는 모든 주요 주제를 포함해야 합니다.
+- 관련자의 가상 인터뷰를 인용 형태로 넣어주세요.
+- 제공된 제목들의 맥락을 유지하면서 일관성 있는 내러티브를 구성하세요."""
+                
+                # 프롬프트 편집 옵션
+                custom_prompt = st.checkbox("직접 프롬프트 작성하기")
+                if custom_prompt:
+                    multi_prompt_text = st.text_area("프롬프트 직접 입력", multi_prompt_text, height=250)
+                
+                # 이미지 생성 옵션
+                generate_multi_image = st.checkbox("기사 생성 후 이미지도 함께 생성하기", value=True, key="multi_image")
+                
+                if st.button("새 기사 생성하기", key="generate_multi"):
+                    if st.session_state.openai_client:
+                        if len(selected_titles) < 1:
+                            st.error("최소 1개 이상의 제목을 선택해주세요.")
+                        else:
+                            with st.spinner("여러 제목으로부터 기사를 생성 중입니다..."):
+                                # 선택된 제목들을 이용하여 새 기사 생성
+                                new_article = generate_article_from_titles(selected_titles, multi_prompt_text)
                                 
-                                # 이미지 생성
-                                image = generate_image(image_prompt)
+                                st.write("**생성된 기사:**")
+                                st.write(new_article)
                                 
-                                if isinstance(image, BytesIO):
-                                    st.subheader("생성된 이미지:")
-                                    st.image(image, use_column_width=True)
-                                else:
-                                    st.error(image)
-                        
-                        # 생성된 기사 저장 옵션
-                        if st.button("생성된 기사 ���장"):
-                            new_article_data = {
-                                'title': f"[생성됨] {selected_article['title']}",
-                                'source': f"AI 생성 (원본: {selected_article['source']})",
-                                'date': datetime.now().strftime("%Y-%m-%d %H:%M"),
-                                'description': new_article[:100] + "...",
-                                'link': "",
-                                'content': new_article
-                            }
-                            articles.append(new_article_data)
-                            save_articles(articles)
-                            st.success("생성된 기사가 저장되었습니다!")
-                else:
-                    st.warning("OpenAI API 키를 사이드바에서 설정해주세요.")
+                                # 이미지 생성 (옵션이 선택된 경우)
+                                if generate_multi_image:
+                                    with st.spinner("기사 관련 이미지를 생성 중입니다..."):
+                                        combined_titles = " / ".join(selected_titles[:3])  # 처음 3개 제목만 사용
+                                        image_prompt = f"""여러 뉴스를 종합한 기사 "{combined_titles}" 관련 이미지를 만들어줘.
+                                        이미지에는 다음 요소가 포함되어야 합니다:
+                                        - 여러 뉴스의 공통 주제를 시각화한 도식
+                                        - 핵심 키워드나 개념
+                                        - 심플하고 통합된 디자인
+                                        """
+                                        
+                                        # 이미지 생성
+                                        image = generate_image(image_prompt)
+                                        
+                                        if isinstance(image, BytesIO):
+                                            st.subheader("생성된 이미지:")
+                                            st.image(image, use_column_width=True)
+                                        else:
+                                            st.error(image)
+                                
+                                # 생성된 기사 저장 옵션
+                                if st.button("생성된 기사 저장", key="save_multi"):
+                                    # 통합 제목 생성 (첫 번째 제목 + 추가 제목 수)
+                                    if len(selected_titles) > 1:
+                                        combined_title = f"{selected_titles[0]} 외 {len(selected_titles)-1}건 관련 소식"
+                                    else:
+                                        combined_title = selected_titles[0]
+                                    
+                                    new_article_data = {
+                                        'title': f"[여러 제목 통합] {combined_title}",
+                                        'source': "AI 생성 (여러 제목 통합)",
+                                        'date': datetime.now().strftime("%Y-%m-%d %H:%M"),
+                                        'description': new_article[:100] + "...",
+                                        'link': "",
+                                        'content': new_article
+                                    }
+                                    articles.append(new_article_data)
+                                    save_articles(articles)
+                                    st.success("생성된 기사가 저장되었습니다!")
+                    else:
+                        st.warning("OpenAI API 키를 사이드바에서 설정해주세요.")
 
 elif menu == "뉴스 기사 예약하기":
     st.header("뉴스 기사 예약하기")