Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,6 +19,9 @@ import schedule
 import threading
 import matplotlib.pyplot as plt
 import kss  # KoNLPy 대신 KSS 사용
+from PIL import Image
+import base64
+from io import BytesIO
 
 # 워드클라우드 추가
 try:
@@ -357,7 +360,7 @@ def generate_article(original_content, prompt_text):
             return "OpenAI API 키가 설정되지 않았습니다."
             
         response = st.session_state.openai_client.chat.completions.create(
-            model="gpt-4.1-nano",  # 또는 사용 가능한 적절한 모델
+            model="gpt-4",  # 또는 사용 가능한 적절한 모델
             messages=[
                 {"role": "system", "content": "당신은 전문적인 뉴스 기자입니다. 주어진 내용을 바탕으로 새로운 기사를 작성해주세요."},
                 {"role": "user", "content": f"다음 내용을 바탕으로 {prompt_text}\n\n{original_content[:1000]}"}
@@ -374,13 +377,31 @@ def generate_image(prompt):
         if not st.session_state.openai_client:
             return "OpenAI API 키가 설정되지 않았습니다."
             
-        response = st.session_state.openai_client.images.generate(
-            model="gpt-image-1",  # 또는 "dall-e-2"
+        # GPT Image 1 모델로 이미지 생성
+        result = st.session_state.openai_client.images.generate(
+            model="gpt-image-1",  # 새로운 모델명 사용
             prompt=prompt,
-            n=1,
             size="1024x1024"
         )
-        return response.data[0].url
+        
+        # base64 이미지 데이터를 디코딩
+        image_base64 = result.data[0].b64_json
+        image_bytes = base64.b64decode(image_base64)
+        
+        # BytesIO 객체로 변환
+        image = BytesIO(image_bytes)
+        
+        # PIL Image로 변환하여 크기 조정 (선택사항)
+        pil_image = Image.open(image)
+        pil_image = pil_image.resize((800, 800), Image.LANCZOS)  # 크기 조정
+        
+        # 다시 BytesIO로 변환
+        output = BytesIO()
+        pil_image.save(output, format="JPEG", quality=80, optimize=True)
+        output.seek(0)
+        
+        return output
+        
     except Exception as e:
         return f"이미지 생성 오류: {str(e)}"
 
@@ -662,9 +683,18 @@ elif menu == "기사 분석하기":
                         with st.spinner("기사의 감정을 분석 중입니다..."):
                             try:
                                 response = st.session_state.openai_client.chat.completions.create(
-                                    model="gpt-4",  # 또는 사용 가능한 적절한 모델
+                                    model="gpt-4",
                                     messages=[
-                                        {"role": "system", "content": "당신은 텍스트의 감정과 논조를 분석하는 전문가입니다. 다음 뉴스 기사의 감정과 논조를 분석하고, '긍정적', '부정적', '중립적' 중 하나로 분류해 주세요. 또한 기사에서 드러나는 핵심 감정 키워드를 5개 추출하고, 각 키워드별로 1-10 사이의 강도 점수를 매겨주세요. JSON 형식으로 다음과 같이 응답해주세요: {'sentiment': '긍정적/부정적/중립적', 'reason': '이유 설명...', 'keywords': [{'word': '키워드1', 'score': 8}, {'word': '키워드2', 'score': 7}, ...]}"},
+                                        {"role": "system", "content": """당신은 텍스트의 감정과 논조를 분석하는 전문가입니다. 
+                                        다음 뉴스 기사의 감정과 논조를 분석하고, 반드시 아래 형식의 JSON으로 응답해주세요:
+                                        {
+                                            "sentiment": "긍정적/부정적/중립적",
+                                            "reason": "이유 설명...",
+                                            "keywords": [
+                                                {"word": "키워드1", "score": 8},
+                                                {"word": "키워드2", "score": 7}
+                                            ]
+                                        }"""},
                                         {"role": "user", "content": f"다음 뉴스 기사를 분석해 주세요:\n\n제목: {selected_article['title']}\n\n내용: {selected_article['content'][:1500]}"}
                                     ],
                                     max_tokens=800
@@ -867,7 +897,6 @@ elif menu == "새 기사 생성하기":
                         # 이미지 생성하기 (옵션이 선택된 경우)
                         if generate_image_too:
                             with st.spinner("기사 관련 이미지를 생성 중입니다..."):
-                                # 이미지 생성 프롬프트 준비
                                 image_prompt = f"""신문기사 제목 "{selected_article['title']}" 을 보고 이미지를 만들어줘 
                                 이미지에는 다음 요소가 포함되어야 합니다:
                                 - 기사를 이해할 수 있는 도식
@@ -876,13 +905,13 @@ elif menu == "새 기사 생성하기":
                                 """
                                 
                                 # 이미지 생성
-                                image_url = generate_image(image_prompt)
+                                image = generate_image(image_prompt)
                                 
-                                if image_url and not image_url.startswith("이미지 생성 오류"):
+                                if isinstance(image, BytesIO):
                                     st.subheader("생성된 이미지:")
-                                    st.image(image_url)
+                                    st.image(image, use_column_width=True)
                                 else:
-                                    st.error(image_url)
+                                    st.error(image)
                         
                         # 생성된 기사 저장 옵션
                         if st.button("생성된 기사 저장"):