Update app.py

app.py

@@ -1,109 +1,90 @@
-# app.py
-#
-# Hugging Face Space for a live U.S.-debt counter
-# now exposes:
-#   • /api/debt   – JSON with current total and per-second rate
-#   • /embed.js   – tiny loader that paints any <div class="ntd-debt-clock">
-#                   inside the host page (works under HF CSP)
-
-import os, time, requests
-from datetime import datetime
-from flask import Flask, jsonify, send_from_directory
-from flask_cors import CORS
-
-# ---------------------------------------------------------------------
-#  Fetch last two debt records from Treasury API and calculate rate
-# ---------------------------------------------------------------------
-
-API_URL = "https://api.fiscaldata.treasury.gov/services/api/fiscal_service/v2/accounting/od/debt_to_penny"
-FIELDS = "record_date,tot_pub_debt_out_amt"
-PARAMS = {
-    "fields": FIELDS,
-    "sort": "-record_date",
-    "page[size]": 2
+from flask import Flask, jsonify
+import requests
+import threading
+import time
+import os
+
+app = Flask(__name__)
+
+# ─────── LIVE STATE ─────────────────────────
+DEBT_STATE = {
+    "starting_debt": 0,
+    "rate_per_sec": 0,
+    "last_updated": time.time()
 }
 
-try:
-    r = requests.get(API_URL, params=PARAMS, timeout=10)
-    r.raise_for_status()
-    data = r.json()["data"]
+# ─────── API FETCH ─────────────────────────
+def get_debt_data():
+    url = "https://api.fiscaldata.treasury.gov/services/api/fiscal_service/v2/accounting/od/debt_to_penny"
+    url += "?fields=record_date,tot_pub_debt_out_amt"
+    url += "&sort=-record_date&page[size]=10"
+
+    resp = requests.get(url, timeout=10)
+    records = resp.json()["data"]
+
+    return [
+        (r["record_date"], float(r["tot_pub_debt_out_amt"]))
+        for r in records
+    ]
+
+# ─────── BACKGROUND REFRESH ────────────────
+def refresh_debt_periodically():
+    while True:
+        try:
+            records = get_debt_data()
+            today, today_amt = records[0]
+            prev_day, prev_amt = None, None
+
+            for r in records[1:]:
+                if r[1] != today_amt:
+                    prev_day, prev_amt = r
+                    break
 
-    # Parse records
-    d1 = float(data[0]["tot_pub_debt_out_amt"].replace(",", ""))
-    d2 = float(data[1]["tot_pub_debt_out_amt"].replace(",", ""))
-    t1 = datetime.strptime(data[0]["record_date"], "%Y-%m-%d").timestamp()
-    t2 = datetime.strptime(data[1]["record_date"], "%Y-%m-%d").timestamp()
+            if prev_day:
+                t1 = time.mktime(time.strptime(today, "%Y-%m-%d"))
+                t2 = time.mktime(time.strptime(prev_day, "%Y-%m-%d"))
+                rate = (today_amt - prev_amt) / (t1 - t2)
 
-    # Calculate rate per second
-    RATE_PER_SECOND = (d1 - d2) / (t1 - t2)
-    STARTING_DEBT = d1
-    START_TIMESTAMP = t1
+                # optional: freeze rate if negative
+                if rate < 0:
+                    rate = 0
 
-except Exception as e:
-    print("Error fetching Treasury data:", e)
-    # fallback if API fails
-    STARTING_DEBT = 34_457_892_345_678.12
-    RATE_PER_SECOND = 10_000.23
-    START_TIMESTAMP = time.time()
+                DEBT_STATE["starting_debt"] = today_amt
+                DEBT_STATE["rate_per_sec"] = rate
+                DEBT_STATE["last_updated"] = time.time()
 
-# ---------------------------------------------------------------------
+                print(f"✅ Refreshed: {today_amt=:.2f}, {rate=:.2f}/sec")
 
-app = Flask(__name__, static_folder="templates")
-CORS(app)
+        except Exception as e:
+            print("❌ Error refreshing debt:", e)
 
+        time.sleep(300)  # 5 minutes
+
+
+# ─────── ROUTES ────────────────────────────
 @app.route("/")
-def root():
-    return send_from_directory("static", "index.html")
+def home():
+    return "✅ Debt clock is running", 200
 
 @app.route("/api/debt")
-def api_debt():
-    elapsed = time.time() - START_TIMESTAMP
-    current = STARTING_DEBT + RATE_PER_SECOND * elapsed
-    return jsonify(startingDebt=current, ratePerSecond=RATE_PER_SECOND)
-
-@app.route("/embed.js")
-def embed():
-    js = """
-    (() => {
-      const API = "https://ntdservices-debt-clock.hf.space/api/debt";
-      const CLS = "ntd-debt-clock";
-      function mount(el) {
-        Object.assign(el.style, {
-  font: "700 5.4rem/1.2 'Courier New', monospace",
-  color: "#0f0",
-  background: "linear-gradient(145deg, #0a0a0a, #1a1a1a)",
-  padding: "12px 20px",
-  borderRadius: "12px",
-  whiteSpace: "nowrap",
-  boxShadow: "0 0 12px rgba(0,255,0,0.5), 0 0 30px rgba(0,255,0,0.2) inset",
-  border: "2px solid rgba(0,255,0,0.3)",
-  letterSpacing: "1.5px",
-  transition: "all 0.3s ease-out",
-  textShadow: "0 0 2px #0f0, 0 0 8px #0f0",
-  marginTop: "-35vh"
-});
-        fetch(API).then(r => r.json()).then(d => {
-          let debt = d.startingDebt;
-          const rate = d.ratePerSecond;
-          let last = performance.now();
-          function tick(now) {
-            const dt = (now - last) / 1000;
-            last = now;
-            debt += rate * dt;
-            el.textContent = "$" + Math.floor(debt).toLocaleString();
-            requestAnimationFrame(tick);
-          }
-          requestAnimationFrame(tick);
-        }).catch(() => { el.textContent = "N/A"; });
-      }
-      document.querySelectorAll("." + CLS).forEach(mount);
-    })();
-    """
-    return js, 200, {
-        "Content-Type": "application/javascript; charset=utf-8",
-        "Cache-Control": "public, max-age=31536000, immutable"
-    }
+def get_debt_api():
+    now = time.time()
+    elapsed = now - DEBT_STATE["last_updated"]
+    est = DEBT_STATE["starting_debt"] + DEBT_STATE["rate_per_sec"] * elapsed
+    return jsonify({
+        "startingDebt": est,
+        "ratePerSecond": DEBT_STATE["rate_per_sec"]
+    })
+
+@app.route("/api/ping")
+def ping():
+    return {"status": "ok", "time": time.strftime("%Y-%m-%d %H:%M:%S")}, 200
 
+
+# ─────── APP START ─────────────────────────
 if __name__ == "__main__":
+    print("✅ Starting debt refresh thread...")
+    threading.Thread(target=refresh_debt_periodically, daemon=True).start()
+
     port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port)