Update app.py

app.py

@@ -1,41 +1,40 @@
 import subprocess
-import sys
 import os
 import matplotlib.font_manager as fm
 import matplotlib.pyplot as plt
 import io
 from PIL import Image
-from datetime import datetime
+from datetime import datetime, timedelta
 import gradio as gr
 import yfinance as yf
 from openai import OpenAI
-import requests
-
-# 패키지 설치 (필요한 라이브러리 설치 및 업데이트)
-def install_packages():
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "pip"])
-    subprocess.check_call([sys.executable, "-m", "pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
-
-# 패키지 설치
-install_packages()
-
-# 나눔고딕 폰트 다운로드 및 설정
-def download_and_setup_font():
-    font_url = 'https://github.com/naver/nanumfont/blob/master/ttf/NanumGothic.ttf?raw=true'
-    font_path = os.path.join(os.path.expanduser("~"), 'NanumGothic.ttf')
-    response = requests.get(font_url)
-    with open(font_path, 'wb') as f:
-        f.write(response.content)
+
+# 1. 필요한 라이브러리 설치 및 업데이트
+subprocess.run(["pip", "install", "--upgrade", "pip"])
+subprocess.run(["pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
+
+# 2. 나눔고딕 폰트 설치 및 적용
+subprocess.run(["apt-get", "update"])
+subprocess.run(["apt-get", "install", "-y", "fonts-nanum"])
+subprocess.run(["fc-cache", "-fv"])
+
+# 3. 라이브러리 임포트 (폰트 설치 후)
+import matplotlib.font_manager as fm
+import matplotlib.pyplot as plt
+
+# 나눔고딕 폰트 경로 설정 및 강제 적용
+font_path = '/usr/share/fonts/truetype/nanum/NanumGothic.ttf'
+
+if os.path.exists(font_path):
     fm.fontManager.addfont(font_path)
     font_prop = fm.FontProperties(fname=font_path)
     plt.rcParams['font.family'] = font_prop.get_name()
     plt.rcParams['axes.unicode_minus'] = False  # 마이너스 부호 깨짐 방지
-
-# 폰트 설정
-download_and_setup_font()
+else:
+    print("폰트를 찾을 수 없습니다.")
 
 # Perplexity AI API 설정
-API_KEY = "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"  # 여기에 Perplexity AI API 키를 입력하세요.
+API_KEY = "여기에_실제_API_키를_입력하세요"  # 여기에 Perplexity AI API 키를 입력하세요.
 
 def get_real_news_summary(company, date):
     # OpenAI 클라이언트 초기화
@@ -43,8 +42,8 @@ def get_real_news_summary(company, date):
     
     # API 요청을 위한 메시지 구성
     messages = [
-        {"role": "system", "content": "You are a helpful assistant that summarizes stock news in Korean."},
-        {"role": "user", "content": f"Summarize the latest stock news for {company} on {date} in Korean. If there's no specific news for that date, provide the most recent relevant information."}
+        {"role": "system", "content": "당신은 한국어로 주식 뉴스를 요약해주는 도움이 되는 어시스턴트입니다."},
+        {"role": "user", "content": f"{date}에 대한 {company}의 최신 주식 뉴스를 한국어로 요약해줘. 해당 날짜에 특정 뉴스가 없다면, 가장 최근의 관련 정보를 제공해줘."}
     ]
     
     try:
@@ -100,7 +99,7 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
             highlight_data = stock_data[stock_data['Change'] <= -percent_change]
             color = "purple"
         else:
-            return "Invalid change type", []
+            return "잘못된 변동 유형입니다.", []
 
         dates = stock_data.index.to_numpy()
         closing_prices = stock_data['Close'].to_numpy()
@@ -110,13 +109,14 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
         plt.scatter(highlight_data.index, highlight_data['Close'], color=color, label=f'{change_type} 포인트')
 
         for index, row in highlight_data.iterrows():
-            plt.text(index, row['Close'], index.strftime('%Y-%m-%d'), fontsize=10, fontweight='bold', color=color, ha='right')
-            plt.axvline(x=index, color=color, linestyle='--', linewidth=1)  # x축과의 연결선 점선으로 표시
+            plt.text(index, row['Close'], index.strftime('%Y-%m-%d'), fontsize=10, fontweight='bold', color=color, ha='right', fontproperties=font_prop)
+            plt.axvline(x=index, color=color, linestyle='--', linewidth=1)
 
-        plt.title(f'{input_value} 주가 추이 ({change_type} 표시)')
-        plt.xlabel('날짜')
-        plt.ylabel('종가')
-        plt.legend()
+        # 그래프 제목, 축 라벨을 한글로 설정
+        plt.title(f'{input_value} 주가 추이 ({change_type} 표시)', fontproperties=font_prop)
+        plt.xlabel('날짜', fontproperties=font_prop)
+        plt.ylabel('종가', fontproperties=font_prop)
+        plt.legend(prop=font_prop)
 
         buf = io.BytesIO()
         plt.savefig(buf, format='png')
@@ -128,7 +128,7 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
 
         return img, gr.update(choices=highlight_dates)
     except Exception as e:
-        return f"Error processing data: {e}", gr.update(choices=[])
+        return f"데이터 처리 중 에러가 발생했습니다: {e}", gr.update(choices=[])
 
 # Gradio 인터페이스 생성 (3열 레이아웃)
 with gr.Blocks() as demo: