Update api.py

api.py

@@ -121,13 +121,20 @@ class URLList(BaseModel):
     urls: List[str] = Field(..., description="List of URLs of news articles to generate summaries")
     key: str = Field(..., description="Authentication Key")
 
-class Summary(BaseModel):
+class SuccessfulResponse(BaseModel):
     urls: List[str] = Field(..., description="List of URLs of news articles inputted by the user")
-    scraped_texts: List[str] = Field(..., description="List of scraped text from input URLs")
-    scrape_errors: List[str] = Field(..., description="List of errors raised during scraping. One item for corresponding URL")
+    scraped_texts: List[str] = Field(..., description="List of scraped text from input URLs".)
+    scrape_errors: List[str] = Field("", description="List of errors raised during scraping. One item for corresponding URL.")
     summaries: List[str] = Field(..., description="List of generated summaries of news articles")
-    summarizer_error: str = Field(..., description="String specifying error raised during summary.")
-    
+    summarizer_error: str = Field("", description="String specifying error raised during summary.")
+
+class AuthenticationError(BaseModel):
+    urls: List[str] = Field(..., description="List of URLs of news articles inputted by the user")
+    scraped_texts: List[str] = Field("", description="List of empty strings as authentication failed")
+    scrape_errors: List[str] = Field("", description="List of empty strings as authentication failed")
+    summaries: List[str] = Field("", description="List of empty strings as authentication failed")
+    summarizer_error: str = Field("Authentication error: Invalid API key.")
+
 
 class NewsSummarizerAPIAuthenticationError(Exception):
     pass 
@@ -138,7 +145,10 @@ def authenticate_key(api_key: str):
         raise NewsSummarizerAPIAuthenticationError("Authentication error: Invalid API key.")
 
 
-@app.post("/generate_summary/", response_model=List[Summary])
+@app.post("/generate_summary/", response_model=List[SuccessfulResponse],
+         responses={
+        401: {"model": AuthenticationError, "description": "authentication Error"}
+         })
 async def read_items(q: URLList):
     try:
         logging.warning("Entering read_items()")