Update src/streamlit_app.py

src/streamlit_app.py

@@ -641,9 +641,25 @@ elif menu == "기사 분석하기":
                         with keyword_tab1:
                             keywords = analyze_keywords(selected_article['content'])
                             
-                            # 시각화
+                            # Plotly를 사용한 시각화
                             df = pd.DataFrame(keywords, columns=['단어', '빈도수'])
-                            st.bar_chart(df.set_index('단어'))
+                            fig = go.Figure(data=[
+                                go.Bar(
+                                    x=df['단어'],
+                                    y=df['빈도수'],
+                                    marker_color='rgb(55, 83, 109)'
+                                )
+                            ])
+                            
+                            fig.update_layout(
+                                title='키워드 빈도 분석',
+                                xaxis_title='키워드',
+                                yaxis_title='빈도수',
+                                height=500,
+                                margin=dict(l=50, r=50, t=80, b=50)
+                            )
+                            
+                            st.plotly_chart(fig, use_container_width=True)
                             
                             st.write("**주요 키워드:**")
                             for word, count in keywords:
@@ -728,11 +744,22 @@ elif menu == "기사 분석하기":
                                         }"""},
                                         {"role": "user", "content": f"다음 뉴스 기사를 분석해 주세요:\n\n제목: {selected_article['title']}\n\n내용: {selected_article['content'][:1500]}"}
                                     ],
-                                    max_tokens=800
+                                    max_tokens=800,
+                                    response_format={ "type": "json_object" }  # JSON 응답 형식 강제
                                 )
                                 
+                                # 응답 내용 확인 및 디버깅
+                                content = response.choices[0].message.content
+                                logging.info(f"API 응답: {content}")
+                                
                                 # JSON 파싱
-                                analysis_result = json.loads(response.choices[0].message.content)
+                                try:
+                                    analysis_result = json.loads(content)
+                                except json.JSONDecodeError as e:
+                                    logging.error(f"JSON 파싱 오류: {str(e)}")
+                                    logging.error(f"파싱 시도한 내용: {content}")
+                                    st.error("API 응답을 파싱하는 중 오류가 발생했습니다. 응답 형식이 올바르지 않습니다.")
+                                    return
                                 
                                 # 결과 시각화
                                 st.subheader("감정 분석 결과")