Upload 14 files

EXTRACTATIONMODEL.py

@@ -11,13 +11,12 @@ from urllib.parse import quote
 from urllib.request import Request, urlopen
 from zoneinfo import ZoneInfo
 
-import numpy as np
-import pandas as pd
-import yfinance as yf
-from sklearn.ensemble import HistGradientBoostingRegressor
-from sklearn.linear_model import LogisticRegression, Ridge
-from sklearn.pipeline import Pipeline
-from sklearn.preprocessing import StandardScaler
+import numpy as np
+import pandas as pd
+from sklearn.ensemble import HistGradientBoostingRegressor
+from sklearn.linear_model import LogisticRegression, Ridge
+from sklearn.pipeline import Pipeline
+from sklearn.preprocessing import StandardScaler
 
 EPS = 1e-9
 
@@ -109,13 +108,20 @@ def download_ohlc_chart(symbol: str, start_date: str) -> pd.DataFrame:
         return pd.DataFrame()
 
 
-def download_ohlc(symbol: str, start_date: str) -> pd.DataFrame:
-    chart = download_ohlc_chart(symbol, start_date)
-    if not chart.empty:
-        return chart
-
-    last_err: Exception | None = None
-    raw = pd.DataFrame()
+def download_ohlc(symbol: str, start_date: str) -> pd.DataFrame:
+    chart = download_ohlc_chart(symbol, start_date)
+    if not chart.empty:
+        return chart
+
+    try:
+        import yfinance as yf
+    except Exception as exc:
+        raise RuntimeError(
+            "Yahoo chart API returned no data and yfinance fallback is unavailable."
+        ) from exc
+
+    last_err: Exception | None = None
+    raw = pd.DataFrame()
     for attempt in range(1, DOWNLOAD_RETRIES + 1):
         try:
             buf = io.StringIO()

requirements.txt

@@ -8,5 +8,6 @@ sentence-transformers
 aiohttp
 scipy
 pytest
-yfinance
-scikit-learn
+yfinance
+websockets>=13.0
+scikit-learn