Update app.py

app.py

@@ -38,11 +38,33 @@ COMMON_TICKERS = {
     "meta": "META",
     "tesla": "TSLA",
     "nvidia": "NVDA",
+    "netflix": "NFLX",
+    "amd": "AMD",
+    "intel": "INTC",
+    "ibm": "IBM",
+    "oracle": "ORCL",
+    "paypal": "PYPL",
+    "adobe": "ADBE",
+    "cisco": "CSCO",
     "bitcoin": "BTC-USD",
     "ethereum": "ETH-USD",
+    "dogecoin": "DOGE-USD",
+    "cardano": "ADA-USD",
+    "xrp": "XRP-USD",
+    "litecoin": "LTC-USD",
     "samsung": "005930.KS",  # 한국 삼성전자
     "hyundai": "005380.KS",  # 현대자동차
     "sk hynix": "000660.KS",  # SK하이닉스
+    "lg": "003550.KS",       # LG
+    "lge": "066570.KS",      # LG전자
+    "ncsoft": "036570.KS",   # 엔씨소프트
+    "kakao": "035720.KS",    # 카카오
+    "naver": "035420.KS",    # 네이버
+    "현대차": "005380.KS",    # 현대자동차
+    "삼성전자": "005930.KS",   # 삼성전자
+    "삼성": "005930.KS",      # 삼성전자
+    "카카오": "035720.KS",    # 카카오
+    "네이버": "035420.KS",    # 네이버
 }
 
 def fetch_articles(query, max_articles=30):
@@ -167,27 +189,73 @@ def get_stock_ticker(asset_name):
     """
     자산명으로부터 주식 티커 심볼을 추출
     """
+    logging.info(f"Identifying ticker for: {asset_name}")
+    
     # 소문자로 변환하여 매핑 확인
-    asset_lower = asset_name.lower()
+    asset_lower = asset_name.lower().strip()
     
     # 직접 티커로 입력한 경우 (대문자 3-5자 형태)
-    if asset_name.isupper() and 3 <= len(asset_name) <= 5:
+    if asset_name.isupper() and 2 <= len(asset_name) <= 6:
+        logging.info(f"Input appears to be a ticker symbol: {asset_name}")
         return asset_name
     
     # 일반적인 종목명 매핑 확인
     if asset_lower in COMMON_TICKERS:
-        return COMMON_TICKERS[asset_lower]
-    
-    # 그 외의 경우 yfinance로 검색 시도
+        ticker = COMMON_TICKERS[asset_lower]
+        logging.info(f"Found ticker in common tickers map: {ticker}")
+        return ticker
+    
+    # 여러 단어로 된 이름의 각 부분에 대한 검색도 시도
+    asset_parts = asset_lower.split()
+    for part in asset_parts:
+        if part in COMMON_TICKERS:
+            ticker = COMMON_TICKERS[part]
+            logging.info(f"Found ticker for part '{part}': {ticker}")
+            return ticker
+    
+    # 그 외의 경우 직접 티커로 시도
+    potential_ticker = asset_name.upper().replace(" ", "")
+    if 2 <= len(potential_ticker) <= 6:
+        # 실제로 존재하는지 확인
+        try:
+            logging.info(f"Trying potential ticker: {potential_ticker}")
+            test_data = yf.download(potential_ticker, period="1d", progress=False)
+            if not test_data.empty:
+                logging.info(f"Valid ticker found: {potential_ticker}")
+                return potential_ticker
+        except Exception as e:
+            logging.debug(f"Error testing potential ticker: {e}")
+    
+    # 그 외의 경우 yfinance로 검색 시도 (info 데이터)
     try:
+        # 일부 티커는 직접 yfinance 기반 검색으로는 오류가 발생할 수 있음
         ticker_search = yf.Ticker(asset_name)
-        # 기본 정보 가져와서 유효한 티커인지 확인
-        info = ticker_search.info
-        if 'symbol' in info:
-            return info['symbol']
-    except:
-        pass
-    
+        try:
+            info = ticker_search.info
+            if 'symbol' in info and info['symbol']:
+                ticker = info['symbol']
+                logging.info(f"Found ticker from info API: {ticker}")
+                return ticker
+        except (ValueError, KeyError, TypeError) as e:
+            logging.debug(f"Error getting ticker info: {e}")
+            pass
+    except Exception as e:
+        logging.debug(f"Error initializing ticker object: {e}")
+    
+    # 추가 시도: 일반적인 미국 증시 티커 형식 확인
+    major_exchanges = ["", ".KS", ".KQ", "-USD"]  # 주요 거래소 접미사 (한국 포함)
+    for exchange in major_exchanges:
+        try:
+            test_ticker = f"{asset_name.upper().replace(' ', '')}{exchange}"
+            logging.info(f"Trying with exchange suffix: {test_ticker}")
+            test_data = yf.download(test_ticker, period="1d", progress=False)
+            if not test_data.empty:
+                logging.info(f"Valid ticker found with suffix: {test_ticker}")
+                return test_ticker
+        except:
+            pass
+    
+    logging.warning(f"Could not identify ticker for: {asset_name}")
     return None
 
 def create_stock_chart(ticker, period="1mo"):
