from fastapi import FastAPI
from pydantic import BaseModel

# импорт твоей логики
from new_sum import generate_summary

# ======================
# APP INIT
# ======================
app = FastAPI()

@app.get("/")
def root():
    return {"status": "ok"}

@app.get("/health")
def health():
    return {"model": "loaded"}

# ======================
# REQUEST SCHEMA
# ======================
class NewsRequest(BaseModel):
    text: str

# ======================
# API ENDPOINT
# ======================
@app.post("/summarize")
def summarize(req: NewsRequest):
    summary = generate_summary(req.text)
    return {"summary": summary}