refactor: added dto

app.py

@@ -4,6 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from extractive import ExtractiveModel, _get_summary
 from cosine import CosineModel, build_summary
 from abstractive import Abstractive
+from dto import SummarizeRequest
 
 from transformers import pipeline
 
@@ -23,16 +24,12 @@ app.add_middleware(
 )
 
 
-class SummaryRequest:
-    def __init__(self, text:str, model:int) -> None:
-        self.text = text
-        self.model = model
-
 
 @app.post("/textsummarize")
-def summarizeText(request_body: SummaryRequest):
+def summarizeText(request_body:SummarizeRequest):
 
-    text,model = request_body["text"],request_body["model"]
+    text = request_body.text
+    model = int(request_body.model)
     summary="error"
 
     if model==0:

dto.py

@@ -0,0 +1,12 @@
+from datetime import datetime
+from typing import List
+from typing import Optional
+
+from pydantic import BaseModel
+from pydantic import Field
+
+
+class SummarizeRequest(BaseModel):
+    """Text Summarize request model."""
+    text: str = Field(..., description="The text you want to summarize", examples=["ከተለያዩ የትግራይ አካባቢዎች በመምጣት ኑሯቸውን ..."])
+    model: Optional[int] = Field(None, description="model index 0 = ExtractiveModel, 1 = Cosine Similarity, 3 = Abstractive - News Headline", examples=[0,1,2])