Create main.py

main.py

@@ -0,0 +1,91 @@
+from fastapi import FastAPI, UploadFile, File, Body
+from apscheduler.schedulers.background import BackgroundScheduler
+from datetime import datetime, timedelta
+import pandas as pd
+import requests
+import os
+
+app = FastAPI()
+scheduler = BackgroundScheduler()
+scheduler.start()
+
+# --- Helper job ---
+def do_post(url: str, payload: dict = None):
+    try:
+        resp = requests.post(url, json=payload or {})
+        print(f"[{datetime.utcnow()}] POST {url} → {resp.status_code}")
+    except Exception as e:
+        print(f"[{datetime.utcnow()}] ERROR posting to {url}: {e}")
+
+# --- 1A: Schedule from uploaded CSV ---
+@app.post("/schedule_csv/")
+async def schedule_csv(
+    csv_file: UploadFile = File(...),
+    delay_hours: float = Body(2.0, description="Delay before first run, in hours"),
+    repeat_hours: float = Body(None, description="If set, job will repeat every N hours")
+):
+    df = pd.read_csv(csv_file.file)
+    urls = df["endpoint"].dropna().tolist()
+    for url in urls:
+        if repeat_hours:
+            scheduler.add_job(
+                do_post,
+                trigger="interval",
+                hours=repeat_hours,
+                next_run_time=datetime.utcnow() + timedelta(hours=delay_hours),
+                args=[url],
+                id=f"interval_{url}"
+            )
+        else:
+            scheduler.add_job(
+                do_post,
+                trigger="date",
+                run_date=datetime.utcnow() + timedelta(hours=delay_hours),
+                args=[url],
+                id=f"once_{url}"
+            )
+    return {"scheduled": len(urls)}
+
+# --- 1B: Schedule from JSON body (n8n could use this) ---
+@app.post("/schedule_json/")
+async def schedule_json(
+    endpoints: list[str] = Body(..., description="List of URLs to call"),
+    delay_hours: float = Body(2.0),
+    repeat_hours: float = Body(None)
+):
+    for url in endpoints:
+        if repeat_hours:
+            scheduler.add_job(
+                do_post,
+                trigger="interval",
+                hours=repeat_hours,
+                next_run_time=datetime.utcnow() + timedelta(hours=delay_hours),
+                args=[url],
+                id=f"interval_{url}"
+            )
+        else:
+            scheduler.add_job(
+                do_post,
+                trigger="date",
+                run_date=datetime.utcnow() + timedelta(hours=delay_hours),
+                args=[url],
+                id=f"once_{url}"
+            )
+    return {"scheduled": len(endpoints)}
+
+# --- 2: Self‑ping to prevent idle shutdown (optional) ---
+HF_SPACE_URL = os.getenv("HF_SPACE_URL")  # e.g. "https://my-username-hf-space.hf.space"
+if HF_SPACE_URL:
+    def heartbeat():
+        try:
+            requests.get(f"{HF_SPACE_URL}/heartbeat")
+            print("Self-ping OK")
+        except:
+            pass
+
+    # ping every hour to keep the Space “active” in HF’s eyes
+    scheduler.add_job(heartbeat, "interval", hours=1, id="self_ping")
+
+@app.get("/heartbeat/")
+async def heartbeat_ok():
+    return {"alive": True}