Update apis/search_api.py

apis/search_api.py

@@ -3,12 +3,12 @@ import os
 import sys
 import uvicorn
 
-from fastapi import FastAPI, Depends
-from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from starlette.middleware.base import BaseHTTPMiddleware
 from pydantic import BaseModel, Field
 from typing import Union
 from sse_starlette.sse import EventSourceResponse, ServerSentEvent
-from utils.logger import logger
 from networks.google_searcher import GoogleSearcher
 from networks.webpage_fetcher import BatchWebpageFetcher
 from documents.query_results_extractor import QueryResultsExtractor
@@ -25,6 +25,7 @@ class SearchAPIApp:
             version="1.0",
         )
         self.setup_routes()
+        self.setup_middleware()
 
     class QueriesToSearchResultsPostItem(BaseModel):
         queries: list = Field(
@@ -134,6 +135,25 @@ class SearchAPIApp:
             summary="Search queries, and extract contents from results",
         )(self.queries_to_search_results)
 
+    def setup_middleware(self):
+        self.app.add_middleware(
+            CORSMiddleware,
+            allow_origins=["*"],
+            allow_credentials=True,
+            allow_methods=["*"],
+            allow_headers=["*"],
+        )
+
+        class NoCacheMiddleware(BaseHTTPMiddleware):
+            async def dispatch(self, request, call_next):
+                response = await call_next(request)
+                response.headers['Cache-Control'] = 'no-cache, no-store, must-revalidate'
+                response.headers['Pragma'] = 'no-cache'
+                response.headers['Expires'] = '0'
+                return response
+
+        self.app.add_middleware(NoCacheMiddleware)
+
 
 class ArgParser(argparse.ArgumentParser):
     def __init__(self, *args, **kwargs):