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"]
)

VIBESTREAM_API = "https://vibestream-production-64f3.up.railway.app"

# Track current job
current_job_id = None
YOUTUBE_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"{VIBESTREAM_API}/live-monitor", json={
            "youtube_url": YOUTUBE_URL,
            "condition": CONDITION,
            "webhook_url": f"{os.environ['WEBHOOK_HOST']}/snow-alert",
            "interval_seconds": 30
        })

        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") == "live_monitor_result":
        data = payload.get("data", {})

        if data.get("triggered"):
            # 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)
    elif payload.get("type") == "job_status":
        status = payload.get("status")
        details = payload.get("details", {})

        if status == "stopped":
            # Check if auto-stopped due to 10-minute limit
            if details.get("auto_stopped"):
                print("Job auto-stopped after 10 minutes, restarting...")
                background_tasks.add_task(start_live_monitor_job)
            else:
                print("Job was manually stopped")

    return {"status": "ok"}