Update app.py

app.py

@@ -1,11 +1,9 @@
-# 필요한 라이브러리 설치 및 업데이트를 최상단에 배치
 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
@@ -47,41 +45,33 @@ API_KEY = "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"  # 여기에 P
 
 # 뉴스 요약을 가져오는 함수 (기존 코드 유지)
 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)
 
@@ -90,16 +80,11 @@ 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)
+        company_name = 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("나스닥")
@@ -112,18 +97,15 @@ def get_dynamic_ticker(input_value):
     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
+        etf_ticker = "QQQ"
     elif market == "코스피":
-        etf_ticker = "EWY"  # 한국 ETF
+        etf_ticker = "EWY"
     elif market == "코스닥":
-        # 코스닥 종목은 접근하기 어려우므로 사전 정의된 리스트 사용
         tickers = ["035420.KQ", "068270.KQ", "035720.KQ"]
     else:
         return None
@@ -158,10 +140,9 @@ def get_top_market_cap_stock(market, sector=None, industry=None):
 # 주가 데이터를 가져오고 조건에 맞는 날짜와 그래프를 반환하는 함수
 def display_stock_with_highlight(input_value, change_type, percent_change):
     try:
-        # 입력값을 티커로 변환
         ticker = get_dynamic_ticker(input_value)
         stock = yf.Ticker(ticker)
-        stock_data = stock.history(period="5y")  # 최근 5년 데이터로 제한
+        stock_data = stock.history(period="5y")
 
         if stock_data.empty:
             return "주가 데이터를 찾을 수 없습니다.", []
@@ -189,7 +170,6 @@ def display_stock_with_highlight(input_value, change_type, percent_change):
             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)
 
-        # 종목명 + '주가 추이'로 제목 설정 (종���명 기반)
         company_name = ticker_to_name.get(ticker, input_value)
         plt.title(f'{company_name} 주가 추이', fontproperties=font_prop)
         plt.xlabel('날짜', fontproperties=font_prop)