fix:수정

app.py

@@ -1,14 +1,15 @@
-import os
 import datetime as dt
 import pandas as pd
 import torch
 import gradio as gr
 import yfinance as yf
-import requests  # ← 추가
 
-from chronos import BaseChronosPipeline  # from 'chronos-forecasting'
+from chronos import BaseChronosPipeline  # pip: chronos-forecasting
 
-# ---- 전역 캐시: 모델을 한 번만 로드해 재사용 ----
+
+# =============================
+#   Chronos 모델 캐시/로더
+# =============================
 _PIPELINE_CACHE = {}
 
 def get_pipeline(model_id: str, device: str = "cpu"):
@@ -20,125 +21,134 @@ def get_pipeline(model_id: str, device: str = "cpu"):
             torch_dtype=torch.float32 if device == "cpu" else torch.bfloat16,
         )
     return _PIPELINE_CACHE[key]
-# ---- 심볼 매핑: 'BTC-USD' → 'bitcoin' (Coingecko id)
-_CG_MAP = {
-    "BTC-USD": "bitcoin",
-    "ETH-USD": "ethereum",
-    "SOL-USD": "solana",
-    "XRP-USD": "ripple",
-    "ADA-USD": "cardano",
-}
-
-def _fetch_coingecko_daily(ticker: str, start: str, end: str):
-    """
-    Coingecko: /coins/{id}/market_chart?vs_currency=usd&days=max
-    반환: (date, price) 일별 데이터프레임
-    """
-    coin_id = _CG_MAP.get(ticker.upper())
-    if not coin_id:
-        raise ValueError("해당 티커는 Coingecko 매핑이 없습니다. (예: BTC-USD, ETH-USD)")
-
-    url = f"https://api.coingecko.com/api/v3/coins/{coin_id}/market_chart"
-    # days=max 로 전체 일봉 받아온 뒤, 날짜 필터링
-    resp = requests.get(url, params={"vs_currency": "usd", "days": "max"}, timeout=30)
-    resp.raise_for_status()
-    data = resp.json()
-    prices = data.get("prices", [])
-    if not prices:
-        raise ValueError("Coingecko 응답에 prices가 없습니다.")
-
-    # prices: [[timestamp_ms, price], ...]
-    df = pd.DataFrame(prices, columns=["ts", "close"])
-    df["ts"] = pd.to_datetime(df["ts"], unit="ms", utc=True).dt.tz_convert(None)
-    df = df.set_index("ts").sort_index()
-    # 날짜 범위 적용
-    s = df["close"].astype(float)
-    if start:
-        s = s[s.index >= pd.to_datetime(start)]
-    if end:
-        s = s[s.index <= pd.to_datetime(end)]
-    return s
 
+
+# =============================
+#   yfinance 전용 견고 로더
+# =============================
 def load_close_series(ticker: str, start: str, end: str, interval: str = "1d"):
     """
-    1) yfinance(history → download)로 시도
-    2) 실패 시 Coingecko 일봉으로 대체 (BTC-USD/ETH-USD 등)
+    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"]
     """
     ticker = ticker.strip().upper()
-    _start = start or "2014-09-17"
-    _end = end or dt.date.today().isoformat()
 
-    # ---- 1차: yfinance 시도
+    # 날짜 보정
+    _start = start or "2014-09-17"                    # BTC-USD 히스토리 시작 근처
+    _end = end or dt.date.today().isoformat()
+    try:
+        sdt = pd.to_datetime(_start)
+        edt = pd.to_datetime(_end)
+        if edt < sdt:
+            sdt, edt = edt, sdt  # 뒤바뀐 경우 교환
+        _start, _end = sdt.date().isoformat(), edt.date().isoformat()
+    except Exception:
+        pass  # 파싱 실패해도 밑의 period 폴백이 커버
+
+    def _extract_close(df):
+        if df is None or df.empty:
+            return None
+        c = df.get("Close")
+        if c is None:
+            return None
+        c = c.dropna().astype(float)
+        return c if not c.empty else None
+
+    # 1) history(start/end)
     try:
         tk = yf.Ticker(ticker)
         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 history")
-
-        s = df["Close"].dropna().astype(float)
-        if s.empty:
-            raise ValueError("empty close after dropna")
-        return s
+        s = _extract_close(df)
+        if s is not None:
+            return s
     except Exception:
         pass
 
+    # 2) download(start/end) + repair=True
     try:
