Update app.py

app.py

@@ -5,6 +5,8 @@ import yfinance as yf
 import time
 import logging
 from datetime import datetime
+import re
+
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
@@ -108,10 +110,16 @@ def download_stock_data(ticker):
         return f"Error: 데이터 다운로드 중 오류가 발생했습니다. {str(e)}"
 
 def extract_ticker(output_part):
-    # 사용자 출력에서 티커를 추출하는 로직 구현
-    # 예시: 'AAPL' 추출
-    ticker = "AAPL"  # 실제 티커 추출 로직으로 대체 필요
-    return ticker
+    # 정규식을 사용하여 티커 추출
+    ticker_pattern = r'\b[A-Za-z]{1,4}\d{1,2}(?:\.[A-Za-z]{2})?\b'
+    ticker_matches = re.findall(ticker_pattern, output_part)
+    
+    if ticker_matches:
+        # 추출된 티커 중 첫 번째 티커 반환
+        return ticker_matches[0]
+    else:
+        # 티커가 없는 경우 None 반환
+        return None
 
 def validate_ticker(ticker):
     logging.debug(f"Validating ticker: {ticker}")