Update api.py

api.py

@@ -133,14 +133,14 @@ class AuthenticationError(BaseModel):
     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.")
+    summarizer_error: str = Field("Error: 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.")
+    summaries: str = Field("", description="Empty string as summarizer encountered an error")
+    summarizer_error: str = Field("Error: Summarizer Error with a message describing the error")
 
 
 class NewsSummarizerAPIAuthenticationError(Exception):
@@ -157,8 +157,8 @@ def authenticate_key(api_key: str):
 
 @app.post("/generate_summary/", response_model=List[SuccessfulResponse],
          responses={
-        401: {"model": AuthenticationError, "description": "authentication Error"}, 
-        500: {"model": AuthenticationError, "description": "authentication Error"}
+        401: {"model": AuthenticationError, "description": "Authentication Error"}, 
+        500: {"model": SummaryError, "description": "Summarizer Error"}
          })
 async def read_items(q: URLList):
     try:
@@ -174,7 +174,8 @@ async def read_items(q: URLList):
         _ = authenticate_key(api_key)
         scraped_texts, scrape_errors = await scrape_urls(urls)
         
-        if [*set(scraped_texts)][0] == "":
+        unique_scraped_texts = [*set(scraped_texts)]
+        if (unique_scraped_texts[0] == "") and (len(unique_scraped_texts) == 1):
             raise NewsSummarizerAPIScrapingError("Scrape Error: Couldn't scrape text from any of the URLs")
             
         summaries = await summ_inference(scraped_texts)
@@ -184,7 +185,7 @@ async def read_items(q: URLList):
         status_code = 500
         if e.__class__.__name__ == "NewsSummarizerAPIAuthenticationError":
             status_code = 401
-        response_json = {'urls': urls, 'scraped_texts': scraped_texts, 'scrape_errors': scrape_errors, 'summaries': "", 'summarizer_error': f'error: {e}'}
+        response_json = {'urls': urls, 'scraped_texts': scraped_texts, 'scrape_errors': scrape_errors, 'summaries': "", 'summarizer_error': f'Error: {e}'}
 
     json_str = json.dumps(response_json, indent=5) # convert dict to JSON str
     return Response(content=json_str, media_type='application/json', status_code=status_code)