from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline

app = FastAPI()
summarizer = pipeline("summarizer", model="facebook/bart-large-cnn")

class SummarizationRequest(BaseModel):
    inputs: str

@app.post("/summarize")
async def summarize_text(request: SummarizationRequest):
    summary = summarizer(request.inputs, max_legth=200, min_length=30, do_sample=False)
    return summary

@app.get("/")
def greet_json():
    return {"message": "BART Summarizer API is running"}