Snow Alert for Pittsburgh

from fastapi import FastAPI, Request, BackgroundTasks
from twilio.rest import Client
import httpx
import os

app = FastAPI()

twilio_client = Client(
    os.environ["TWILIO_ACCOUNT_SID"],
    os.environ["TWILIO_AUTH_TOKEN"]
)

TRIO_API_BASE = "https://trio.machinefi.com/api"
TRIO_API_KEY = os.environ["TRIO_API_KEY"]

# Track current job
current_job_id = None
STREAM_URL = "https://youtube.com/watch?v=PITTSBURGH_LIVE_CAM"
CONDITION = "Is it snowing? Look for falling snowflakes or white accumulation."

async def start_live_monitor_job():
    """Start or restart the live monitor job."""
    global current_job_id

    async with httpx.AsyncClient() as client:
        response = await client.post(
            f"{TRIO_API_BASE}/live-monitor",
            json={
                "stream_url": STREAM_URL,
                "condition": CONDITION,
                "webhook_url": f"{os.environ['WEBHOOK_HOST']}/snow-alert",
                "interval_seconds": 30,
            },
            headers={"Authorization": f"Bearer {TRIO_API_KEY}"},
        )

        if response.status_code == 200:
            current_job_id = response.json()["job_id"]
            print(f"Started job: {current_job_id}")
        else:
            print(f"Failed to start job: {response.text}")

@app.on_event("startup")
async def startup():
    """Start monitoring on server startup."""
    await start_live_monitor_job()

@app.post("/snow-alert")
async def snow_alert(request: Request, background_tasks: BackgroundTasks):
    payload = await request.json()

    # Handle trigger events
    if payload.get("type") == "watch_triggered":
        data = payload.get("data", {})

        # Send SMS alert
        twilio_client.messages.create(
            body=f"It's snowing in Pittsburgh!\n\n{data.get('explanation', '')}",
            from_=os.environ["TWILIO_PHONE"],
            to=os.environ["MY_PHONE"],
        )
        print("Snow alert sent!")

    # Handle job status events (auto-stop after 10 min or after trigger)
    elif payload.get("type") == "job_stopped":
        data = payload.get("data", {})

        # Restart for continuous coverage when the job auto-stops
        if data.get("auto_stopped"):
            reason = data.get("reason", "unknown")
            print(f"Job auto-stopped ({reason}), restarting...")
            background_tasks.add_task(start_live_monitor_job)
        else:
            print("Job was manually stopped")

    return {"status": "ok"}