Update app.py

app.py

@@ -1,40 +1,71 @@
-import subprocess
-import os
-import matplotlib.font_manager as fm
-import matplotlib.pyplot as plt
-import io
-from PIL import Image
-from datetime import datetime, timedelta
-import gradio as gr
-import yfinance as yf
-from openai import OpenAI
+# A 코드
+# 1. 나눔고딕 폰트 설치 및 적용
+!apt-get install -y fonts-nanum
+!fc-cache -fv
 
-# 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
+import matplotlib.font_manager as fm
+import yfinance as yf
+import numpy as np
+import os
 
 # 나눔고딕 폰트 경로 설정 및 강제 적용
 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  # 마이너스 부호 깨짐 방지
 else:
     print("폰트를 찾을 수 없습니다.")
 
+# 나눔고딕 폰트 강제 적용
+font_prop = fm.FontProperties(fname=font_path)
+plt.rcParams['font.family'] = font_prop.get_name()
+plt.rcParams['axes.unicode_minus'] = False  # 마이너스 부호 깨짐 방지
+
+# 2. yfinance로 주가 데이터 가져오기 (삼성전자 예시)
+ticker = '005930.KS'  # 삼성전자 티커
+stock = yf.Ticker(ticker)
+stock_data = stock.history(period="max")
+
+# 주가 데이터 출력 확인
+print(stock_data.head())  # 주가 데이터 일부 확인
+
+# 3. 주가 데이터 시각화 (pandas 데이터프레임을 numpy 배열로 변환하여 처리)
+plt.figure(figsize=(10, 6))
+
+# 인덱스와 종가 데이터를 numpy 배열로 변환
+dates = stock_data.index.to_numpy()
+closing_prices = stock_data['Close'].to_numpy()
+
+# 주가 데이터 시각화
+plt.plot(dates, closing_prices, label='삼성전자 종가')
+
+# 그래프 제목, 축 라벨을 한글로 설정
+plt.title('삼성전자 주가 추이', fontproperties=font_prop)
+plt.xlabel('날짜', fontproperties=font_prop)
+plt.ylabel('종가', fontproperties=font_prop)
+plt.legend(prop=font_prop)
+
+# B 코드
+import subprocess
+
+# 필요한 라이브러리 설치 및 업데이트
+subprocess.run(["pip", "install", "--upgrade", "pip"])
+subprocess.run(["pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
+
+# 이후 코드 실행
+import requests
+import gradio as gr
+import yfinance as yf
+import matplotlib.pyplot as plt
+import io
+from PIL import Image
+from datetime import datetime, timedelta
+from openai import OpenAI
+import os
+
 # Perplexity AI API 설정
-API_KEY = "여기에_실제_API_키를_입력하세요"  # 여기에 Perplexity AI API 키를 입력하세요.
+API_KEY = "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"  # 여기에 Perplexity AI API 키를 입력하세요.
 
 def get_real_news_summary(company, date):
     # OpenAI 클라이언트 초기화
@@ -42,8 +73,8 @@ def get_real_news_summary(company, date):
     
     # API 요청을 위한 메시지 구성
     messages = [
-        {"role": "system", "content": "당신은 한국어로 주식 뉴스를 요약해주는 도움이 되는 어시스턴트입니다."},
-        {"role": "user", "content": f"{date}에 대한 {company}의 최신 주식 뉴스를 한국어로 요약해줘. 해당 날짜에 특정 뉴스가 없다면, 가장 최근의 관련 정보를 제공해줘."}
+        {"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."}
     ]
     
     try:
@@ -99,7 +130,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 "잘못된 변동 유형입니다.", []
+            return "Invalid change type", []
 
         dates = stock_data.index.to_numpy()
         closing_prices = stock_data['Close'].to_numpy()
@@ -109,14 +140,13 @@ 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', fontproperties=font_prop)
-            plt.axvline(x=index, color=color, linestyle='--', linewidth=1)
+            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.title(f'{input_value} 주가 추이 ({change_type} 표시)', fontproperties=font_prop)
-        plt.xlabel('날짜', fontproperties=font_prop)
-        plt.ylabel('종가', fontproperties=font_prop)
-        plt.legend(prop=font_prop)
+        plt.title(f'{input_value} 주가 추이 ({change_type} 표시)')
+        plt.xlabel('날짜')
+        plt.ylabel('종가')
+        plt.legend()
 
         buf = io.BytesIO()
         plt.savefig(buf, format='png')
@@ -128,7 +158,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"데이터 처리 중 에러가 발생했습니다: {e}", gr.update(choices=[])
+        return f"Error processing data: {e}", gr.update(choices=[])
 
 # Gradio 인터페이스 생성 (3열 레이아웃)
 with gr.Blocks() as demo:
@@ -142,12 +172,13 @@ with gr.Blocks() as demo:
 
             submit_btn = gr.Button("Submit")
 
-            # 예제
+            # 예제 (이전 레이아웃으로 복원)
             examples = [["SK바이오팜"], 
-                        ["Apple"], 
-                        ["삼성전자"], 
-                        ["005930.KS"], 
-                        ["AAPL"]]
+                        ["나스닥 시총 1위"], 
+                        ["나스닥 제약주 시총 1위"], 
+                        ["나스닥 바이오텍 시총 1위"], 
+                        ["코스피 시총 1위"], 
+                        ["코스닥 시총 1위"]]
             gr.Examples(examples=examples, inputs=[input_value])
 
         with gr.Column():  # 그래프를 출력할 두 번째 열