from fastapi import FastAPI
from pydantic import BaseModel

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

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

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

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