@@ -196,12 +264,28 @@ def create_stock_chart(ticker, period="1mo"):
     """
     try:
         logging.info(f"Fetching stock data for {ticker}")
-        stock_data = yf.download(ticker, period=period)
+        # Graceful handling for problematic symbols
+        try:
+            stock_data = yf.download(ticker, period=period, progress=False)
+        except Exception as dl_error:
+            logging.error(f"Error downloading stock data: {dl_error}")
+            # Try alternative symbol format
+            if "-" in ticker:
+                alt_ticker = ticker.replace("-", ".")
+                logging.info(f"Trying alternative ticker format: {alt_ticker}")
+                stock_data = yf.download(alt_ticker, period=period, progress=False)
+            else:
+                raise dl_error
         
         if stock_data.empty:
             logging.warning(f"No stock data found for ticker: {ticker}")
             return None
             
+        # 데이터 확인 및 디버그 로깅
+        logging.info(f"Downloaded data shape: {stock_data.shape}")
+        logging.info(f"Data columns: {stock_data.columns.tolist()}")
+        
+        # 그래프 작성
         fig, ax = plt.subplots(figsize=(10, 6))
         
         # 종가 그래프
@@ -212,10 +296,19 @@ def create_stock_chart(ticker, period="1mo"):
             stock_data['MA20'] = stock_data['Close'].rolling(window=20).mean()
             ax.plot(stock_data.index, stock_data['MA20'], label='20-day MA', color='orange')
         
-        # 거래량 서브플롯 추가
-        ax2 = ax.twinx()
-        ax2.bar(stock_data.index, stock_data['Volume'], alpha=0.3, color='gray', label='Volume')
-        ax2.set_ylabel('Volume')
+        # 거래량 서브플롯 추가 (거래량이 있는 경우만)
+        if 'Volume' in stock_data.columns and not stock_data['Volume'].isna().all():
+            ax2 = ax.twinx()
+            ax2.bar(stock_data.index, stock_data['Volume'], alpha=0.3, color='gray', label='Volume')
+            ax2.set_ylabel('Volume')
+            
+            # 범례 추가 (거래량 있는 경우)
+            lines, labels = ax.get_legend_handles_labels()
+            lines2, labels2 = ax2.get_legend_handles_labels()
+            ax.legend(lines + lines2, labels + labels2, loc='upper left')
+        else:
+            # 거래량 없는 경우 종가만 표시
+            ax.legend(loc='upper left')
         
         # 차트 스타일링
         ax.set_title(f"{ticker} Stock Price")
@@ -223,15 +316,10 @@ def create_stock_chart(ticker, period="1mo"):
         ax.set_ylabel('Price')
         ax.grid(True, alpha=0.3)
         
-        # 범례 추가
-        lines, labels = ax.get_legend_handles_labels()
-        lines2, labels2 = ax2.get_legend_handles_labels()
-        ax.legend(lines + lines2, labels + labels2, loc='upper left')
-        
         plt.tight_layout()
         
         # 이미지 저장
-        chart_path = f"stock_chart_{ticker.replace('-', '_')}.png"
+        chart_path = f"stock_chart_{ticker.replace('-', '_').replace('.', '_')}.png"
         plt.savefig(chart_path)
         plt.close()
         
@@ -239,7 +327,18 @@ def create_stock_chart(ticker, period="1mo"):
         return chart_path
     except Exception as e:
         logging.error(f"Error creating stock chart for {ticker}: {e}")
-        return None
+        # 오류 발생 시에도 그래프 생성 시도 (기본 텍스트 안내)
+        try:
+            fig, ax = plt.subplots(figsize=(10, 6))
+            ax.text(0.5, 0.5, f"Unable to load data for {ticker}\nError: {str(e)}", 
+                    horizontalalignment='center', verticalalignment='center', transform=ax.transAxes)
+            ax.set_axis_off()
+            chart_path = f"stock_chart_error_{ticker.replace('-', '_').replace('.', '_')}.png"
+            plt.savefig(chart_path)
+            plt.close()
+            return chart_path
+        except:
+            return None
 
 def analyze_asset_sentiment(asset_name):
     logging.info(f"Starting sentiment analysis for asset: {asset_name}")
@@ -422,7 +521,13 @@ with gr.Blocks() as iface:
                     wrap=False,
                 )
     
-    analyze_button.click(
+    # 디버그 메시지를 위한 함수 추가
+def debug_callback(message):
+    logging.info(f"DEBUG: {message}")
+    return f"Debug: {message}"
+
+# 분석 버튼 콜백 함수 연결
+analyze_button.click(
         analyze_asset_sentiment,
         inputs=[input_asset],
         outputs=[articles_output, sentiment_summary, stock_chart, ticker_info],