-        df = yf.download(ticker, start=_start, end=_end, interval=interval, progress=False, threads=False)
-        if not df.empty and "Close" in df and not df["Close"].dropna().empty:
-            return df["Close"].dropna().astype(float)
+        df = yf.download(
+            ticker, start=_start, end=_end, interval=interval,
+            progress=False, threads=False, repair=True
+        )
+        s = _extract_close(df)
+        if s is not None:
+            return s
     except Exception:
         pass
 
-    # ---- 2차: Coingecko fallback (일봉만)
-    try:
-        s = _fetch_coingecko_daily(ticker, _start, _end)
-        if s.empty:
-            raise ValueError("Coingecko 데이터가 비어 있습니다.")
-        # interval이 일봉이 아니면 일봉으로 강제 전환 안내 (호출 측에서 메시지로 보여줌)
-        if interval != "1d":
-            raise RuntimeError("FALLBACK_DAILY_ONLY")
-        return s
-    except RuntimeError as r:
-        if str(r) == "FALLBACK_DAILY_ONLY":
-            # 호출부에서 메시지 처리할 수 있게 예외를 다시 던짐
-            raise RuntimeError("FALLBACK_DAILY_ONLY")
-        raise
-    except Exception as e:
-        raise ValueError(f"데이터를 가져오지 못했습니다 (yfinance/Coingecko 실패): {e}")
+    # 3) period 폴백
+    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일 이내만 가능
+
+    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)
+            if s is not None:
+                return s
+        except Exception:
+            pass
+        # download(period=…)
+        try:
+            df = yf.download(
+                ticker, period=per, interval=interval,
+                progress=False, threads=False, repair=True
+            )
+            s = _extract_close(df)
+            if s is not None:
+                return s
+        except Exception:
+            pass
+
+    raise ValueError(
+        "yfinance에서 데이터를 가져오지 못했습니다. "
+        "간격(interval)이나 기간(start/end 혹은 period)을 조정해 다시 시도해 보세요."
+    )
+
 
+# =============================
+#   예측 함수 (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)
-        fallback_note = ""
-    except RuntimeError as r:
-        if str(r) == "FALLBACK_DAILY_ONLY":
-            # 일봉으로 재시도
-            series = load_close_series(ticker, start_date, end_date, "1d")
-            interval = "1d"
-            fallback_note = "※ Coingecko 대체 소스 사용으로 간격을 '1d(일봉)'로 자동 전환했습니다."
-        else:
-            return None, pd.DataFrame(), f"데이터 로딩 오류: {r}"
     except Exception as e:
+        # Gradio v4: Plot.update 없음 → None 반환
         return None, pd.DataFrame(), f"데이터 로딩 오류: {e}"
 
     pipe = get_pipeline(model_id, device)
     H = int(horizon)
 
-    import numpy as np
+    # Chronos 입력: 1D 텐서 (float)
     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 매핑
     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")
     plt.plot(future_index, q50.numpy(), label="forecast(q50)")
@@ -147,19 +157,21 @@ def run_forecast(ticker, start_date, end_date, horizon, model_id, device, interv
     plt.legend()
     plt.tight_layout()
 
-    base_note = "※ 데모 목적입니다. 투자 판단의 책임은 본인에게 있습니다."
-    note = (fallback_note + "  " + base_note).strip()
+    note = "※ 데모 목적입니다. 투자 판단의 책임은 본인에게 있습니다."
     return fig, df_fcst, note
 
-# ---- Gradio UI ----
+
+# =============================
+#   Gradio UI
+# =============================
 with gr.Blocks(title="Chronos Stock/Crypto Forecast") as demo:
     gr.Markdown("# Chronos 주가·크립토 예측 데모")
     with gr.Row():
         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="2014-09-17", label="시작일 (YYYY-MM-DD, 예: 2014-09-17)")
-        end = gr.Textbox(value=dt.date.today().isoformat(), label="종료일 (YYYY-MM-DD, 비워두면 오늘)")
+        start = gr.Textbox(value="2014-09-17", label="시작일 (YYYY-MM-DD)")
+        end = gr.Textbox(value=dt.date.today().isoformat(), label="종료일 (YYYY-MM-DD)")
     with gr.Row():
         model_id = gr.Dropdown(
             choices=[

requirements.txt

@@ -1,7 +1,6 @@
 gradio>=4.44
 pandas>=2.2
-yfinance==0.2.40   # (우린 실시간 안 써서 이대로 안전)
-requests>=2.31
+yfinance==0.2.40
 matplotlib>=3.8
 torch>=2.2 ; platform_system != "Darwin"
 chronos-forecasting>=1.0