minor improvements

1 files changed, 27 insertions, 0 deletions

diff --git a/app/utils/cors.py b/app/utils/cors.py
new file mode 100644
index 0000000..e7b54e8
--- /dev/null
+++ b/app/utils/cors.py
@@ -0,0 +1,27 @@
+import os
+
+from dotenv import load_dotenv
+from fastapi.middleware.cors import CORSMiddleware
+
+load_dotenv()
+
+
+def setup_cors(app):
+
+    dev_origin = os.getenv("FRONTEND_URL", "http://localhost:3000")
+    prod_origin = os.getenv("PROD_FRONTEND_URL")
+
+    origins = [dev_origin, prod_origin]
+
+    filtered_origins = []
+    for origin in origins:
+        if origin:
+            filtered_origins.append(origin)
+
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=filtered_origins,
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )