Update api.py

api.py

@@ -120,6 +120,13 @@ class URLList(BaseModel):
     urls: List[str]
     key: str
 
+class Summary(BaseModel):
+    urls: List[str]
+    scraped_texts: List[str]
+    scrape_errors: List[str]
+    summaries: List[str]
+    summarizer_error: str
+    
 
 class NewsSummarizerAPIAuthenticationError(Exception):
     pass 
@@ -130,7 +137,7 @@ def authenticate_key(api_key: str):
         raise NewsSummarizerAPIAuthenticationError("Authentication error: Invalid API key.")
 
 
-@app.post("/generate_summary/")
+@app.post("/generate_summary/", response_model=List[Summary])
 async def read_items(q: URLList):
     try:
         logging.warning("Entering read_items()")