Update api.py

api.py

@@ -124,21 +124,31 @@ class URLList(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")
+    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="Blank as the response code is 200")
 
 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")
+    scraped_texts: str = Field("", description="Empty string as authentication failed")
+    scrape_errors: str = Field("", description="Empty string as authentication failed")
+    summaries: str = Field("", description="Empty string as authentication failed")
+    summarizer_error: str = Field("Authentication error: Invalid API key.")
+
+class SummaryError(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")
+    summaries: List[str] = Field("", description="List of empty summaries as summarizer returned an error")
     summarizer_error: str = Field("Authentication error: Invalid API key.")
 
 
 class NewsSummarizerAPIAuthenticationError(Exception):
     pass 
 
+class NewsSummarizerAPIScrapingError(Exception):
+    pass 
+
 
 def authenticate_key(api_key: str):
     if api_key != os.getenv('API_KEY'):
@@ -147,7 +157,8 @@ def authenticate_key(api_key: str):
 
 @app.post("/generate_summary/", response_model=List[SuccessfulResponse],
          responses={
-        401: {"model": AuthenticationError, "description": "authentication Error"}
+        401: {"model": AuthenticationError, "description": "authentication Error"}, 
+        500: {"model": AuthenticationError, "description": "authentication Error"}
          })
 async def read_items(q: URLList):
     try:
@@ -162,6 +173,10 @@ async def read_items(q: URLList):
         api_key = request_json['key']
         _ = authenticate_key(api_key)
         scraped_texts, scrape_errors = await scrape_urls(urls)
+        
+        if [*set(scraped_texts)][0] == "":
+            raise NewsSummarizerAPIScrapingError("Scrape Error: Couldn't scrape text from any of the URLs")
+            
         summaries = await summ_inference(scraped_texts)
         status_code = 200
         response_json = {'urls': urls, 'scraped_texts': scraped_texts, 'scrape_errors': scrape_errors, 'summaries': summaries, 'summarizer_error': ''}