Update app.py

app.py

@@ -1,52 +1,165 @@
-import yfinance as yf
+# 필요한 라이브러리 설치 및 업데이트를 최상단에 배치
 import subprocess
+
+# 필요한 라이브러리 설치 및 업데이트
+subprocess.run(["pip", "install", "--upgrade", "pip"])
+subprocess.run(["pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
+
+# 라이브러리 임포트
+import yfinance as yf
+import os
 import matplotlib.font_manager as fm
 import matplotlib.pyplot as plt
+import numpy as np
+import re  # 한글, 숫자, 기호를 남기기 위한 정규 표현식에 사용
 import gradio as gr
 import io
 from PIL import Image
 from datetime import datetime, timedelta
-import os
+from openai import OpenAI
 
-# 필요한 라이브러리 설치 및 업데이트
-subprocess.run(["pip", "install", "--upgrade", "pip"])
-subprocess.run(["pip", "install", "--upgrade", "openai", "yfinance", "gradio", "matplotlib", "Pillow"])
+# 1. 나눔고딕 폰트 설치 및 적용
+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  # 마이너스 부호 깨짐 방지
 
 # Perplexity AI API 설정
 API_KEY = "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"  # 여기에 Perplexity AI API 키를 입력하세요.
 
-# yfinance로 나스닥 헬스케어 시총 1위 종목 가져오기
-def get_nasdaq_healthcare_leader():
-    # 헬스케어 섹터 상위 기업들을 가져옴
-    healthcare_tickers = ["JNJ", "PFE", "MRNA", "VRTX", "AMGN"]  # 예시로 나스닥 헬스케어 상위 티커들
+# 뉴스 요약을 가져오는 함수 (기존 코드 유지)
+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 = get_dynamic_ticker(input_value)
+        company_name = input_value if ticker == input_value else ticker_to_name.get(ticker, input_value)
+        return handle_click(company_name, selected_date)
+
+# 종목명과 티커를 매핑하는 딕셔너리
+ticker_to_name = {}
+name_to_ticker = {}
+
+# 모든 예제를 동적으로 매칭하기 위한 함수
+def get_dynamic_ticker(input_value):
+    if input_value == "나스닥 시총 1위":
+        return get_top_market_cap_stock("나스닥")
+    elif input_value == "나스닥 바이오텍 시총 1위":
+        return get_top_market_cap_stock("나스닥", industry="Biotechnology")
+    elif input_value == "나스닥 헬스케어 시총 1위":
+        return get_top_market_cap_stock("나스닥", sector="Healthcare")
+    elif input_value == "코스피 ��총 1위":
+        return get_top_market_cap_stock("코스피")
+    elif input_value == "코스닥 시총 1위":
+        return get_top_market_cap_stock("코스닥")
+    else:
+        # 사전에 등록된 종목명 반환
+        return name_to_ticker.get(input_value, input_value)
+
+# 시가총액 상위 종목을 가져오는 함수
+def get_top_market_cap_stock(market, sector=None, industry=None):
+    # yfinance는 종목 목록을 제공하지 않으므로, ETF를 활용
+    if market == "나스닥":
+        etf_ticker = "QQQ"  # 나스닥 100 ETF
+    elif market == "코스피":
+        etf_ticker = "EWY"  # 한국 ETF
+    elif market == "코스닥":
+        # 코스닥 종목은 접근하기 어려우므로 사전 정의된 리스트 사용
+        tickers = ["035420.KQ", "068270.KQ", "035720.KQ"]
+    else:
+        return None
+
+    if market in ["나스닥", "코스피"]:
+        etf = yf.Ticker(etf_ticker)
+        holdings = etf.info.get('holdings', [])
+        tickers = [holding['symbol'] for holding in holdings]
+
     largest_market_cap = 0
-    leader_ticker = ""
+    top_ticker = None
 
-    for ticker in healthcare_tickers:
+    for ticker in tickers:
         stock = yf.Ticker(ticker)
-        stock_data = stock.history(period="1d")  # 전일 종가 기준
-        market_cap = stock.info.get('marketCap', 0)
+        stock_info = stock.info
+        market_cap = stock_info.get('marketCap', 0)
+        stock_sector = stock_info.get('sector', None)
+        stock_industry = stock_info.get('industry', None)
+
+        if sector and stock_sector != sector:
+            continue
+        if industry and stock_industry != industry:
+            continue
 
         if market_cap > largest_market_cap:
             largest_market_cap = market_cap
-            leader_ticker = ticker
-
-    return leader_ticker
+            top_ticker = ticker
+            ticker_to_name[ticker] = stock_info.get('shortName', ticker)
 
-# 종목명과 티커를 매핑하는 딕셔너리 확장 및 조건별 종목 매핑
-name_to_ticker = {
-    "나스닥 시총 1위": "AAPL",  # Apple
-    "나스닥 바이오텍 시총 1위": "VRTX",  # Vertex Pharmaceuticals
-    "나스닥 헬스케어 시총 1위": get_nasdaq_healthcare_leader(),  # yfinance로 동적으로 설정
-    "코스피 시총 1위": "005930.KS",  # 삼성전자
-    "코스닥 시총 1위": "196170.KQ",  # 알테오젠
-}
+    return top_ticker
 
 # 주가 데이터를 가져오고 조건에 맞는 날짜와 그래프를 반환하는 함수
 def display_stock_with_highlight(input_value, change_type, percent_change):
     try:
         # 입력값을 티커로 변환
-        ticker = name_to_ticker.get(input_value, input_value)
+        ticker = get_dynamic_ticker(input_value)
         stock = yf.Ticker(ticker)
         stock_data = stock.history(period="5y")  # 최근 5년 데이터로 제한
 
@@ -74,11 +187,13 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
 
         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.axvline(x=index, color=color, linestyle='--', linewidth=1)
 
-        plt.title(f'{input_value} 주가 추이')
-        plt.xlabel('날짜')
-        plt.ylabel('종가')
+        # 종목명 + '주가 추이'로 제목 설정 (종목명 기반)
+        company_name = ticker_to_name.get(ticker, input_value)
+        plt.title(f'{company_name} 주가 추이', fontproperties=font_prop)
+        plt.xlabel('날짜', fontproperties=font_prop)
+        plt.ylabel('종가', fontproperties=font_prop)
         plt.legend()
 
         buf = io.BytesIO()
@@ -105,11 +220,11 @@ with gr.Blocks() as demo:
 
             submit_btn = gr.Button("Submit")
 
-            # 예제 (이전 레이아웃으로 복원)
+            # 예제 (동적으로 매칭)
             examples = [["SK바이오팜"],
                         ["나스닥 시총 1위"],
                         ["나스닥 바이오텍 시총 1위"],
-                        ["나스닥 헬스케어 시총 1위"],  # 수정된 예제
+                        ["나스닥 헬스케어 시총 1위"],
                         ["코스피 시총 1위"],
                         ["코스닥 시총 1위"]]
             gr.Examples(examples=examples, inputs=[input_value])