Update app.py

app.py

@@ -277,7 +277,7 @@ def create_stock_chart(ticker, period="1mo"):
             else:
                 raise dl_error
         
-        if stock_data.empty:
+        if len(stock_data) == 0:
             logging.warning(f"No stock data found for ticker: {ticker}")
             return None
             
@@ -288,27 +288,59 @@ def create_stock_chart(ticker, period="1mo"):
         # 그래프 작성
         fig, ax = plt.subplots(figsize=(10, 6))
         
-        # 종가 그래프
-        ax.plot(stock_data.index, stock_data['Close'], label='Close Price', color='blue')
-        
-        # 이동평균선 추가 (20일)
-        if len(stock_data) > 20:
-            stock_data['MA20'] = stock_data['Close'].rolling(window=20).mean()
-            ax.plot(stock_data.index, stock_data['MA20'], label='20-day MA', color='orange')
-        
-        # 거래량 서브플롯 추가 (거래량이 있는 경우만)
-        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')
+        # 종가 그래프 - 멀티인덱스 처리
+        if isinstance(stock_data.columns, pd.MultiIndex):
+            # 멀티인덱스인 경우 ('Close', ticker) 형태
+            close_col = ('Close', ticker)
+            if close_col in stock_data.columns:
+                ax.plot(stock_data.index, stock_data[close_col], label='Close Price', color='blue')
+                
+                # 이동평균선 추가 (20일)
+                if len(stock_data) > 20:
+                    stock_data['MA20'] = stock_data[close_col].rolling(window=20).mean()
+                    ax.plot(stock_data.index, stock_data['MA20'], label='20-day MA', color='orange')
+                
+                # 거래량 서브플롯 추가 (거래량이 있는 경우만)
+                volume_col = ('Volume', ticker)
+                if volume_col in stock_data.columns and not stock_data[volume_col].isna().all():
+                    ax2 = ax.twinx()
+                    ax2.bar(stock_data.index, stock_data[volume_col], 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')
+            else:
+                raise ValueError(f"Close column not found in data columns: {stock_data.columns}")
         else:
-            # 거래량 없는 경우 종가만 표시
-            ax.legend(loc='upper left')
+            # 일반 인덱스인 경우
+            if 'Close' in stock_data.columns:
+                ax.plot(stock_data.index, stock_data['Close'], label='Close Price', color='blue')
+                
+                # 이동평균선 추가 (20일)
+                if len(stock_data) > 20:
+                    stock_data['MA20'] = stock_data['Close'].rolling(window=20).mean()
+                    ax.plot(stock_data.index, stock_data['MA20'], label='20-day MA', color='orange')
+                
+                # 거래량 서브플롯 추가 (거래량이 있는 경우만)
+                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')
+            else:
+                raise ValueError(f"Close column not found in data columns: {stock_data.columns}")
         
         # 차트 스타일링
         ax.set_title(f"{ticker} Stock Price")