Update app.py

app.py

@@ -2,9 +2,10 @@ import discord
 import logging
 import os
 import asyncio
-import yfinance as yf
-from huggingface_hub import InferenceClient
 import subprocess
+from huggingface_hub import InferenceClient
+from yahooquery import Ticker
+from datetime import datetime
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
@@ -95,13 +96,7 @@ async def generate_response(message):
     response = await asyncio.get_event_loop().run_in_executor(None, lambda: hf_client.chat_completion(
         messages=messages, max_tokens=1000, temperature=0.7, top_p=0.85))
 
-    full_response = []
-    for part in response:
-        logging.debug(f'스트림에서 받은 부분: {part}')
-        if part.choices and part.choices[0].delta and part.choices[0].delta.content:
-            full_response.append(part.choices[0].delta.content)
-
-    full_response_text = ''.join(full_response)
+    full_response_text = ''.join(response)
     logging.debug(f'모델 전체 응답: {full_response_text}')
 
     # 어시스턴트 응답을 대화 히스토리에 추가
@@ -114,34 +109,33 @@ async def generate_response(message):
 
 def get_ticker_from_name(name):
     try:
-        search_results = yf.Ticker(name).info
-        if 'symbol' in search_results:
-            return search_results['symbol']
-        else:
-            # 종목명이 아닌 경우, 가장 유사한 티커 검색
-            ticker_search = yf.utils.get_tickers()
-            matches = [t for t in ticker_search if name.lower() in t.lower()]
-            if matches:
-                return matches[0]  # 가장 유사한 티커 반환
-            else:
-                return None
+        ticker_search = Ticker(name)
+        search_results = ticker_search.summary_detail
+        if search_results and name in search_results:
+            return search_results[name]['symbol']
+        return None
     except Exception as e:
         logging.error(f'Error retrieving ticker for {name}: {e}')
         return None
 
 def get_company_info(ticker):
     try:
-        stock = yf.Ticker(ticker)
-        info = stock.info
+        stock = Ticker(ticker)
+        info = stock.summary_detail
+        if not info:
+            return f"티커 {ticker}에 대한 정보를 가져오는 데 실패했습니다."
+
         history = stock.history(period="10y")
+        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
         summary = f"""
-        **회사 이름:** {info.get('longName', 'N/A')}
+        **회사 이름:** {info[ticker].get('longName', 'N/A')}
         **티커:** {ticker}
-        **산업:** {info.get('industry', 'N/A')}
-        **섹터:** {info.get('sector', 'N/A')}
-        **시가 총액:** {info.get('marketCap', 'N/A')}
-        **현재가:** {info.get('previousClose', 'N/A')}
+        **산업:** {info[ticker].get('industry', 'N/A')}
+        **섹터:** {info[ticker].get('sector', 'N/A')}
+        **시가 총액:** {info[ticker].get('marketCap', 'N/A')}
+        **현재가:** {info[ticker].get('previousClose', 'N/A')}
+        **데이터 업데이트:** {now}
 
         **최근 1일 차트:**\n{get_chart_summary(history, period='1d')}
         **최근 1주 차트:**\n{get_chart_summary(history, period='1wk')}
@@ -160,7 +154,9 @@ def get_company_info(ticker):
         return f"티커 {ticker}에 대한 정보를 가져오는 데 실패했습니다."
 
 def get_chart_summary(history, period):
-    # 차트 요약 생성 (예: 평균, 최고, 최저 등)
+    if history.empty:
+        return "차트 데이터 없음"
+
     if period == '1d':
         recent = history.tail(1)
     elif period == '1wk':
@@ -172,15 +168,24 @@ def get_chart_summary(history, period):
     else:  # '10y'
         recent = history
 
-    average_price = recent['Close'].mean()
-    highest_price = recent['Close'].max()
-    lowest_price = recent['Close'].min()
+    average_price = recent['close'].mean()
+    highest_price = recent['close'].max()
+    lowest_price = recent['close'].min()
     return f"평균가: {average_price:.2f}, 최고가: {highest_price:.2f}, 최저가: {lowest_price:.2f}"
 
 def get_news_summary(ticker):
-    # 최신 뉴스 요약 생성 (Yahoo Finance API 등을 활용하여 뉴스 요약 추가 가능)
-    # 현재는 예시로 간단한 텍스트 반환
-    return f"{ticker} 관련 최신 뉴스: (뉴스 서비스와의 통합 필요)"
+    try:
+        stock = Ticker(ticker)
+        news = stock.news()
+        if not news:
+            return "최신 뉴스 없음"
+
+        top_news = news[:3]
+        news_summary = "\n".join([f"{i+1}. [{article['title']}]({article['link']}) - {article['providerPublishTime']}" for i, article in enumerate(top_news)])
+        return news_summary
+    except Exception as e:
+        logging.error(f'Error retrieving news for {ticker}: {e}')
+        return "뉴스 요약을 가져오는 데 실패했습니다."
 
 if __name__ == "__main__":
     discord_token = os.getenv('DISCORD_TOKEN')