Update app.py

app.py

@@ -1,9 +1,4 @@
 import subprocess
-
-# 필요한 라이브러리 설치 및 업데이트
-subprocess.run(["pip", "install", "--upgrade", "pip"])
-subprocess.run(["pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
-
 import os
 import matplotlib.font_manager as fm
 import matplotlib.pyplot as plt
@@ -16,13 +11,171 @@ from PIL import Image
 from datetime import datetime, timedelta
 from openai import OpenAI
 
-# ... (중간 코드는 그대로 유지)
+# 1. 필요한 라이브러리 설치 및 업데이트
+def install_libraries():
+    try:
+        subprocess.run(["pip", "install", "--upgrade", "pip"])
+        subprocess.run(["pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
+    except Exception as e:
+        print(f"라이브러리 설치 중 오류가 발생했습니다: {e}")
+
+install_libraries()
+
+# 2. 나눔고딕 폰트 설치 및 적용
+def install_nanum_font():
+    try:
+        subprocess.run(["apt-get", "update"], check=True)
+        subprocess.run(["apt-get", "install", "-y", "fonts-nanum"], check=True)
+        subprocess.run(["fc-cache", "-fv"], check=True)
+    except Exception as e:
+        print(f"폰트 설치 중 오류가 발생했습니다: {e}")
+
+install_nanum_font()
+
+# 나눔고딕 폰트 경로 설정 및 강제 적용
+font_path = '/usr/share/fonts/truetype/nanum/NanumGothic.ttf'
+if os.path.exists(font_path):
+    fm.fontManager.addfont(font_path)
+else:
+    print("폰트를 찾을 수 없습니다.")
+# 나눔고딕 폰트 강제 적용
+font_prop = fm.FontProperties(fname=font_path)
+plt.rcParams['font.family'] = font_prop.get_name()
+plt.rcParams['axes.unicode_minus'] = False  # 마이너스 부호 깨짐 방지
+
+# 3. Perplexity AI API 설정
+API_KEY = "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"
+
+# 뉴스 요약을 가져오는 함수
+def get_real_news_summary(company, date):
+    # OpenAI 클라이언트 초기화
+    client = OpenAI(api_key=API_KEY, base_url="https://api.perplexity.ai")
+    
+    # 날짜 형식을 맞춰주기 위한 처리
+    target_date = datetime.strptime(date, '%Y-%m-%d')
+    start_date = (target_date - timedelta(days=1)).strftime('%Y-%m-%d')
+    end_date = (target_date + timedelta(days=1)).strftime('%Y-%m-%d')
+
+    # API 요청을 위한 메시지 구성
+    messages = [
+        {"role": "system", "content": "You are a helpful assistant that summarizes stock news strictly in Korean."},
+        {"role": "user", "content": f"Summarize the stock news for {company} between {start_date} and {end_date} in Korean. Only focus on news within this date range."}
+    ]
+
+    try:
+        # API 요청
+        response = client.chat.completions.create(
+            model="llama-3.1-sonar-large-128k-online",
+            messages=messages
+        )
+
+        # 응답에서 요약 추출
+        summary = response.choices[0].message.content
+
+        # 한글, 숫자, 공백, 특수 기호만 남기는 정규 표현식
+        korean_only_summary = re.sub(r'[^\w\s#.,!%()\-\[\]]', '', summary)
+
+        # ##로 시작하는 부분을 **으로 감싸서 Bold 처리
+        formatted_summary = re.sub(r'##\s*(.+)', r'**\1**', korean_only_summary)
+
+        return formatted_summary
+    except Exception as e:
+        return f"뉴스 요약 중 에러가 발생했습니다: {str(e)}"
+
+# 뉴스 요약을 가져오는 함수
+def handle_click(company_name, date_clicked):
+    return get_real_news_summary(company_name, date_clicked)
+
+# Gradio에서 사용할 함수 (뉴스 요약 포함)
+def update_news(input_value, selected_date):
+    if selected_date == "" or selected_date is None:
+        return "날짜를 선택해주세요."
+    else:
+        # 종목명을 가져와서 Perplexity로 검색
+        ticker = name_to_ticker.get(input_value, input_value)
+        company_name = input_value if ticker == input_value else list(name_to_ticker.keys())[list(name_to_ticker.values()).index(ticker)]
+        return handle_click(company_name, selected_date)
+
+# 종목명과 티커를 매핑하는 딕셔너리 확장 및 조건별 종목 매핑
+name_to_ticker = {
+    "삼성전자": "005930.KS",
+    "SK바이오팜": "326030.KS",
+    "Apple": "AAPL",
+    "Nvidia": "NVDA",
+    "Vertex": "VRTX",
+    "현대차": "005380.KS",
+    "카카오": "035720.KS",
+    "LG화학": "051910.KS",
+    "셀트리온": "068270.KS",
+    "네이버": "035420.KS",
+    "에코프로비엠": "247540.KS",
+    "알테오젠": "196170.KQ",
+    # 최신 시가총액 상위 종목을 반영하여 업데이트
+    "나스닥 시총 1위": "AAPL",  # Apple
+    "나스닥 바이오텍 시총 1위": "VRTX",  # Vertex Pharmaceuticals
+    "나스닥 헬스케어 시총 1위": "LLY",  # Eli Lilly
+    "코스피 시총 1위": "005930.KS",  # 삼성전자
+    "코스닥 시총 1위": "196170.KQ",  # 알테오젠
+}
+
+# 주가 데이터를 가져오고 조건에 맞는 날짜와 그래프를 반환하는 함수
+def display_stock_with_highlight(input_value, change_type, percent_change):
+    try:
+        # 입력값을 티커로 변환
+        ticker = name_to_ticker.get(input_value, input_value)
+        stock = yf.Ticker(ticker)
+        stock_data = stock.history(period="5y")  # 최근 5년 데이터로 제한
+
+        if stock_data.empty:
+            return "주가 데이터를 찾을 수 없습니다.", []
+
+        stock_data['Change'] = stock_data['Close'].pct_change() * 100
+
+        percent_change = float(percent_change)
+
+        if change_type == "상승":
+            highlight_data = stock_data[stock_data['Change'] >= percent_change]
+            color = "darkorange"
+        elif change_type == "하락":
+            highlight_data = stock_data[stock_data['Change'] <= -percent_change]
+            color = "purple"
+        else:
+            return "Invalid change type", []
+
+        dates = stock_data.index.to_numpy()
+        closing_prices = stock_data['Close'].to_numpy()
+
+        plt.figure(figsize=(10, 6))
+        plt.plot(dates, closing_prices, color='gray', label=input_value)
+        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축과의 연결선 점선으로 표시
+
+        # 종목명 + '주가 추이'로 제목 설정 (종목명 기반)
+        company_name = list(name_to_ticker.keys())[list(name_to_ticker.values()).index(ticker)]
+        plt.title(f'{company_name} 주가 추이', fontproperties=font_prop)
+        plt.xlabel('날짜', fontproperties=font_prop)
+        plt.ylabel('종가', fontproperties=font_prop)
+        plt.legend()
+
+        buf = io.BytesIO()
+        plt.savefig(buf, format='png')
+        plt.close()
+        buf.seek(0)
+        img = Image.open(buf)
+
+        highlight_dates = highlight_data.index.strftime('%Y-%m-%d').tolist()
+
+        return img, gr.update(choices=highlight_dates)
+    except Exception as e:
+        return f"Error processing data: {e}", gr.update(choices=[])
 
 # Gradio 인터페이스 생성 (3열 레이아웃)
 with gr.Blocks() as demo:
     gr.Markdown("## 주가 그래프와 뉴스 요약")
 
-    # CSS로 3열 뉴스 요약 타이틀이 달린 회색 네모 칸 생성
     demo.css("#news_output_box { background-color: #f0f0f0; padding: 20px; }")
 
     with gr.Row():
@@ -33,7 +186,7 @@ with gr.Blocks() as demo:
 
             submit_btn = gr.Button("Submit")
 
-            # 예제 (이전 레이아웃으로 복원)
+            # 예제
             examples = [["SK바이오팜"],
                         ["나스닥 시총 1위"],
                         ["나스닥 헬스케어 시총 1위"],
@@ -47,7 +200,6 @@ with gr.Blocks() as demo:
             date_dropdown = gr.Dropdown(label="조건에 해당하는 날짜 선택", choices=[])
 
         with gr.Column():  # 뉴스 요약을 출력할 세 번째 열
-            # news_output을 gr.Markdown으로 변경, 빈 칸을 회색 배경으로 설정
             news_output = gr.Markdown(label="뉴스 요약", value="", elem_id="news_output_box")
 
     # Submit 버튼 클릭 시 그래프 및 날짜 드롭다운 업데이트