feat: returning post_id

app.py

@@ -7,14 +7,20 @@ summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 class SummarizationRequest(BaseModel):
     inputs: str
+    post_id: str
 
-@app.post("/summarize")
+class SummarizationResponse(BaseModel):
+    post_id: str
+    summary: str
+
+@app.post("/summarize", response_model=SummarizationResponse)
 async def summarize_text(request: SummarizationRequest):
     input_length = len(request.inputs.split())
     max_length = max(50, int(input_length * 0.5))
     min_length = max(20, int(input_length * 0.2))
+
     summary = summarizer(request.inputs, max_length=max_length, min_length=min_length, do_sample=False)
-    return {"summary": summary[0]["summary_text"]}
+    return {"post_id": request.post_id, "summary": summary[0]["summary_text"]}
 
 @app.get("/")
 def greet_json():