Jelajahi Sumber

Add CORS for local testing

Daoud Clarke 3 tahun lalu
induk
melakukan
a93fbe9d66
1 mengubah file dengan 12 tambahan dan 5 penghapusan
  1. 12 5
      mwmbl/main.py

+ 12 - 5
mwmbl/main.py

@@ -5,6 +5,7 @@ from multiprocessing import Process
 
 import uvicorn
 from fastapi import FastAPI
+from starlette.middleware.cors import CORSMiddleware
 
 from mwmbl import background
 from mwmbl.indexer import historical, retrieve, preprocess, update_pages
@@ -41,11 +42,7 @@ def run():
         print("Creating a new index")
         TinyIndex.create(item_factory=Document, index_path=args.index, num_pages=NUM_PAGES, page_size=PAGE_SIZE)
 
-    Process(target=background.run, args=(args.index,)).start()
-    # Process(target=historical.run).start()
-    # Process(target=retrieve.run).start()
-    # Process(target=preprocess.run, args=(args.index,)).start()
-    # Process(target=update_pages.run, args=(args.index,)).start()
+    # Process(target=background.run, args=(args.index,)).start()
 
     completer = Completer()
 
@@ -55,6 +52,16 @@ def run():
         # Initialize FastApi instance
         app = FastAPI()
 
+        origins = ["*"]
+
+        app.add_middleware(
+            CORSMiddleware,
+            allow_origins=origins,
+            allow_credentials=True,
+            allow_methods=["*"],
+            allow_headers=["*"],
+        )
+
         search_router = search.create_router(ranker)
         app.include_router(search_router)
         app.include_router(crawler_router)