fix:수정

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import torch
 import gradio as gr
 import yfinance as yf
+import requests
 
 from chronos import BaseChronosPipeline  # pip: chronos-forecasting
 
@@ -24,17 +25,98 @@ def get_pipeline(model_id: str, device: str = "cpu"):
 
 
 # =============================
-#   yfinance 전용 견고 로더
+#   Binance (무인증) 폴백
 # =============================
-def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
+_BINANCE_INTERVAL = {"1d": "1d", "1h": "1h", "30m": "30m", "15m": "15m", "5m": "5m"}
+
+def _yf_to_binance_symbol(ticker: str) -> str | None:
+    """
+    BTC-USD -> BTCUSDT, ETH-USD -> ETHUSDT ...
+    주식/원화/기타 심볼은 None (Binance 폴백하지 않음)
+    """
+    t = ticker.upper()
+    if t.endswith("-USD") and len(t) >= 6:
+        base = t[:-4]  # remove "-USD"
+        return f"{base}USDT"
+    return None
+
+def _fetch_binance_klines(ticker: str, interval: str, start: str | None, end: str | None) -> pd.Series:
     """
-    yfinance만 사용.
-    1) Ticker().history(start/end)
-    2) download(start/end, repair=True)
-    3) period 기반 폴백:
-       - 1d  → ["max", "10y", "5y", "2y", "1y"]
-       - 1h  → ["730d", "365d", "60d"]
-       - 30m/15m/5m → ["60d", "30d", "14d"]
+    Binance Klines (무인증)
+    https://api.binance.com/api/v3/klines
+    반환: pandas.Series(index=datetime, values=float close)
+    """
+    if interval not in _BINANCE_INTERVAL:
+        raise ValueError("Binance는 해당 interval을 지원하지 않습니다.")
+
+    symbol = _yf_to_binance_symbol(ticker)
+    if not symbol:
+        raise ValueError("이 티커는 Binance 폴백 대상이 아닙니다.")
+
+    base = "https://api.binance.com/api/v3/klines"
+
+    def to_ms(s: str) -> int:
+        return int(pd.to_datetime(s).timestamp() * 1000)
+
+    start_ms = to_ms(start) if start else None
+    end_ms = to_ms(end) if end else None
+
+    rows = []
+    cur_start = start_ms
+    while True:
+        params = {"symbol": symbol, "interval": _BINANCE_INTERVAL[interval], "limit": 1000}
+        if cur_start is not None:
+            params["startTime"] = cur_start
+        if end_ms is not None:
+            params["endTime"] = end_ms
+
+        r = requests.get(base, params=params, timeout=30)
+        r.raise_for_status()
+        data = r.json()
+        if not data:
+            break
+
+        rows.extend(data)
+
+        last_close_time = data[-1][6]  # closeTime (ms)
+        next_start = last_close_time + 1
+        if cur_start is not None and next_start <= cur_start:
+            break
+        cur_start = next_start
+
+        if len(data) < 1000:
+            break
+
+    if not rows:
+        raise ValueError("Binance에서 데이터가 비어 있습니다.")
+
+    df = pd.DataFrame(rows, columns=[
+        "openTime","open","high","low","close","volume","closeTime",
+        "quoteAssetVolume","numTrades","takerBuyBase","takerBuyQuote","ignore"
+    ])
+    df["ts"] = pd.to_datetime(df["closeTime"], unit="ms")
+    s = df.set_index("ts")["close"].astype(float).sort_index()
+
+    if start:
+        s = s[s.index >= pd.to_datetime(start)]
+    if end:
+        s = s[s.index <= pd.to_datetime(end)]
+    if s.empty:
+        raise ValueError("Binance 시리즈가 비어 있습니다.")
+    return s
+
+
+# =============================
+#   yfinance 전용 견고 로더 (+Binance 폴백)
+# =============================
+def load_close_series(ticker: str, start: str, end: str, interval: str = "1d") -> pd.Series:
+    """
+    yfinance만으로 먼저 시도:
+      1) Ticker().history(start/end)
+      2) download(start/end, repair=True)
+      3) period 폴백 (interval별 후보 순회)
+    그래도 실패할 경우:
+      4) Binance (무인증) 폴백 — BTC-USD 같은 암호화폐만 대상
     """
     ticker = ticker.strip().upper()
 
