Update src/streamlit_app.py

src/streamlit_app.py

@@ -398,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]}"}
@@ -409,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:
@@ -931,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 ="""다음 기사 양식을 따라서 다시 작성해줘. 
 역할: 당신은 신문사의 기자입니다.
 작업: 최근 일어난 사건에 대한 보도자료를 작성해야 합니다. 자료는 사실을 기반으로 하며, 객관적이고 정확해야 합니다.
 지침:
@@ -952,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("뉴스 기사 예약하기")