Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import matplotlib.pyplot as plt
+import seaborn as sns
 from transformers import pipeline
 import openai
 import os
@@ -8,6 +9,11 @@ import os
 # OpenAI API 키 설정 (환경변수에서 가져오는 방법)
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
+# 한글 폰트 설정
+import matplotlib
+matplotlib.rcParams['font.family'] = 'NanumGothic'  # 한글 폰트를 설정합니다.
+matplotlib.rcParams['axes.unicode_minus'] = False  # 마이너스 기호 깨짐 방지
+
 # 네이버 뉴스 API를 통해 실제 뉴스 기사 가져오기
 def fetch_naver_news(query, display=5):
     client_id = "I_8koTJh3R5l4wLurQbG"  # 네이버 개발자 센터에서 발급받은 Client ID
@@ -108,13 +114,17 @@ def analyze_news_political_viewpoint(query):
 
 # 성향 분포 시각화 (막대 그래프)
 def visualize_sentiment_distribution(sentiment_counts):
-    fig, ax = plt.subplots()
+    fig, ax = plt.subplots(figsize=(8, 5))
     labels = list(sentiment_counts.keys())
     sizes = list(sentiment_counts.values())
-    ax.bar(labels, sizes, color=["blue", "red", "gray"])
-    ax.set_xlabel('성향')
-    ax.set_ylabel('건수')
-    ax.set_title('뉴스 성향 분포')
+
+    # 색상 설정 (부드러운 팔레트)
+    color_palette = sns.color_palette("pastel")[0:len(sizes)]
+    
+    ax.bar(labels, sizes, color=color_palette)
+    ax.set_xlabel('성향', fontsize=14)
+    ax.set_ylabel('건수', fontsize=14)
+    ax.set_title('뉴스 성향 분포', fontsize=16)
     st.pyplot(fig)
 
 # Streamlit 애플리케이션