Update app/sentiment.py

app/sentiment.py

@@ -1,53 +1,70 @@
 """
-Safe, lazy-loading sentiment pipeline for HF Spaces with proper cache redirection.
+Background price cache with multi-API fallback and rate-limit handling
 """
-import os
-import hashlib
+import httpx
 import logging
-from functools import lru_cache
-
-# ─── Ensure all HF caching uses a writable directory ─────────────────────────
-# Set HF_HOME for huggingface_hub and TRANSFORMERS_CACHE for transformers
-os.environ["HF_HOME"] = "/tmp/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
-os.makedirs("/tmp/huggingface", exist_ok=True)
-
-from transformers import pipeline
-
-class SentimentCache:
-    """Handles in-memory caching and streaming of sentiment results."""
-    latest_id: int = 0
-    latest_result: dict = {}
-    _pipeline = None  # Will hold the loaded pipeline
-
-    @classmethod
-    def _get_pipeline(cls):
-        if cls._pipeline is None:
-            logging.info("🔄 Loading sentiment model…")
-            cls._pipeline = pipeline(
-                "sentiment-analysis",
-                model="distilbert-base-uncased-finetuned-sst-2-english"
-            )
-        return cls._pipeline
-
-    @classmethod
-    def _hash(cls, text: str) -> str:
-        return hashlib.sha256(text.encode()).hexdigest()
-
-    @classmethod
-    @lru_cache(maxsize=128)
-    def _analyze(cls, text: str):
-        pipe = cls._get_pipeline()
-        return pipe(text)[0]
-
-    @classmethod
-    def compute(cls, text: str):
-        """Trigger inference and update latest result."""
-        res = cls._analyze(text)
-        cls.latest_id += 1
-        cls.latest_result = {
-            "text": text,
-            "label": res.get("label"),
-            "score": round(res.get("score", 0.0), 4)
-        }
-        logging.info("✅ Sentiment computed: %s", cls.latest_result)
+
+# Primary and secondary APIs for crypto prices
+COINGECKO_URL = (
+    "https://api.coingecko.com/api/v3/simple/price"
+    "?ids=bitcoin,ethereum,dogecoin&vs_currencies=usd"
+)
+COINCAP_URL = (
+    "https://api.coincap.io/v2/assets?ids=bitcoin,ethereum,dogecoin"
+)
+
+# Shared price cache
+CURRENT_PRICES = {"bitcoin": "--", "ethereum": "--", "dogecoin": "--"}
+
+# How often to retry each API before falling back (seconds)
+RETRY_DELAY = 5
+
+
+def fetch_prices() -> None:
+    """
+    Try CoinGecko first; on 429 or error, fall back to CoinCap.
+    Updates CURRENT_PRICES in-place.
+    """
+    global CURRENT_PRICES
+    apis = [
+        ("CoinGecko", COINGECKO_URL),
+        ("CoinCap", COINCAP_URL)
+    ]
+
+    for name, url in apis:
+        try:
+            resp = httpx.get(url, timeout=10)
+            # Handle CoinCap JSON structure separately
+            if name == "CoinGecko":
+                resp.raise_for_status()
+                data = resp.json()
+                prices = {
+                    "bitcoin": data["bitcoin"]["usd"],
+                    "ethereum": data["ethereum"]["usd"],
+                    "dogecoin": data["dogecoin"]["usd"]
+                }
+            else:
+                resp.raise_for_status()
+                data = resp.json().get("data", [])
+                prices = {item["id"]: float(item["priceUsd"]) for item in data}
+
+            CURRENT_PRICES.update(prices)
+            logging.info("✅ [%s] prices updated: %s", name, prices)
+            return
+
+        except httpx.HTTPStatusError as e:
+            status = e.response.status_code
+            if status == 429:
+                logging.warning("⚠️ [%s] rate limit (429). Retrying fallback.", name)
+            else:
+                logging.warning("⚠️ [%s] HTTP error %s: %s", name, status, e)
+        except Exception as e:
+            logging.warning("⚠️ [%s] fetch error: %s", name, e)
+
+        # If we reach here, wait a bit before trying next API
+        try:
+            import time; time.sleep(RETRY_DELAY)
+        except Exception:
+            pass
+
+    logging.error("❌ All price APIs failed. Keeping previous prices: %s", CURRENT_PRICES)