Update api.py

api.py

@@ -6,7 +6,7 @@ import tensorflow as tf
 from scraper import scrape_text
 from fastapi import FastAPI, Response
 from typing import List
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, ValidationError
 import uvicorn
 import json
 import logging
@@ -135,6 +135,21 @@ def authenticate_key(api_key: str):
         raise NewsSummarizerAPIAuthenticationError("Authentication error: Invalid API key.")
 
 
+@app.exception_handler(ValidationError)
+async def validation_exception_handler(request: Request, exc: ValidationError):
+    urls = request.query_params.getlist("urls")
+    error_details = exc.errors()
+    error_messages = []
+    for error in error_details:
+        loc = '->'.join(map(str, error['loc']))
+        msg = error['msg']
+        error_messages.append(f"Error at {loc}: {msg}")
+    error_message = "; ".join(error_messages) if error_messages else "Validation error"
+    response_json = {'urls': urls, 'scraped_texts': '', 'scrape_errors': '', 'summaries': "", 'summarizer_error': f'Error: {error_message}'}
+    json_str = json.dumps(response_json, indent=5) # convert dict to JSON str
+    return Response(content=json_str, media_type='application/json', status_code=422)
+
+
 @app.post("/generate_summary/", tags=["Generate Summary"], response_model=List[SuccessfulResponse],
          responses={
         401: {"model": AuthenticationError, "description": "Authentication Error"},