@@ -45,12 +127,12 @@ def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
         sdt = pd.to_datetime(_start)
         edt = pd.to_datetime(_end)
         if edt < sdt:
-            sdt, edt = edt, sdt  # 뒤바뀐 경우 교환
+            sdt, edt = edt, sdt
         _start, _end = sdt.date().isoformat(), edt.date().isoformat()
     except Exception:
-        pass  # 파싱 실패해도 밑의 period 폴백이 커버
+        pass
 
-    def _extract_close(df):
+    def _extract_close(df: pd.DataFrame | None) -> pd.Series | None:
         if df is None or df.empty:
             return None
         c = df.get("Close")
@@ -85,12 +167,11 @@ def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
     if interval == "1d":
         period_candidates = ["max", "10y", "5y", "2y", "1y"]
     elif interval == "1h":
-        period_candidates = ["730d", "365d", "60d"]   # 1시간봉은 과거 제한 큼
-    else:  # 30m/15m/5m 등 분봉
-        period_candidates = ["60d", "30d", "14d"]     # 분봉은 보통 60~30일 이내만 가능
+        period_candidates = ["730d", "365d", "60d"]
+    else:  # 30m/15m/5m
+        period_candidates = ["60d", "30d", "14d"]
 
     for per in period_candidates:
-        # Ticker().history(period=…)
         try:
             df = tk.history(period=per, interval=interval, auto_adjust=True, actions=False)
             s = _extract_close(df)
@@ -98,7 +179,6 @@ def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
                 return s
         except Exception:
             pass
-        # download(period=…)
         try:
             df = yf.download(
                 ticker, period=per, interval=interval,
@@ -110,9 +190,15 @@ def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
         except Exception:
             pass
 
+    # 4) Binance 폴백 (암호화폐만)
+    try:
+        s = _fetch_binance_klines(ticker, interval, _start, _end)
+        return s
+    except Exception:
+        pass
+
     raise ValueError(
-        "yfinance에서 데이터를 가져오지 못했습니다. "
-        "간격(interval)이나 기간(start/end 혹은 period)을 조정해 다시 시도해 보세요."
+        "데이터를 가져오지 못했습니다. 간격(interval)이나 기간(start/end 혹은 period)을 조정해 다시 시도해 보세요."
     )
 
 
@@ -123,26 +209,25 @@ def run_forecast(ticker, start_date, end_date, horizon, model_id, device, interv
     try:
         series = load_close_series(ticker, start_date, end_date, interval)
     except Exception as e:
-        # Gradio v4: Plot.update 없음 → None 반환
         return None, pd.DataFrame(), f"데이터 로딩 오류: {e}"
 
     pipe = get_pipeline(model_id, device)
     H = int(horizon)
 
-    # Chronos 입력: 1D 텐서 (float)
+    # Chronos 입력
     context = torch.tensor(series.values, dtype=torch.float32)
 
     # 예측: (num_series=1, num_quantiles=3, H) with q=[0.1, 0.5, 0.9]
     preds = pipe.predict(context=context, prediction_length=H)[0]
     q10, q50, q90 = preds[0], preds[1], preds[2]
 
-    # 표 데이터
+    # 표
     df_fcst = pd.DataFrame(
         {"q10": q10.numpy(), "q50": q50.numpy(), "q90": q90.numpy()},
         index=pd.RangeIndex(1, H + 1, name="step"),
     )
 
-    # 미래 x축: interval→pandas freq 매핑
+    # 미래 x축
     import matplotlib.pyplot as plt
     freq_map = {"1d": "D", "1h": "H", "30m": "30T", "15m": "15T", "5m": "5T"}
     freq = freq_map.get(interval, "D")