fix: 비트코인 로직 수정

app.py

@@ -15,26 +15,44 @@ def get_pipeline(model_id: str, device: str = "cpu"):
     if key not in _PIPELINE_CACHE:
         _PIPELINE_CACHE[key] = BaseChronosPipeline.from_pretrained(
             model_id,
-            device_map=device,  # "cpu" / "cuda" (Spaces 기본은 cpu)
+            device_map=device,  # "cpu" / "cuda"
             torch_dtype=torch.float32 if device == "cpu" else torch.bfloat16,
         )
     return _PIPELINE_CACHE[key]
 
-# ---- 주가 데이터 로딩 (yfinance) ----
+# ---- 주가/크립토 데이터 로딩 (yfinance, 견고화) ----
 def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
-    # 한국 주식은 예: 005930.KS (삼성전자)
-    df = yf.download(ticker, start=start, end=end, interval=interval, progress=False)
-    if df.empty or "Close" not in df:
-        raise ValueError("데이터가 없거나 'Close' 열을 찾을 수 없습니다. 티커/날짜를 확인하세요.")
+    """
+    BTC-USD 등 크립토 심볼에서 간헐적으로 timezone/파싱 오류가 나므로
+    history() 경로를 우선 사용하고, 실패 시 한 번 재시도.
+    """
+    # 기본값 보정: 너무 최근만 고르면 빈 데이터가 나올 수 있어 일봉은 과거부터 권장
+    _start = start or "2014-09-17"  # BTC-USD 최초 상장일 근처
+    _end = end or dt.date.today().isoformat()
+
+    tk = yf.Ticker(ticker)
+    try:
+        df = tk.history(start=_start, end=_end, interval=interval, auto_adjust=True, actions=False)
+        if df.empty or "Close" not in df:
+            raise ValueError("empty")
+    except Exception:
+        # fallback: download() 경로 시도
+        df = yf.download(ticker, start=_start, end=_end, interval=interval, progress=False, threads=False)
+        if df.empty or "Close" not in df:
+            raise ValueError("데이터가 없거나 'Close' 열이 없습니다. 티커/날짜/간격을 확인하세요.")
+
     s = df["Close"].dropna().astype(float)
+    if s.empty:
+        raise ValueError("다운로드 결과가 비어 있습니다. 기간/간격을 줄이거나 다시 시도하세요.")
     return s
 
 # ---- 예측 함수 (Gradio가 호출) ----
 def run_forecast(ticker, start_date, end_date, horizon, model_id, device, interval):
     try:
-        series = load_close_series(ticker, start_date, end_date, interval)
+        series = load_close_series(ticker.strip(), start_date, end_date, interval)
     except Exception as e:
-        return gr.Plot.update(None), pd.DataFrame(), f"데이터 로딩 오류: {e}"
+        # Gradio v4에서는 Plot.update가 없음 → None 반환으로 정리
+        return None, pd.DataFrame(), f"데이터 로딩 오류: {e}"
 
     pipe = get_pipeline(model_id, device)
     H = int(horizon)
@@ -42,8 +60,7 @@ def run_forecast(ticker, start_date, end_date, horizon, model_id, device, interv
     # Chronos 입력: 1D 텐서 (float)
     context = torch.tensor(series.values, dtype=torch.float32)
 
-    # 출력: (num_series=1, num_quantiles=3, H)
-    #    보통 q=[0.1, 0.5, 0.9]
+    # 예측: (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]
 
@@ -53,15 +70,18 @@ def run_forecast(ticker, start_date, end_date, horizon, model_id, device, interv
         index=pd.RangeIndex(1, H + 1, name="step"),
     )
 
-    # 그래프
+    # 미래 x축: interval에 맞는 pandas 주기
     import matplotlib.pyplot as plt
+    freq_map = {"1d": "D", "1h": "H", "30m": "30T", "15m": "15T", "5m": "5T"}
+    freq = freq_map.get(interval, "D")
+    future_index = pd.date_range(series.index[-1], periods=H + 1, freq=freq)[1:]
+
+    # 그래프
     fig = plt.figure(figsize=(10, 4))
     plt.plot(series.index, series.values, label="history")
-    # 미래 구간 x축 만들기: 종가가 일 단위라 'D' 주기 사용
-    future_index = pd.date_range(series.index[-1], periods=H + 1, freq="D")[1:]
     plt.plot(future_index, q50.numpy(), label="forecast(q50)")
     plt.fill_between(future_index, q10.numpy(), q90.numpy(), alpha=0.2, label="q10–q90")
-    plt.title(f"{ticker} forecast by Chronos-Bolt")
+    plt.title(f"{ticker} forecast by Chronos-Bolt ({interval}, H={H})")
     plt.legend()
     plt.tight_layout()
 
@@ -69,14 +89,14 @@ def run_forecast(ticker, start_date, end_date, horizon, model_id, device, interv
     return fig, df_fcst, note
 
 # ---- Gradio UI ----
-with gr.Blocks(title="Chronos Stock Forecast") as demo:
-    gr.Markdown("# Chronos 주가 예측 데모")
+with gr.Blocks(title="Chronos Stock/Crypto Forecast") as demo:
+    gr.Markdown("# Chronos 주가·크립토 예측 데모")
     with gr.Row():
-        ticker = gr.Textbox(value="AAPL", label="티커 (예: AAPL, MSFT, 005930.KS)")
-        horizon = gr.Slider(5, 60, value=20, step=1, label="예측 길이 H (일)")
+        ticker = gr.Textbox(value="BTC-USD", label="티커 (예: AAPL, MSFT, 005930.KS, BTC-USD)")
+        horizon = gr.Slider(5, 365, value=90, step=1, label="예측 스텝 H (간격 단위�� 동일)")
     with gr.Row():
-        start = gr.Textbox(value=(dt.date.today()-dt.timedelta(days=365)).isoformat(), label="시작일 (YYYY-MM-DD)")
-        end = gr.Textbox(value=dt.date.today().isoformat(), label="종료일 (YYYY-MM-DD)")
+        start = gr.Textbox(value="2014-09-17", label="시작일 (YYYY-MM-DD, 예: 2014-09-17)")
+        end = gr.Textbox(value=dt.date.today().isoformat(), label="종료일 (YYYY-MM-DD, 비워두면 오늘)")
     with gr.Row():
         model_id = gr.Dropdown(
             choices=[
@@ -89,7 +109,11 @@ with gr.Blocks(title="Chronos Stock Forecast") as demo:
             label="모델"
         )
         device = gr.Dropdown(choices=["cpu"], value="cpu", label="디바이스")
-        interval = gr.Dropdown(choices=["1d"], value="1d", label="간격")
+        interval = gr.Dropdown(
+            choices=["1d", "1h", "30m", "15m", "5m"],
+            value="1d",
+            label="간격"
+        )
     btn = gr.Button("예측 실행")
 
     plot = gr.Plot(label="History + Forecast")