src/api.py

from fastapi import FastAPI, APIRouter, Depends, HTTPException, Request, Response
from fastapi.responses import JSONResponse
import schedule
import time
import requests
import threading
import asyncio
import uvicorn
from multiprocessing import Process

from src.llm.routes import router as conversation_router
from src.llm.core.config import settings
from src.llm.agents.conversation_agent import ConversationAgent


app = FastAPI(
    title="TheryAI API",
    description="API for TheryAI",
    version="0.1.0",
    docs_url="/docs",
    redoc_url="/redoc",
    openapi_url="/openapi.json",
    debug=True,
)



app.include_router(conversation_router)

@app.get("/")
async def home():
    return {"message": "Welcome to TheryAI API"}

@app.get("/health")
async def health():
    status: dict = {"redis": "unknown", "postgres": "unknown"}

    # ── Redis check ─────────────────────────────────────────────────────
    try:
        import redis as redis_lib
        r = redis_lib.from_url(settings.effective_redis_url, socket_connect_timeout=3)
        r.ping()
        status["redis"] = "ok"
    except Exception as exc:
        status["redis"] = f"error: {exc}"

    # ── Postgres check ───────────────────────────────────────────────────
    if settings.POSTGRES_URL:
        try:
            import psycopg2
            conn = psycopg2.connect(settings.POSTGRES_URL, connect_timeout=3)
            conn.close()
            status["postgres"] = "ok"
        except Exception as exc:
            status["postgres"] = f"error: {exc}"
    else:
        status["postgres"] = "not configured"

    # ── Overall status ───────────────────────────────────────────────────
    all_ok = all(v == "ok" for v in status.values() if v != "not configured")
    return JSONResponse(
        content={"status": "ok" if all_ok else "degraded", **status},
        status_code=200 if all_ok else 503,
    )

def ping_server():
    try:
        print("Pinging server")
        response = requests.get("thery.up.railway.app")
    except requests.exceptions.RequestException as e:
        print("Server is down")
        # send email to admin
    
schedule.every(10).minutes.do(ping_server)


def run_schedule():
    while True:
        schedule.run_pending()
        time.sleep(1)


thread = threading.Thread(target=run_schedule)
thread.daemon = True
thread.start()


def run_fastapi():
    uvicorn.run(
        "src.api:app",
        host="0.0.0.0",
        port=8000,
        log_level="info",
        reload=True,
    )
    
if __name__ == "__main__":
    run_fastapi()